Fix assert when not compiling compute input

Fix assert when not compiling compute input

Author: pratikashar

IGC/Compiler/DebugInfo/ScalarVISAModule.h

@@ -98,6 +98,12 @@ class ScalarVisaModule final : public IGC::VISAModule {
 
   bool usesSlot1ScratchSpill() const override;
 
+  virtual std::optional<enum ShaderType> getShaderType() const override {
+    if (!m_pShader)
+      return std::nullopt;
+    return m_pShader->GetShaderType();
+  }
+
 private:
   /// @brief Constructor.
   /// @param m_pShader holds the Shader object that provides information

IGC/DebugInfo/VISADebugEmitter.cpp

@@ -442,7 +442,9 @@ void DebugEmitter::prepareElfForZeBinary(bool is64Bit, char *pElfBuffer, size_t
       }
     }
   } else {
-    IGC_ASSERT_MESSAGE(is64Bit, "64-bit ELF file only supported");
+    auto ST = m_pVISAModule->getShaderType();
+    if (ST && (ST == ShaderType::COMPUTE_SHADER || ST == ShaderType::OPENCL_SHADER))
+      IGC_ASSERT_MESSAGE(is64Bit, "64-bit ELF file only supported");
   }
 }
 

IGC/DebugInfo/VISAModule.hpp

@@ -20,8 +20,10 @@ SPDX-License-Identifier: MIT
 
 #include "LexicalScopes.hpp"
 #include "VISAIDebugEmitter.hpp"
+#include "IGC/common/Types.hpp"
 
 #include <map>
+#include <optional>
 #include <string>
 #include <vector>
 
@@ -542,6 +544,8 @@ class VISAModule {
 
   virtual llvm::StringRef GetVISAFuncName() const = 0;
 
+  virtual std::optional<enum ShaderType> getShaderType() const { return std::nullopt; }
+
   const InstInfoMap *GetInstInfoMap() { return &m_instInfoMap; }
 
   VISAModule &operator=(VISAModule &other) = delete;