Skip to content

Commit 432969e

Browse files
committed
[NVPTX] A small fix for MMA and sparse MMA tests. PR150950.
1 parent 3c4d5d5 commit 432969e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

llvm/test/CodeGen/NVPTX/wmma.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -96,9 +96,9 @@ def __init__(self, geom, frag, ptx_elt_type, is_mma_sparse=False):
9696
"m16n8k32:b:bf16": 4,
9797
"m16n8k32:b:f16": 4,
9898
"m16n8k32:c:f16": 2,
99-
"m16n8k32:c:f32": 4 if is_mma_sparse else 8,
99+
"m16n8k32:c:f32": 4,
100100
"m16n8k32:d:f16": 2,
101-
"m16n8k32:d:f32": 4 if is_mma_sparse else 8,
101+
"m16n8k32:d:f32": 4,
102102
"m16n8k16:a:tf32": 4,
103103
"m16n8k16:b:tf32": 4,
104104
"m16n8k16:c:tf32": 4,

0 commit comments

Comments
 (0)