Use os.sep in filter_traceback function (#3322)

Signed-off-by: Anatoly Myachev <[email protected]>
intel · Jan 31, 2025 · 3467f99 · 3467f99
1 parent ccf97fd
commit 3467f99
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/triton/compiler/compiler.py b/python/triton/compiler/compiler.py
@@ -190,9 +190,10 @@ def filter_traceback(e: BaseException):
         filter_traceback(e.__context__)
 
     # If a user has a file that matches one of these, they're out of luck.
+    sep = os.sep
     BAD_FILES = [
-        "/triton/compiler/code_generator.py",
-        "/ast.py",
+        f"{sep}triton{sep}compiler{sep}code_generator.py",
+        f"{sep}ast.py",
     ]
 
     tb = e.__traceback__