Skip to content

Commit 71f4284

Browse files
committed
Add abbreviation for bfloat16 data type in mfma_macro_generator.py
- Introduced a new abbreviation "bf16" for the bfloat16 data type in the mfma_macro_generator.py file, enhancing clarity and consistency in data type representation.
1 parent 15035cd commit 71f4284

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tilelang/intrinsics/mfma_macro_generator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,6 +139,7 @@ def _initialize_mfma_prefix(self, k_dim=16):
139139
}[out_dtype]
140140

141141
in_dtype_abbrv = {
142+
"bfloat16": "bf16",
142143
"float16": "f16",
143144
"float32": "f32",
144145
"int8": "i8",

0 commit comments

Comments
 (0)