transformers/tests/trainer
Yao Matrix a5a0c7b888
switch to device agnostic device calling for test cases (#38247)
* use device agnostic APIs in test cases

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* fix style

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* add one more

Signed-off-by: YAO Matrix <matrix.yao@intel.com>

* xpu now supports integer device id, aligning to CUDA behaviors

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* update to use device_properties

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* fix style

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* update comment

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* fix comments

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

* fix style

Signed-off-by: Matrix Yao <matrix.yao@intel.com>

---------

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-26 10:18:53 +02:00
..
__init__.py
test_data_collator.py
test_trainer_callback.py
test_trainer_distributed_loss.py
test_trainer_distributed_worker_seed.py
test_trainer_distributed.py
test_trainer_fsdp.py
test_trainer_seq2seq.py [generation] Less verbose warnings by default (#38179) 2025-05-19 10:03:37 +00:00
test_trainer_tpu.py
test_trainer_utils.py
test_trainer.py switch to device agnostic device calling for test cases (#38247) 2025-05-26 10:18:53 +02:00