[Mlir-commits] [llvm] [mlir] [LLVM][NVPTX] Add Rubin extensions for G2S Tensor intrinsics (PR #220029)

[Rajat Bajpai]

================
@@ -1168,48 +1168,85 @@ static bool upgradeArmOrAarch64IntrinsicFunction(bool IsArm, Function *F,
   return false; // No other 'arm.*', 'aarch64.*'.
 }
 
-static Intrinsic::ID shouldUpgradeNVPTXTMAG2SIntrinsics(Function *F,
-                                                        StringRef Name) {
+static Intrinsic::ID
+shouldUpgradeNVPTXTMAG2SIntrinsics(Function *F, StringRef Name,
+                                   SmallVectorImpl<Type *> &OvlTys) {
   if (Name.consume_front("cp.async.bulk.tensor.g2s.")) {
+    // G2S CTA mode check is needed to differentiate the g2s and g2s_cta
+    // intrinsics.
+    bool IsCTA = Name.consume_front("cta.");
----------------
rajatbajpai wrote:

Sure, separated g2s and g2s + cta variants.

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