Revert "Disambiguate benchmarks directory search. (#6434)" (#6440)

pytorch · Feb 1, 2024 · bc4fee2 · bc4fee2
1 parent 599b925
commit bc4fee2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/benchmarks/torchbench_model.py b/benchmarks/torchbench_model.py
@@ -193,10 +193,11 @@ def get_skip_data(self):
     its lists of models into sets of models.
     """
 
-    benchmarks_dir = self._find_near_file(("benchmarks/dynamo",))
+    benchmarks_dir = self._find_near_file(("benchmarks",))
     assert benchmarks_dir is not None, "PyTorch benchmarks folder not found."
 
-    skip_file = os.path.join(benchmarks_dir, "torchbench_skip_models.yaml")
+    skip_file = os.path.join(benchmarks_dir, "dynamo",
+                             "torchbench_skip_models.yaml")
     with open(skip_file) as f:
       data = yaml.safe_load(f)