Gaudi3 CI (#38790)

2025-06-23 10:56:51 +02:00
parent 2166b6b4ff
commit 984ff89e73
16 changed files with 618 additions and 14 deletions
--- a/tests/trainer/test_trainer_distributed_loss.py
+++ b/tests/trainer/test_trainer_distributed_loss.py
@@ -18,11 +18,13 @@ from transformers.testing_utils import (
    execute_subprocess_async,
    get_torch_dist_unique_port,
    require_torch_multi_accelerator,
+    run_first,
    torch_device,
 )


 class TestTrainerDistributedLoss(TestCasePlus):
+    @run_first
    @require_torch_multi_accelerator
    def test_trainer(self):
        device_count = backend_device_count(torch_device)