Tooling changes in vISA Backend

Tooling changes in vISA Backend to streamline debugging and testing.

Author: jmechalas

IGC/Compiler/CISACodeGen/CISABuilder.cpp

@@ -4618,7 +4618,7 @@ void CEncoder::InitEncoder(bool canAbortOnSpill, bool hasStackCall, bool hasInli
 
   std::string kernelName = std::string(m_program->entry->getName());
   std::string asmName;
-  if (m_enableVISAdump || context->m_instrTypes.hasDebugInfo) {
+  if (m_enableVISAdump || vbuilder->GetOption(vISA_ShaderStatsDumpless) || context->m_instrTypes.hasDebugInfo) {
     asmName = GetDumpFileName("asm");
   } else {
     asmName = "kernel.asm";

visa/BinaryEncodingIGA.cpp

@@ -1619,16 +1619,11 @@ SendDesc BinaryEncodingIGA::encodeExDescImm(G4_INST *sendInst,
   const G4_Operand *exDescG4 = sendInst->getSrc(3);
   const G4_SendDescRaw *descG4 = (G4_SendDescRaw *)sendInst->getMsgDesc();
   vISA_ASSERT(descG4 != nullptr, "expected raw descriptor");
-  if (sendInst->getBuilder().getOption(vISA_ShaderDataBaseStats)) {
+  if (sendInst->getBuilder().getOption(vISA_ShaderStatsDumpless)) {
     auto JitInfo = kernel.fg.builder->getJitInfo();
     JitInfo->sendInfo.src0Vec.push_back(descG4->getSrc0LenRegs());
     JitInfo->sendInfo.src1Vec.push_back(descG4->getSrc1LenRegs());
     JitInfo->sendInfo.destVec.push_back(descG4->getDstLenRegs());
-    printSendDataToFile(descG4->getSrc0LenRegs(),
-                        descG4->getSrc1LenRegs(),
-                        descG4->getDstLenRegs(),
-                        sendInst->getBuilder().getOptions()->getOptionCstr(
-                            vISA_ShaderDataBaseStatsFilePath));
   }
 
   sdos.xlen = (int)descG4->extMessageLength();
@@ -1684,16 +1679,11 @@ SendDesc BinaryEncodingIGA::encodeExDescRegA0(G4_INST *sendInst,
   G4_Operand *exDescG4 = sendInst->getSrc(3);
   const G4_SendDescRaw *descG4 = sendInst->getMsgDescRaw();
   vISA_ASSERT(descG4 != nullptr, "expected raw descriptor");
-  if (sendInst->getBuilder().getOption(vISA_ShaderDataBaseStats)) {
+  if (sendInst->getBuilder().getOption(vISA_ShaderStatsDumpless)) {
     auto JitInfo = kernel.fg.builder->getJitInfo();
     JitInfo->sendInfo.src0Vec.push_back(descG4->getSrc0LenRegs());
     JitInfo->sendInfo.src1Vec.push_back(descG4->getSrc1LenRegs());
     JitInfo->sendInfo.destVec.push_back(descG4->getDstLenRegs());
-    printSendDataToFile(descG4->getSrc0LenRegs(),
-                        descG4->getSrc1LenRegs(),
-                        descG4->getDstLenRegs(),
-                        sendInst->getBuilder().getOptions()->getOptionCstr(
-                            vISA_ShaderDataBaseStatsFilePath));
   }
   SendDesc exDescIga;
   exDescIga.type = SendDesc::Kind::REG32A;

visa/BuildCISAIR.h

@@ -78,6 +78,9 @@ class CISA_IR_Builder : public VISABuilder {
   VISA_BUILDER_API int GetuInt32Option(vISAOptions option) override {
       return m_options.getuInt32Option(option);
   }
+  VISA_BUILDER_API bool GetOption(vISAOptions option) override {
+      return m_options.getOption(option);
+  }
   VISA_BUILDER_API void SetOption(vISAOptions option, bool val) override {
     m_options.setOption(option, val);
   }

visa/VISAKernelImpl.cpp

@@ -34,6 +34,9 @@ SPDX-License-Identifier: MIT
 #include "KernelCost.hpp"
 #include "include/RelocationInfo.h"
 #include "visa_igc_common_header.h"
+#ifdef _WIN32
+#include "inc/common/DriverStore.h"
+#endif
 
 #include <cctype>
 #include "common/LLVMWarningsPush.hpp"
@@ -468,6 +471,24 @@ void *VISAKernelImpl::encodeAndEmit(unsigned int &binarySize) {
     dumpPerfStatsInJson(m_asmName);
   }
 
+#ifdef _WIN32
+  if (m_options->getOption(vISA_ShaderStatsDumpless)) {
+      typedef void(__stdcall* PFNOPENWRAPPERDLL)(
+          void* buf, size_t buf_size, char* dump_stem, char* platform,
+          void* sendinfo, size_t sendinfo_size);
+      HMODULE handle = LoadDependency("WrapperLib.dll");
+      if (handle) {
+          auto OpenWrapper =
+              (PFNOPENWRAPPERDLL)GetProcAddress(handle, "process_buffer");
+          if (OpenWrapper) {
+              OpenWrapper(binary, binarySize, (char*)m_asmName.c_str(),
+                  (char*)m_kernel->getGenxPlatformString(),
+                  &m_jitInfo->sendInfo, sizeof(m_jitInfo->sendInfo));
+          }
+      }
+  }
+#endif
+
   return binary;
 }
 

visa/include/VISABuilderAPIDefinition.h

@@ -1315,6 +1315,7 @@ class VISABuilder {
                                        bool emit_visa_only = false) = 0;
 
   VISA_BUILDER_API virtual int GetuInt32Option(vISAOptions option) = 0;
+  VISA_BUILDER_API virtual bool GetOption(vISAOptions option) = 0;
   VISA_BUILDER_API virtual void SetOption(vISAOptions option, bool val) = 0;
   VISA_BUILDER_API virtual void SetOption(vISAOptions option, uint32_t val) = 0;
   VISA_BUILDER_API virtual void SetOption(vISAOptions option,

visa/include/VISAOptionsDefs.h

@@ -43,6 +43,8 @@ DEF_VISA_OPTION(vISA_DumpPerfStatsVerbose, ET_BOOL, "-dumpVISAJsonStatsVerbose",
                 "dump the verbose stats to default json file name", false)
 DEF_VISA_OPTION(vISA_DumpSendInfoStats, ET_BOOL, "-dumpVISASendInfoStats",
                 "dumps the sendinfo stats with the stats.json file", false)
+DEF_VISA_OPTION(vISA_ShaderStatsDumpless, ET_BOOL, "-noDumpShaderStats",
+        "executes pathway for no dumping and running shader stats", false)
 DEF_VISA_OPTION(VISA_FullIRVerify, ET_BOOL, "-fullIRVerify", UNUSED, false)
 // dump each option while it is being set by setOption()
 DEF_VISA_OPTION(vISA_dumpVISAOptions, ET_BOOL, "-dumpVisaOptions", UNUSED,