Expose method to check autoGRF in CodeGenContext

Create method to check autoGRF to CodeGenContext

Author: admitric

IGC/Compiler/CISACodeGen/CISABuilder.cpp

@@ -4644,34 +4644,20 @@ namespace IGC
                     SaveOption(vISA_TotalGRFNum, unsigned(0));
                     SaveOption(vISA_AutoGRFSelection, true);
                 }
+                else if (ClContext->getExpGRFSize() > 0) {
+                    // Explicit GRF size set per module (by compiler option)
+                    SaveOption(vISA_TotalGRFNum, ClContext->getExpGRFSize());
+                }
                 else if (ClContext->getNumThreadsPerEU() > 0)
                 {
                     // Number of threads per EU is set per module (by compiler option)
                     SaveOption(vISA_HWThreadNumberPerEU, unsigned(ClContext->getNumThreadsPerEU()));
                 }
-                else if (ClContext->getNumThreadsPerEU() == 0)
-                {
+                else if (ClContext->isAutoGRFSelectionEnabled()) {
                     // "Auto" mode per module (by compiler option) - use compiler heuristics to determine number of threads per EU
                     SaveOption(vISA_TotalGRFNum, unsigned(0));
                     SaveOption(vISA_AutoGRFSelection, true);
                 }
-                else if (ClContext->getExpGRFSize() > 0) {
-                    // Explicit GRF size set per module (by compiler option)
-                    SaveOption(vISA_TotalGRFNum, ClContext->getExpGRFSize());
-                }
-                else if ((m_program->m_Platform->supportsAutoGRFSelection() &&
-                    (context->m_DriverInfo.supportsAutoGRFSelection() ||
-                      ClContext->m_InternalOptions.IntelEnableAutoLargeGRF ||
-                      ClContext->m_Options.IntelEnableAutoLargeGRF)
-                    ) && !ClContext->m_InternalOptions.Intel128GRFPerThread &&
-                    !ClContext->m_Options.Intel128GRFPerThread &&
-                    !ClContext->m_InternalOptions.Intel256GRFPerThread &&
-                    !ClContext->m_Options.Intel256GRFPerThread
-                    )
-                {
-                    // When user hasn't specified number of threads, we can rely on compiler heuristics
-                    SaveOption(vISA_AutoGRFSelection, true);
-                }
 
                 // Emit warnings if mismatch is found in user input
                 // Mismatch between number of threads and GRF size (per module)

IGC/Compiler/CISACodeGen/OpenCLKernelCodeGen.cpp

@@ -113,6 +113,27 @@ namespace IGC
         return CodeGenContext::getNumGRFPerThread(returnDefault);
     }
 
+    bool OpenCLProgramContext::isAutoGRFSelectionEnabled() const
+    {
+        if (getNumThreadsPerEU() == 0)
+            return true;
+
+        if ((platform.supportsAutoGRFSelection() &&
+            (m_DriverInfo.supportsAutoGRFSelection() ||
+                m_InternalOptions.IntelEnableAutoLargeGRF ||
+                m_Options.IntelEnableAutoLargeGRF)
+            ) && !m_InternalOptions.Intel128GRFPerThread &&
+            !m_Options.Intel128GRFPerThread &&
+            !m_InternalOptions.Intel256GRFPerThread &&
+            !m_Options.Intel256GRFPerThread
+            )
+        {
+            return true;
+        }
+
+        return false;
+    }
+
     bool OpenCLProgramContext::forceGlobalMemoryAllocation() const
     {
         return m_InternalOptions.ForceGlobalMemoryAllocation ||

IGC/Compiler/CISACodeGen/OpenCLKernelCodeGen.hpp

@@ -131,6 +131,7 @@ namespace IGC
         bool needsDivergentBarrierHandling() const;
         unsigned GetSlmSizePerSubslice();
         float GetSpillThreshold(SIMDMode dispatchSize);
+        bool isAutoGRFSelectionEnabled() const override;
 
         void clearBeforeRetry() {
             m_programOutput.clearBeforeRetry();

IGC/Compiler/CodeGenPublic.h

@@ -1095,6 +1095,7 @@ namespace IGC
         virtual uint32_t getExpGRFSize() const;
         virtual uint32_t getNumGRFPerThread(bool returnDefault = true);
         virtual void setNumGRFPerThread(uint32_t value) { m_NumGRFPerThread = value; }
+        virtual bool isAutoGRFSelectionEnabled() const { return false; };
         virtual bool forceGlobalMemoryAllocation() const;
         virtual bool allocatePrivateAsGlobalBuffer() const;
         virtual bool noLocalToGenericOptionEnabled() const;