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

remove multiple archtictures to isa head and adding gemm tuning scripts (#261)

Browse Source 
* Remove adding multiple architectures to isa head

* Add mask for gpu memory load in scripts for tuning gemm 'script/amd/gemm/matmul.py'

* Move the scripts to a better place 'scripts/amd/gemm/'
This commit is contained in:
Shucai Xiao
2023-07-18 14:21:16 -05:00
committed by GitHub
parent d6e51fd221
commit 1c86e3238a
3 changed files with 211 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
python/triton/compiler/compiler.py
View File

@@ -222,9 +222,6 @@ def get_amdgpu_arch_fulldetails():
arch_name = arch_name_features[0]
arch_features = ""
if (len(arch_name_features) == 3):
arch_features = "+" + re.search('\\w+', arch_name_features[1]).group(0) + ","\
"-" + re.search('\\w+', arch_name_features[2]).group(0)
return [arch_triple, arch_name, arch_features, warp_size]
except BaseException:
return None