Adds vISA option to enable JSON shader dump.

Adds vISA option to enable JSON shader dump.

Author: daniel-man

visa/BinaryEncodingIGA.cpp

@@ -9,14 +9,15 @@ SPDX-License-Identifier: MIT
 #include "BinaryEncodingIGA.h"
 #include "GTGPU_RT_ASM_Interface.h"
 #include "iga/IGALibrary/api/igaEncoderWrapper.hpp"
+#include "iga/IGALibrary/Frontend/FormatterJSON.hpp"
 #include "Timer.h"
 #include "BuildIR.h"
 #include "Common_ISA_framework.h"
 
+#include <fstream>
 #include <map>
 #include <utility>
 
-
 using namespace iga;
 using namespace vISA;
 
@@ -45,6 +46,8 @@ class BinaryEncodingIGA
     // translates and encodes (formerly "DoAll")
     void Encode();
 
+    void EmitJSON(int dumpJSON);
+
     ///////////////////////////////////////////////////////////////////////////
     // these function translate G4 IR to IGA IR
     Instruction *translateInstruction(G4_INST *g4inst, Block*& bbNew);
@@ -1034,6 +1037,19 @@ void BinaryEncodingIGA::Encode()
         kernel.fg.builder->getJitInfo()->offsetToSkipSetFFIDGP1 =
             kernel.getComputeFFIDGP1NextOff();
     }
+
+    int dumpJSON = kernel.fg.builder->getuint32Option(vISA_dumpIgaJson);
+    if (dumpJSON) {
+        EmitJSON(dumpJSON);
+    }
+}
+
+void BinaryEncodingIGA::EmitJSON(int dumpJSON) {
+    std::string jsonFileName = fileName + ".json";
+    std::ofstream ofs(jsonFileName, std::ofstream::out);
+    FormatOpts fos(*platformModel);
+    fos.printJson = true;
+    FormatJSON(ofs, fos, *IGAKernel, nullptr);
 }
 
 Instruction *BinaryEncodingIGA::translateInstruction(
@@ -1991,4 +2007,3 @@ bool vISA::InstSupportsSrcModifierIGA(TARGET_PLATFORM p, const G4_INST &i, const
         return false;
     }
 }
-

visa/include/VISAOptionsDefs.h

@@ -296,6 +296,10 @@ DEF_VISA_OPTION(vISA_encoderFile,       ET_CSTR, "-encoderStatisticsFile", "USAG
 DEF_VISA_OPTION(vISA_CISAbinary,        ET_CSTR, "-CISAbinary",            "USAGE: File Name with isaasm paths. ",  NULL)
 DEF_VISA_OPTION(vISA_DumpRegInfo, ET_BOOL, "-dumpRegInfo",            UNUSED, false)
 DEF_VISA_OPTION(vISA_PrintHexFloatInAsm, ET_BOOL, "-printHexFloatInAsm", UNUSED, false)
+DEF_VISA_OPTION(vISA_dumpIgaJson, ET_INT32, "-dumpIgaJson",
+  "Emits a .json file (peer to .asm) with final IGA output in JSON format; "
+  "1 enables basic output, 2 enables with def/use dataflow information "
+  "(may increase compile time significantly for large shaders)", 0)
 
 //=== misc options ===
 DEF_VISA_OPTION(vISA_PlatformSet,         ET_INT32, NULLSTR,              UNUSED, -1 /*GENX_NONE*/)