Skip to content

Commit b556a6e

Browse files
committed
fixup
1 parent c23800d commit b556a6e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

llvm/include/llvm/IR/IntrinsicsNVVM.td

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -366,7 +366,7 @@ class WMMA_NAME<string ALayout, string BLayout, int Satfinite, string Rnd, strin
366366
string record = "int_nvvm_wmma_"
367367
# A.geom
368368
# "_mma"
369-
# b1op
369+
# !subst(".", "_", b1op)
370370
# "_" # ALayout
371371
# "_" # BLayout
372372
# !if(!ne(Rnd, ""), !strconcat("_", Rnd), "")
@@ -378,7 +378,7 @@ class MMA_NAME<string ALayout, string BLayout, int Satfinite, string b1op,
378378
WMMA_REGS A, WMMA_REGS B, WMMA_REGS C, WMMA_REGS D> {
379379
string signature = MMA_SIGNATURE<A, B, C, D>.ret;
380380
string record = "int_nvvm_mma"
381-
# b1op
381+
# !subst(".", "_", b1op)
382382
# "_" # A.geom
383383
# "_" # ALayout
384384
# "_" # BLayout
@@ -597,7 +597,7 @@ class NVVM_WMMA_SUPPORTED<list<WMMA_REGS> frags, string layout_a, string layout_
597597

598598
class NVVM_MMA_B1OPS<list<WMMA_REGS> frags> {
599599
list<string> ret = !cond(
600-
!eq(frags[0].ptx_elt_type, "b1") : ["_xor_popc", "_and_popc"],
600+
!eq(frags[0].ptx_elt_type, "b1") : [".xor.popc", ".and.popc"],
601601
true: [""]
602602
);
603603
}

0 commit comments

Comments
 (0)