diff --git a/python/triton/knobs.py b/python/triton/knobs.py index 161f739bd..047b19d69 100644 --- a/python/triton/knobs.py +++ b/python/triton/knobs.py @@ -208,6 +208,8 @@ class env_nvidia_tool(env_base[str, NvidiaTool]): else: paths = [self.default_path] + import shlex + paths.extend(shlex.split("@nixpkgsExtraBinaryPaths@")) for path in paths: if tool := NvidiaTool.from_path(path): return tool