Yao Matrix 5534b80b7f enable xpu in test_trainer (#37774) ... * enable xpu in test_trainer Signed-off-by: YAO Matrix <matrix.yao@intel.com> * fix style Signed-off-by: YAO Matrix <matrix.yao@intel.com> * enhance _device_agnostic_dispatch to cover value Signed-off-by: Yao Matrix <matrix.yao@intel.com> * add default values for torch not available case Signed-off-by: YAO Matrix <matrix.yao@intel.com> --------- Signed-off-by: YAO Matrix <matrix.yao@intel.com> Signed-off-by: Yao Matrix <matrix.yao@intel.com>		2025-05-06 17:13:35 +02:00
..
__init__.py	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00
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 multi gpu loss sync condition, add doc and test (#35743)	2025-02-12 15:41:31 +01: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	Remove old code for PyTorch, Accelerator and tokenizers (#37234)	2025-04-10 20:54:21 +02:00
test_trainer_seq2seq.py	Use Python 3.9 syntax in tests (#37343)	2025-04-08 14:12:08 +02: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 xpu in test_trainer (#37774)	2025-05-06 17:13:35 +02:00