add isel

Author: abhilash1910

llvm/lib/Target/NVPTX/NVPTXISelLowering.cpp

@@ -3973,6 +3973,17 @@ bool NVPTXTargetLowering::getTgtMemIntrinsic(
     return true;
   }
 
+  case Intrinsic::nvvm_prefetch_tensormap: {
+    auto &DL = I.getDataLayout();  
+    Info.opc = ISD::INTRINSIC_VOID;
+    Info.memVT = getPointerTy(DL);  
+    Info.ptrVal = I.getArgOperand(0);
+    Info.offset = 0;
+    Info.flags = MachineMemOperand::MOLoad | MachineMemOperand::MODereferenceable;
+    Info.align.reset();
+    return true;
+  }
+  
   case Intrinsic::nvvm_ldu_global_i:
   case Intrinsic::nvvm_ldu_global_f:
   case Intrinsic::nvvm_ldu_global_p: {

llvm/lib/Target/NVPTX/NVPTXIntrinsics.td

@@ -753,30 +753,11 @@ foreach dim = [1, 2, 3, 4, 5] in {
 //Prefetchu and Prefetch
 
 class PREFETCH_CONST_CHK<dag frag>
-          : PatFrag<!setdagop(frag, ops), frag, [{
-            auto *Addr = N->getOperand(2).getNode();
-            auto *MemNode = dyn_cast<MemSDNode>(Addr);
-            bool result = MemNode->getMemOperand()->getAddrSpace() == llvm::ADDRESS_SPACE_CONST;
-            return result;
-          }]>;
-
-
+          : PatFrag<!setdagop(frag, ops), frag, AS_match.const>;
 class PREFETCH_GENERIC_CHK<dag frag>
-          : PatFrag<!setdagop(frag, ops), frag,  [{
-            auto *Addr = N->getOperand(2).getNode();
-            auto *MemNode = dyn_cast<MemSDNode>(Addr);
-            bool result= MemNode->getMemOperand()->getAddrSpace() == llvm::ADDRESS_SPACE_GENERIC;
-            return result;
-          }]>;
-
-
+          : PatFrag<!setdagop(frag, ops), frag, AS_match.generic>;
 class PREFETCH_PARAM_CHK<dag frag>
-          : PatFrag<!setdagop(frag, ops), frag, [{
-            auto *Addr = N->getOperand(2).getNode();
-            auto *MemNode = dyn_cast<MemSDNode>(Addr);
-            bool result = MemNode->getMemOperand()->getAddrSpace() == llvm::ADDRESS_SPACE_PARAM;
-            return result;
-          }]>;
+          : PatFrag<!setdagop(frag, ops), frag, AS_match.param>;
 
 defvar frag_pat = (int_nvvm_prefetch_tensormap node:$addr);