ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit '4d58c70e6cf76ce6cb40aa6035ebccbb28493f71' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2026-01-20 17:18:34 +00:00
parent a7320b9717
commit 6a0cbcb01d
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(