Change /usr/local/lib/python3.12/dist-packages/torch/_inductor/bin/ptxas

nWEIdia · nWEIdia · commit 1a3aea468052 · 2025-09-26T19:34:42.000-07:00
to /usr/local/lib/python3.12/dist-packages/torch/bin/ptxas
diff --git a/torch/_inductor/runtime/compile_tasks.py b/torch/_inductor/runtime/compile_tasks.py
@@ -40,7 +40,7 @@ def _reload_python_module(
 def _set_triton_ptxas_path() -> None:
     if os.environ.get("TRITON_PTXAS_PATH") is not None:
         return
-    ptxas = Path(__file__).absolute().parents[1] / "bin" / "ptxas"
+    ptxas = Path(__file__).absolute().parents[2] / "bin" / "ptxas"
     if not ptxas.exists():
         return
     if ptxas.is_file() and os.access(ptxas, os.X_OK):
