ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 18:42:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit 'e6e2e04edbd5766afb388fc4ba64d57a9b52452e' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-11-19 18:15:38 +00:00
parent 29d162bec7
commit 7ed276c492
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)