We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6fb00f3 commit 85685a8Copy full SHA for 85685a8
python/triton/backends/compiler.py
@@ -1,6 +1,7 @@
1
import os
2
import re
3
import subprocess
4
+import sysconfig
5
6
from abc import ABCMeta, abstractmethod, abstractclassmethod
7
from dataclasses import dataclass
@@ -25,7 +26,7 @@ def __init__(self, target: GPUTarget) -> None:
25
26
27
@staticmethod
28
def _path_to_binary(binary: str):
- binary += ".exe" if os.name == "nt" else ""
29
+ binary += sysconfig.get_config_var("EXE")
30
base_dir = os.path.join(os.path.dirname(__file__), os.pardir)
31
paths = [
32
os.environ.get(f"TRITON_{binary.upper()}_PATH", ""),
0 commit comments