[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