ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-19 22:39:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Inductor] Copy logic for ck-tile gemm instance configuration in Inductor max-autotune integration and test it (#2910)

Browse Source 
* add op, gen_instances and test

---------

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>
This commit is contained in:
Max Podkorytov
2025-11-19 09:38:02 -08:00
committed by GitHub
parent 7fe7aa76f5
commit e6e2e04edb
4 changed files with 221 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/test/test_gen_instances.py
View File

@@ -15,6 +15,9 @@ from ck4inductor.grouped_conv_fwd.gen_instances import (
from ck4inductor.batched_universal_gemm.gen_instances import (
gen_ops_library as gen_batched_gemm_ops_library,
)
from ck4inductor.ck_tile_universal_gemm.gen_instances import (
ops as gen_ck_tile_gemm_ops_library,
)
log = logging.getLogger(__name__)
@@ -43,3 +46,9 @@ class TestGenInstances(unittest.TestCase):
log.debug("%d gemm instances from library" % len(instances))
self.assertTrue(instances)
def test_gen_ck_tile_universal_gemm_instances(self):
instances = gen_ck_tile_gemm_ops_library()
log.debug("%d ck-tile gemm instances from library" % len(instances))
self.assertTrue(instances)