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

[FRONTEND][OPTIMIZER] small perf improvements (#1945)

Browse Source
This commit is contained in:
Philippe Tillet
2023-07-14 15:11:36 -07:00
committed by GitHub
parent 80163a9c1e
commit 8207eabd7b
6 changed files with 40 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/setup.py
View File

@@ -125,8 +125,8 @@ def download_and_copy_ptxas():
base_dir = os.path.dirname(__file__)
src_path = "bin/ptxas"
version = "12.1.105"
url = f"https://conda.anaconda.org/nvidia/label/cuda-12.1.1/linux-64/cuda-nvcc-{version}-0.tar.bz2"
version = "12.2.91"
url = f"https://conda.anaconda.org/nvidia/label/cuda-12.2.0/linux-64/cuda-nvcc-{version}-0.tar.bz2"
dst_prefix = os.path.join(base_dir, "triton")
dst_suffix = os.path.join("third_party", "cuda", src_path)
dst_path = os.path.join(dst_prefix, dst_suffix)