[Mlir-commits] [mlir] [MLIR][NVVM] Add support for dp4a instructions (PR #139043)
Durgadoss R
llvmlistbot at llvm.org
Thu May 8 22:36:45 PDT 2025
================
@@ -1590,6 +1590,14 @@ static void nvvmInferResultRanges(Operation *op, Value result,
}
}
+#define GET_DP4A_ID(a_sign, is_b_signed) \
+ is_b_signed ? llvm::Intrinsic::nvvm_idp4a_##a_sign##_s \
+ : llvm::Intrinsic::nvvm_idp4a_##a_sign##_u
+
+llvm::Intrinsic::ID Dp4aOp::getIntrinsicID(bool a_signed, bool b_signed) {
+ return a_signed ? GET_DP4A_ID(s, b_signed) : GET_DP4A_ID(u, b_signed);
----------------
durga4github wrote:
Can we use GET_ID(is_a_signed, is_b_signed) ?
Since this is only 4 individual intrinsics, a simple lambda (without any macro) will be cleaner
https://github.com/llvm/llvm-project/pull/139043
More information about the Mlir-commits
mailing list