Add upperBoundGRF

Pass the value to vISA_MaxGRFNum
intel · Feb 26, 2025 · 2768d02 · 2768d02
1 parent c7a0598
commit 2768d02
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 3 deletions.
diff --git a/IGC/Compiler/CISACodeGen/CISABuilder.cpp b/IGC/Compiler/CISACodeGen/CISABuilder.cpp
@@ -5556,7 +5556,8 @@ namespace IGC
         }
     }
 
-    void CEncoder::InitEncoder(bool canAbortOnSpill, bool hasStackCall, bool hasInlineAsmCall, bool hasAdditionalVisaAsmToLink, int numThreadsPerEU, uint lowerBoundGRF, VISAKernel* prevKernel)
+    void CEncoder::InitEncoder(bool canAbortOnSpill, bool hasStackCall, bool hasInlineAsmCall, bool hasAdditionalVisaAsmToLink, int numThreadsPerEU,
+        uint lowerBoundGRF, uint upperBoundGRF, VISAKernel* prevKernel)
     {
         m_aliasesMap.clear();
         m_encoderState.m_SubSpanDestination = false;
@@ -5626,6 +5627,11 @@ namespace IGC
             SaveOption(vISA_MinGRFNum, lowerBoundGRF);
         }
 
+        if (upperBoundGRF > 0)
+        {
+            SaveOption(vISA_MaxGRFNum, upperBoundGRF);
+        }
+
         // Pass all build options to builder
         SetBuilderOptions(vbuilder);
 

diff --git a/IGC/Compiler/CISACodeGen/CISABuilder.hpp b/IGC/Compiler/CISACodeGen/CISABuilder.hpp
@@ -115,7 +115,8 @@ namespace IGC
     class CEncoder
     {
     public:
-        void InitEncoder(bool canAbortOnSpill, bool hasStackCall, bool hasInlineAsmCall, bool hasAdditionalVisaAsmToLink, int numThreadsPerEU, uint lowerBoundGRF, VISAKernel* prevKernel);
+        void InitEncoder(bool canAbortOnSpill, bool hasStackCall, bool hasInlineAsmCall, bool hasAdditionalVisaAsmToLink, int numThreadsPerEU,
+            uint lowerBoundGRF, uint upperBoundGRF, VISAKernel* prevKernel);
         void InitBuildParams(llvm::SmallVector<std::unique_ptr< char, std::function<void(char*)>>, 10> & params);
         void InitVISABuilderOptions(TARGET_PLATFORM VISAPlatform, bool canAbortOnSpill, bool hasStackCall, bool enableVISA_IR);
         SEncoderState CopyEncoderState();

diff --git a/IGC/Compiler/CISACodeGen/EmitVISAPass.cpp b/IGC/Compiler/CISACodeGen/EmitVISAPass.cpp
@@ -913,6 +913,7 @@ bool EmitPass::runOnFunction(llvm::Function& F)
         }
 
         uint lowerBoundGRF = 0;
+        uint upperBoundGRF = 0;
         int numThreadsPerEU = -1;
         if (F.hasFnAttribute("num-thread-per-eu"))
         {
@@ -921,7 +922,7 @@ bool EmitPass::runOnFunction(llvm::Function& F)
         }
         // call builder after pre-analysis pass where scratchspace offset to VISA is calculated
         m_encoder->InitEncoder(m_canAbortOnSpill, m_currShader->HasStackCalls(), hasInlineAsmCall, hasAdditionalVisaAsmToLink,
-            numThreadsPerEU, lowerBoundGRF, prevKernel);
+            numThreadsPerEU, lowerBoundGRF, upperBoundGRF, prevKernel);
 
         if (!m_encoder->IsCodePatchCandidate())
             createVMaskPred(m_vMaskPredForSubplane);