Yao Matrix 3bd1c20149 enable misc cases on XPU & use device agnostic APIs for cases in tests (#38192) ... * use device agnostic APIs in tests Signed-off-by: Matrix Yao <matrix.yao@intel.com> * more Signed-off-by: Matrix Yao <matrix.yao@intel.com> * fix style Signed-off-by: Matrix Yao <matrix.yao@intel.com> * add reset_peak_memory_stats API Signed-off-by: YAO Matrix <matrix.yao@intel.com> * update --------- Signed-off-by: Matrix Yao <matrix.yao@intel.com> Signed-off-by: YAO Matrix <matrix.yao@intel.com> Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>		2025-05-20 10:09:01 +02:00
..
__init__.py
test_data_collator.py	add FlashAttentionKwargs and seq_idx to flat collator (#36456)	2025-04-16 15:45:03 +02:00
test_trainer_callback.py	fix: prevent second save in the end of training if last step was saved already (#36219)	2025-02-20 17:38:52 +01:00
test_trainer_distributed_loss.py	fix bug in distributed loss test (#38166)	2025-05-16 16:21:35 +00:00
test_trainer_distributed_worker_seed.py	enable trainer test cases on xpu (#38138)	2025-05-15 12:17:44 +00:00
test_trainer_distributed.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_trainer_fsdp.py	fix FSDP + torch.compile bug when saving pretrained model (#37725)	2025-05-06 17:51:28 +02:00
test_trainer_seq2seq.py	[generation] Less verbose warnings by default (#38179)	2025-05-19 10:03:37 +00:00
test_trainer_tpu.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_trainer_utils.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02:00
test_trainer.py	enable misc cases on XPU & use device agnostic APIs for cases in tests (#38192)	2025-05-20 10:09:01 +02:00