GPU Test skips - fbcode/pytorch/data/test

Differential Revision: D60701549 Pull Request resolved: #1314
pytorch · Aug 14, 2024 · 5b7068c · 5b7068c
1 parent c0fa044
commit 5b7068c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_distributed.py b/test/test_distributed.py
@@ -235,7 +235,7 @@ def test_elastic_training_dl2(self, backend) -> None:
                 "--run_path",
                 f"--nnodes={nnodes}",
                 f"--nproc_per_node={world_size}",
-                abs_path("bin/elastic_training.py"),
+                abs_path("elastic_training.py"),
                 "--" + backend,
                 "--dl2",
             ],
@@ -258,7 +258,7 @@ def test_elastic_training_dl1(self, backend) -> None:
                 "--run_path",
                 f"--nnodes={nnodes}",
                 f"--nproc_per_node={world_size}",
-                abs_path("bin/elastic_training.py"),
+                abs_path("elastic_training.py"),
                 "--" + backend,
                 "--dl1",
             ],