[llvm] [LLVM][NVPTX] Add support for ldmatrix extensions introduced in PTX 8.6 (PR #124899)

Pradeep Kumar via llvm-commits llvm-commits at lists.llvm.org
Tue Mar 11 01:28:53 PDT 2025


================
@@ -7194,7 +7197,27 @@ class WMMA_REGINFO<WMMA_REGS r, string op>
 
     !and(!eq(op,"ldmatrix"),
          !eq(ptx_elt_type,"b16"),
-         !eq(geom, "m8n8")) : [hasSM<75>, hasPTX<65>]);
+         !eq(geom, "m8n8")) : [hasSM<75>, hasPTX<65>],
+
+    !and(!eq(op,"ldmatrix"),
+         !eq(ptx_elt_type,"b8"),
+         !eq(geom, "m16n16")) : [hasSM<100>, hasAAFeatures, hasPTX<86>],
----------------
schwarzschild-radius wrote:

Thanks to @durga4github's PR (https://github.com/llvm/llvm-project/pull/127990), I have updated my change to use `hasArchAccel` predicated instead of `hasAAFeatures`

https://github.com/llvm/llvm-project/pull/124899


More information about the llvm-commits mailing list