Add analysis of stateless memory load/store

Add the analysis to StatelessToStatefull pass to check if there is
non kernel argument memory load/store.

Change-Id: Ieff012618efceeb637446927e9914ecf3ec5f2b6

Author: DianaChen

IGC/Compiler/CISACodeGen/helper.cpp

@@ -1371,9 +1371,9 @@ namespace IGC
             opcode == llvm_gradientYfine);
     }
 
-    bool IsStatelessMemLoadIntrinsic(const llvm::GenIntrinsicInst& inst)
+    bool IsStatelessMemLoadIntrinsic(llvm::GenISAIntrinsic::ID id)
     {
-        switch(inst.getIntrinsicID())
+        switch(id)
         {
         case GenISAIntrinsic::GenISA_simdBlockRead:
                 return true;
@@ -1383,9 +1383,9 @@ namespace IGC
         return false;
     }
 
-    bool IsStatelessMemStoreIntrinsic(const llvm::GenIntrinsicInst& inst)
+    bool IsStatelessMemStoreIntrinsic(llvm::GenISAIntrinsic::ID id)
     {
-        switch (inst.getIntrinsicID()) {
+        switch (id) {
         case GenISAIntrinsic::GenISA_simdBlockWrite:
             return true;
         default:
@@ -1394,7 +1394,7 @@ namespace IGC
         return false;
     }
 
-    bool IsStatelessMemAtomicIntrinsic(const llvm::GenIntrinsicInst& inst)
+    bool IsStatelessMemAtomicIntrinsic(GenIntrinsicInst& inst, GenISAIntrinsic::ID id)
     {
         // This includes:
         // GenISA_intatomicraw

IGC/Compiler/CISACodeGen/helper.h

@@ -175,9 +175,9 @@ namespace IGC
     bool IsSIMDBlockIntrinsic(llvm::Instruction* inst);
     bool isSubGroupIntrinsic(const llvm::Instruction* I);
 
-    bool IsStatelessMemLoadIntrinsic(const llvm::GenIntrinsicInst& inst);
-    bool IsStatelessMemStoreIntrinsic(const llvm::GenIntrinsicInst& inst);
-    bool IsStatelessMemAtomicIntrinsic(const llvm::GenIntrinsicInst& inst);
+    bool IsStatelessMemLoadIntrinsic(llvm::GenISAIntrinsic::ID id);
+    bool IsStatelessMemStoreIntrinsic(llvm::GenISAIntrinsic::ID id);
+    bool IsStatelessMemAtomicIntrinsic(llvm::GenIntrinsicInst& inst, llvm::GenISAIntrinsic::ID id);
 
     bool isURBWriteIntrinsic(const llvm::Instruction* inst);
 

IGC/Compiler/Optimizer/OpenCLPasses/StatelessToStatefull/StatelessToStatefull.cpp

@@ -604,6 +604,22 @@ void StatelessToStatefull::visitCallInst(CallInst& I)
                 m_changed = true;
             }
         }
+        // check if there's non-kernel-arg load/store
+        if (IsStatelessMemStoreIntrinsic(intrinID) ||
+            IsStatelessMemLoadIntrinsic(intrinID)  ||
+            IsStatelessMemAtomicIntrinsic(*Inst, intrinID)) {
+
+            Value* ptr = Inst->getOperand(0);
+            if (!pointerIsFromKernelArgument(*ptr)) {
+                CodeGenContext* ctx = getAnalysis<CodeGenContextWrapper>().getCodeGenContext();
+                if (IsStatelessMemStoreIntrinsic(intrinID))
+                    ctx->m_hasNonKernelArgStore = true;
+                else if (IsStatelessMemLoadIntrinsic(intrinID))
+                    ctx->m_hasNonKernelArgLoad = true;
+                else
+                    ctx->m_hasNonKernelArgAtomic = true;
+            }
+        }
     }
 }
 
@@ -652,6 +668,12 @@ void StatelessToStatefull::visitLoadInst(LoadInst& I)
 
         m_changed = true;
     }
+
+    // check if there's non-kernel-arg load/store
+    if (!pointerIsFromKernelArgument(*ptr)) {
+        CodeGenContext* ctx = getAnalysis<CodeGenContextWrapper>().getCodeGenContext();
+        ctx->m_hasNonKernelArgLoad = true;
+    }
 }
 
 void StatelessToStatefull::visitStoreInst(StoreInst& I)
@@ -693,6 +715,11 @@ void StatelessToStatefull::visitStoreInst(StoreInst& I)
             m_changed = true;
         }
     }
+
+    if (!pointerIsFromKernelArgument(*ptr)) {
+        CodeGenContext* ctx = getAnalysis<CodeGenContextWrapper>().getCodeGenContext();
+        ctx->m_hasNonKernelArgStore = true;
+    }
 }
 
 CallInst* StatelessToStatefull::createBufferPtr(unsigned addrSpace, Constant* argNumber, Instruction* InsertBefore)