Skip to content

Commit ed27288

Browse files
committed
Fix test gpu
1 parent f40a096 commit ed27288

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

test/test_gpu/main.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,11 @@
1818
fbcode_skip_file_path = "fb/skip_tests_h100_fbcode.yaml"
1919
SKIP_FILE = importlib.resources.files(__package__).joinpath(fbcode_skip_file_path)
2020
else:
21-
SKIP_FILE = "skip_tests_h100_pytorch.yaml"
21+
import os
22+
23+
SKIP_FILE = os.path.abspath(
24+
os.path.dirname(__file__).join("skip_tests_h100_pytorch.yaml")
25+
)
2226

2327
with open(SKIP_FILE, "r") as f:
2428
skip_tests = yaml.safe_load(f)

0 commit comments

Comments
 (0)