@@ -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
598598class 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