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

Fixed a bug related to split_k and prune unnecessary tuning space (#332)

Browse Source 
* refine tuning scrit by adding prune_configs, also fixed a bug in generating tuning configs

* fixed a bug in returning the empty config
This commit is contained in:
Shucai Xiao
2023-09-21 23:47:14 -05:00
committed by GitHub
parent a8574be74d
commit 10795d8fd3
2 changed files with 41 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/triton/runtime/autotuner.py
View File

@@ -96,7 +96,7 @@ class Autotuner(KernelInterface):
key_values.append(kwargs[name])
key = tuple(key_values)
return self.cache[key] if key in self.cache else Config()
return self.cache[key] if key in self.cache else Config({})
def run(self, *args, **kwargs):