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

not using custom data type by default, now we can have ISA-level same code as opt_padding

Browse Source
This commit is contained in:
carlushuang
2024-03-17 23:23:32 +00:00
parent ee397d0ab2
commit f55c7629bc
26 changed files with 629 additions and 328 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -11,7 +11,7 @@ import copy
import fnmatch
DTYPE_MAP = {
"fp16": "ck_tile::half_t",
"fp16": "ck_tile::fp16_t",
"bf16": "ck_tile::bf16_t",
"fp8" : "ck_tile::fp8_t"
}