Add compiler option to force private memory allocation in global memory

Change-Id: I55f67ccbafb57faedc2f4f0e41e913828ac72333

Author: jfuentes

IGC/Compiler/CodeGenContext.cpp

@@ -508,6 +508,12 @@ namespace IGC
         return CodeGenContext::getNumGRFPerThread();
     }
 
+    bool OpenCLProgramContext::forceGlobalMemoryAllocation() const
+    {
+        return m_InternalOptions.IntelForceGlobalMemoryAllocation;
+    }
+
+
     void CodeGenContext::initLLVMContextWrapper(bool createResourceDimTypes)
     {
         llvmCtxWrapper = new LLVMContextWrapper(createResourceDimTypes);
@@ -669,6 +675,11 @@ namespace IGC
         return 128;
     }
 
+    bool CodeGenContext::forceGlobalMemoryAllocation() const
+    {
+        return false;
+    }
+
     bool CodeGenContext::isPOSH() const
     {
         return this->getModule()->getModuleFlag(

IGC/Compiler/CodeGenPublic.h

@@ -845,6 +845,7 @@ namespace IGC
         virtual void resetOnRetry();
         virtual uint32_t getNumThreadsPerEU() const;
         virtual uint32_t getNumGRFPerThread() const;
+        virtual bool forceGlobalMemoryAllocation() const;
         bool isPOSH() const;
 
         CompilerStats& Stats()
@@ -1073,6 +1074,10 @@ namespace IGC
                 {
                     EnableTakeGlobalAddress = true;
                 }
+                if (strstr(options, "-cl-intel-force-global-mem-allocation"))
+                {
+                    IntelForceGlobalMemoryAllocation = true;
+                }
             }
 
 
@@ -1089,6 +1094,7 @@ namespace IGC
             bool PromoteStatelessToBindless = false;
             bool PreferBindlessImages = false;
             bool EnableTakeGlobalAddress = false;
+            bool IntelForceGlobalMemoryAllocation = false;
 
         };
 
@@ -1162,6 +1168,7 @@ namespace IGC
         float getProfilingTimerResolution();
         uint32_t getNumGRFPerThread() const;
         uint32_t getNumThreadsPerEU() const;
+        bool forceGlobalMemoryAllocation() const;
     private:
         llvm::DenseMap<llvm::Function*, std::string> m_hashes_per_kernel;
     };

IGC/Compiler/Optimizer/OpenCLPasses/PrivateMemory/PrivateMemoryResolution.cpp

@@ -361,7 +361,8 @@ bool PrivateMemoryResolution::safeToUseScratchSpace(llvm::Module& M) const
     bool supportsStatelessSpacePrivateMemory = Ctx.m_DriverInfo.supportsStatelessSpacePrivateMemory();
     bool bOCLLegacyStatelessCheck = true;
 
-    if ((modMD.compOpt.OptDisable && bOCLLegacyStatelessCheck) || !supportsScratchSpacePrivateMemory) {
+    if ((modMD.compOpt.OptDisable && bOCLLegacyStatelessCheck) || !supportsScratchSpacePrivateMemory
+        || Ctx.forceGlobalMemoryAllocation()) {
         return false;
     }