From ee24094b81df4bcd0e126e7466050633cc2bf973 Mon Sep 17 00:00:00 2001 From: zpcore Date: Wed, 31 Jan 2024 18:26:42 +0000 Subject: [PATCH] Fix model skip yaml file not found issue --- benchmarks/torchbench_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmarks/torchbench_model.py b/benchmarks/torchbench_model.py index 0e59bc6941ae..53a5a2815555 100644 --- a/benchmarks/torchbench_model.py +++ b/benchmarks/torchbench_model.py @@ -164,7 +164,7 @@ def _find_near_file(self, names): Looks for `names` in the current directory, up to its two direct parents. """ - for dir in ("./", "../", "../../"): + for dir in ("./", "../", "../../", "../../../"): for name in names: path = os.path.join(dir, name) if exists(path): @@ -193,7 +193,7 @@ def get_skip_data(self): its lists of models into sets of models. """ - benchmarks_dir = self._find_near_file(("benchmarks",)) + benchmarks_dir = self._find_near_file(("pytorch/benchmarks",)) assert benchmarks_dir is not None, "PyTorch benchmarks folder not found." skip_file = os.path.join(benchmarks_dir, "dynamo",