github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[BUILD] Reverted ptxas change and fixed bug in cache key computation (#1971)

Browse Source
This commit is contained in:
Philippe Tillet
2023-07-19 20:58:24 -07:00
committed by GitHub
parent 28a61484bc
commit 3452615d79
4 changed files with 29 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/triton/runtime/jit.py
View File

@@ -11,7 +11,7 @@ from collections import defaultdict, namedtuple
from typing import (Callable, Generic, Iterable, List, Optional, TypeVar, Union, cast,
overload)
from ..common.backend import get_backend
from ..common.backend import get_backend, path_to_ptxas
TRITON_PATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
TRITON_VERSION = "2.1.0"
@@ -117,10 +117,8 @@ def version_key():
with open(lib.module_finder.find_spec(lib.name).origin, "rb") as f:
contents += [hashlib.md5(f.read()).hexdigest()]
# ptxas version
try:
ptxas_version = hashlib.md5(subprocess.check_output(["ptxas", "--version"])).hexdigest()
except Exception:
ptxas_version = ''
ptxas = path_to_ptxas()[0]
ptxas_version = hashlib.md5(subprocess.check_output([ptxas, "--version"])).hexdigest()
return '-'.join(TRITON_VERSION) + '-' + ptxas_version + '-' + '-'.join(contents)