[llvm] [LLVM][NVPTX] Enable family specific support for a few intrinsics (PR #173268)

Rajat Bajpai via llvm-commits llvm-commits at lists.llvm.org
Mon Dec 22 22:19:56 PST 2025


================
@@ -4857,12 +4857,16 @@ class WMMA_REGINFO<WMMA_REGS r, string op, string metadata = "", string kind = "
   // all fragments of the instruction are viable.
   list<Predicate> Predicates = !cond(
     !or(!eq(op, "mma.block_scale"),
-        !eq(op, "mma.sp.block_scale")) : [hasSM120a, hasPTX<88>],
+        !eq(op, "mma.sp.block_scale")) : [hasSM<120>, hasFamilySpecificFeatures],
----------------
rajatbajpai wrote:

While this is technically correct, it is not the most maintainable solution. Using `hasPTXWithFamilySMs` would make it easier to handle any future exceptions in a single place, such as the sm_101 PTX 9.0+ case.

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


More information about the llvm-commits mailing list