ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 07:51:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Undo disabling data types

Browse Source
This commit is contained in:
PoYen, Chen
2024-06-11 14:37:18 +00:00
parent 2532908699
commit 8eb6e451f2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/ck_tile/01_fmha/generate.py
View File

@@ -12,8 +12,8 @@ import fnmatch
DTYPE_MAP = {
"fp16": "ck_tile::fp16_t",
# "bf16": "ck_tile::bf16_t",
# "fp8" : "ck_tile::fp8_t"
"bf16": "ck_tile::bf16_t",
"fp8" : "ck_tile::fp8_t"
}
DTYPE_BITS = {