Skip to content
This repository was archived by the owner on Oct 11, 2025. It is now read-only.

Commit 7c44357

Browse files
[mlir][python] Add T.tf32 and missing tests for tf32 (#116725)
1 parent 8e26cc8 commit 7c44357

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

mlir/python/mlir/extras/types.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
Float8E4M3Type,
2222
Float8E5M2Type,
2323
Float8E8M0FNUType,
24+
FloatTF32Type,
2425
FunctionType,
2526
IndexType,
2627
IntegerType,
@@ -70,6 +71,7 @@ def ui(width):
7071

7172
f16 = lambda: F16Type.get()
7273
f32 = lambda: F32Type.get()
74+
tf32 = lambda: FloatTF32Type.get()
7375
f64 = lambda: F64Type.get()
7476
bf16 = lambda: BF16Type.get()
7577

0 commit comments

Comments
 (0)