Early Register Estimator - filter dump

Adds new flag to dump only selected functions in Early Register Estimator.

Author: pkwasnie-intel

IGC/Compiler/CISACodeGen/IGCLivenessAnalysis.cpp

@@ -20,6 +20,7 @@ SPDX-License-Identifier: MIT
 
 #include <fstream>
 #include <queue>
+#include <regex>
 
 char IGCLivenessAnalysis::ID = 0;
 // Register pass to igc-opt
@@ -381,6 +382,10 @@ void IGCLivenessAnalysis::intraBlock(llvm::BasicBlock &BB, std::string &Output,
 
 void IGCLivenessAnalysis::dumpRegPressure(llvm::Function &F,
                                           unsigned int SIMD) {
+    const char *Filter = IGC_GET_REGKEYSTRING(DumpRegPressureEstimateFilter);
+    if (Filter && *Filter != '\0' && !std::regex_search(F.getName().str(), std::regex(Filter)))
+        return;
+
     // force print all
     PrinterType = 5;
 

IGC/common/igc_flags.h

@@ -742,6 +742,7 @@ DECLARE_IGC_REGKEY(bool, RematSameBBScope, true, "Confine rematerialization only
 DECLARE_IGC_REGKEY(bool, EnableRemat, false, "Enable clone adress arithmetic pass not only on retry", false)
 DECLARE_IGC_REGKEY(bool, RematReassocBefore, false, "Enable short sequence of passes before clone address arithmetic pass to potentially decrese amount of operations that will be rematerialized", false)
 DECLARE_IGC_REGKEY(bool, DumpRegPressureEstimate, false,  "Dump RegPressureEstimate to a file", false)
+DECLARE_IGC_REGKEY(debugString, DumpRegPressureEstimateFilter, 0, "Only dump RegPressureEstimate for functions matching the given regex", false)
 DECLARE_IGC_REGKEY(bool, EnableReusingXYZWStoreConstPayload, true, "Enable reusing XYZW stores const payload", false)
 DECLARE_IGC_REGKEY(bool, EnableReusingLSCStoreConstPayload,  false, "Enable reusing LSC stores const payload", false)
 DECLARE_IGC_REGKEY(DWORD, RegPressureVerbocity,   0,  "Different printing types", false)