[IGC feature][CodePatch] Debug print for code patching

Author: houjenko

IGC/Compiler/CISACodeGen/CISABuilder.cpp

@@ -376,6 +376,11 @@ namespace IGC
 
     uint32_t CEncoder::getGRFSize() const { return m_program->getGRFSize(); }
 
+
+    std::string CEncoder::GetShaderName() {
+        return IGC::Debug::GetDumpNameObj(m_program, "").str();
+    }
+
     void CEncoder::SetProgram(CShader* program)
     {
         m_program = program;

IGC/Compiler/CISACodeGen/CISABuilder.hpp

@@ -149,6 +149,7 @@ namespace IGC
         void DeclareInput(CVariable* var, uint offset, uint instance);
         void MarkAsOutput(CVariable* var);
         void Compile(bool hasSymbolTable = false);
+        std::string GetShaderName();
         void ReportCompilerStatistics(VISAKernel* pMainKernel, SProgramOutput* pOutput);
         int GetThreadCount(SIMDMode simdMode);
 

IGC/Compiler/CISACodeGen/EmitVISAPass.cpp

@@ -881,7 +881,20 @@ bool EmitPass::runOnFunction(llvm::Function& F)
         m_currShader->AllocatePayload();
 
         if (m_currShader->ProgramOutput()->m_scratchSpaceUsedBySpills)
+        {
+            if (IGC_GET_FLAG_VALUE(CodePatchExperiments))
+            {
+                errs() << "Skip Prologue : " << m_encoder->GetShaderName() << "\n";
+            }
             return false;
+        }
+        if (m_encoder->IsCodePatchCandidate())
+        {
+            if (IGC_GET_FLAG_VALUE(CodePatchExperiments))
+            {
+                errs() << "Prologue/CodePatch : " << m_encoder->GetShaderName() << "\n";
+            }
+        }
     }
 
     if (m_currShader->GetDebugInfoData())

IGC/common/igc_flags.def

@@ -394,6 +394,7 @@ DECLARE_IGC_REGKEY(bool, EnableTrackPtr,                false, "Track Staging Co
 DECLARE_IGC_REGKEY(bool, ExtraRetrySIMD16,              false,  "Enable extra simd16 with retry for STAGE1_BEST_PREF", false)
 DECLARE_IGC_REGKEY(bool, SaveRestoreIR,                 true,  "Save/Restore IR for staged compilation to avoid duplicated compilations", false)
 DECLARE_IGC_REGKEY(DWORD, CodePatch,                    0,     "Enable Pixel Shader code patching to directly emit code after stitching", false)
+DECLARE_IGC_REGKEY(DWORD, CodePatchExperiments,         0,      "Experiment with code patching when != 0", false)
 DECLARE_IGC_REGKEY(DWORD, FirstStagedSIMD,              0,      "Force Pixel shader to be 1: FastSIMD (SIMD8), 2: BestSIMD (SIMD16 or SIMD8), 3: FatestSIMD (SIMD8 opt off)", false)
 DECLARE_IGC_REGKEY(bool, ForceFastestSIMD, false,  "Force pixel shader to return SIMD8 as fast as possible.", false)
 DECLARE_IGC_REGKEY(bool, ForceBestSIMD, false,  "Force pixel shader to return the best SIMD, either SIMD16 or SIMD8.", false)