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

[CK TILE GEMM] Add bf8 support to tile engine streamk generator (#3543)

Browse Source
This commit is contained in:
Cong Ma
2026-01-20 10:01:33 -07:00
committed by GitHub
parent 6300ad3c62
commit 4d58c70e6c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tile_engine/ops/gemm_streamk/gemm_streamk_instance_builder.py
View File

@@ -307,6 +307,7 @@ class GemmKernelBuilder:
"fp16": "ck_tile::fp16_t",
"fp8": "ck_tile::fp8_t",
"bf16": "ck_tile::bf16_t",
"bf8": "ck_tile::bf8_t",
"fp32": "float",
"fp64": "double",
}
@@ -776,7 +777,7 @@ def main():
parser.add_argument(
"--datatype",
required=True,
choices=["fp16", "fp8", "bf16", "fp32", "fp64"],
choices=["fp16", "fp8", "bf16", "bf8", "fp32", "fp64"],
help="Data type",
)
parser.add_argument(