Skip to content

Commit 6fdf438

Browse files
committed
add a passing test
1 parent a0e1a5f commit 6fdf438

File tree

8 files changed

+51
-5
lines changed

8 files changed

+51
-5
lines changed

Jenkinsfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -573,7 +573,7 @@ def Build_CK(Map conf=[:]){
573573
. ${env.WORKSPACE}/bin/activate
574574
python3 -m pip install pytest build setuptools setuptools_scm
575575
python3 -m pip install .
576-
python3 -m pytest python/test/test_gen_instances.py
576+
python3 -m pytest python/test/ck4inductor/test_gen_instances.py
577577
"""
578578
}
579579
dir("build"){

pyproject.toml

Lines changed: 17 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ name = "rocm-composable-kernel"
77
dynamic = ["version"]
88
description = "Composable Kernel, performance-critical kernels for machine learning workloads"
99
readme = "README.md"
10-
requires-python = ">=3.8"
10+
requires-python = ">=3.10"
1111
license = {file = "LICENSE"}
1212
classifiers = [
1313
"Programming Language :: Python :: 3",
@@ -21,8 +21,13 @@ dependencies = []
2121
"Bug Tracker" = "https://github.com/rocm/composable_kernel/issues"
2222

2323
[tool.setuptools]
24+
2425
packages = [
2526
"genck",
27+
"genck.ops",
28+
"genck.ops.ck_tile",
29+
"genck.ops.ck_tile.gemm",
30+
"genck.emitters",
2631
"ck4inductor",
2732
"ck4inductor.include",
2833
"ck4inductor.library",
@@ -32,18 +37,26 @@ packages = [
3237
]
3338

3439
[tool.setuptools.package-dir]
40+
41+
genck = "python/codegen"
42+
"genck.ops" = "python/codegen/ops"
43+
"genck.ops.ck_tile" = "python/codegen/ops/ck_tile"
44+
"genck.ops.ck_tile.gemm" = "python/codegen/ops/ck_tile/gemm"
45+
"genck.emitters" = "python/codegen/emitters"
3546
ck4inductor = "python/ck4inductor"
3647
"ck4inductor.universal_gemm" = "python/ck4inductor/universal_gemm"
3748
"ck4inductor.batched_universal_gemm" = "python/ck4inductor/batched_universal_gemm"
3849
"ck4inductor.grouped_conv_fwd" = "python/ck4inductor/grouped_conv_fwd"
3950
"ck4inductor.include" = "include"
4051
"ck4inductor.library" = "library"
4152

42-
"genck" = "python/codegen"
43-
4453
[tool.setuptools.package-data]
4554
"ck4inductor.include" = ["ck/**/*.hpp"]
46-
"ck4inductor.library" = ["src/tensor_operation_instance/gpu/gemm_universal/**/*.hpp", "src/tensor_operation_instance/gpu/gemm_universal_batched/**/*.hpp", "include/ck/library/tensor_operation_instance/gpu/grouped_conv_fwd/**/*.hpp"]
55+
"ck4inductor.library" = [
56+
"src/tensor_operation_instance/gpu/gemm_universal/device_gemm_xdl_universal_*/*.hpp",
57+
"src/tensor_operation_instance/gpu/gemm_universal_batched/**/*.hpp",
58+
"include/ck/library/tensor_operation_instance/gpu/grouped_conv_fwd/**/*.hpp"
59+
]
4760

4861
[tool.setuptools.dynamic]
4962
version = { attr = "setuptools_scm.get_version" }

python/codegen/__init__.py

Whitespace-only changes.

python/codegen/ops/__init__.py

Whitespace-only changes.

python/codegen/ops/ck_tile/__init__.py

Whitespace-only changes.

python/codegen/ops/ck_tile/gemm/__init__.py

Whitespace-only changes.
Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
import unittest
2+
3+
class TestCanRenderInstance(unittest.TestCase):
4+
def test_can_render_instance(self):
5+
from genck.ops.ck_tile.gemm.instance import GEMM
6+
from genck.ops.ck_tile.gemm.render import render
7+
8+
test_instance = GEMM(
9+
layout_a="Row",
10+
layout_b="Col",
11+
layout_c="Row",
12+
datatype_a="BF16",
13+
datatype_b="BF16",
14+
datatype_c="BF16",
15+
tile_m=256,
16+
tile_n=256,
17+
tile_k=64,
18+
warp_m=2,
19+
warp_n=2,
20+
warp_k=1,
21+
warp_tile_m=32,
22+
warp_tile_n=32,
23+
warp_tile_k=16,
24+
m_is_padded="false",
25+
n_is_padded="false",
26+
k_is_padded="false",
27+
pipeline="CompV3",
28+
scheduler="Intrawave",
29+
epilogue="Default",
30+
)
31+
32+
rendered_instance = render(test_instance)
33+
self.assertIn("ck_tile_gemm_universal", rendered_instance)

0 commit comments

Comments
 (0)