Detect GEP instructions into vectors

Detect GEP instructions into vectors

Author: mmerecki

IGC/Compiler/CISACodeGen/CheckInstrTypes.cpp

@@ -171,6 +171,7 @@ void CheckInstrTypes::print(llvm::raw_ostream& OS) const
     OS << "\nhasDynamicGenericLoadStore: " << g_InstrTypes.hasDynamicGenericLoadStore;
     OS << "\nhasUnmaskedRegion: " << g_InstrTypes.hasUnmaskedRegion;
     OS << "\nhasSLM: " << g_InstrTypes.hasSLM;
+    OS << "\nhasPrivateGEPIntoVector: " << g_InstrTypes.hasPrivateGEPIntoVector;
     OS << "\nnumCall: " << g_InstrTypes.numCall;
     OS << "\nnumBarrier: " << g_InstrTypes.numBarrier;
     OS << "\nnumLoadStore: " << g_InstrTypes.numLoadStore;
@@ -754,6 +755,20 @@ void CheckInstrTypes::visitGetElementPtrInst(llvm::GetElementPtrInst& I)
     {
         g_InstrTypes.hasGenericAddressSpacePointers = true;
     }
+    else if (I.getPointerAddressSpace() == ADDRESS_SPACE::ADDRESS_SPACE_PRIVATE &&
+        g_InstrTypes.hasPrivateGEPIntoVector == false &&
+        I.getNumIndices() > 1)
+    {
+        llvm::SmallVector<llvm::Value*, 4> indices(I.indices());
+        indices.pop_back();
+        llvm::Type* lastIndexedType = llvm::GetElementPtrInst::getIndexedType(
+            I.getSourceElementType(),
+            indices);
+        if (lastIndexedType->isVectorTy())
+        {
+            g_InstrTypes.hasPrivateGEPIntoVector = true;
+        }
+    }
 }
 
 #undef PASS_FLAG

IGC/Compiler/CodeGenPublic.h

@@ -348,6 +348,7 @@ namespace IGC
         bool hasDynamicGenericLoadStore{};
         bool hasUnmaskedRegion{};
         bool hasSLM{};
+        bool hasPrivateGEPIntoVector{};
         unsigned int numCall{};
         unsigned int numBarrier{};
         unsigned int numLoadStore{};