From 1798d883e307a151dda05cec13307161c0d81022 Mon Sep 17 00:00:00 2001 From: Jovian Anthony Jaison Date: Tue, 13 Aug 2024 16:07:46 -0700 Subject: [PATCH] GPU Test skips - fbcode/pytorch/data/test (#1314) Summary: Pull Request resolved: https://github.com/pytorch/data/pull/1314 This diff is part of a series of fixes for skipping GPU tests in the PyTorch data library. Reviewed By: johnsushant Differential Revision: D60701549 --- test/test_distributed.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_distributed.py b/test/test_distributed.py index 09ab0fa4f..84cfff046 100644 --- 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", ],