refactor: L0 module input handling refactor

Related-To: NEO-14135

Signed-off-by: Chodor, Jaroslaw <[email protected]>

Author: jchodor

level_zero/core/source/module/CMakeLists.txt

@@ -1,5 +1,5 @@
 #
-# Copyright (C) 2023 Intel Corporation
+# Copyright (C) 2023-2025 Intel Corporation
 #
 # SPDX-License-Identifier: MIT
 #
@@ -12,6 +12,7 @@ target_sources(${L0_STATIC_LIB_NAME}
                ${CMAKE_CURRENT_SOURCE_DIR}/module_build_log.h
                ${CMAKE_CURRENT_SOURCE_DIR}/module_imp.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/module_imp.h
+               ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/translation_unit_ext.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}module_imp_helper.cpp
 )
 

level_zero/core/source/module/module_imp.cpp

@@ -254,8 +254,8 @@ ze_result_t ModuleTranslationUnit::staticLinkSpirV(std::vector<const char *> inp
     return this->compileGenBinary(linkInputArgs, true);
 }
 
-ze_result_t ModuleTranslationUnit::buildFromSpirV(const char *input, uint32_t inputSize, const char *buildOptions, const char *internalBuildOptions,
-                                                  const ze_module_constants_t *pConstants) {
+ze_result_t ModuleTranslationUnit::buildFromIntermediate(IGC::CodeType::CodeType_t intermediateType, const char *input, uint32_t inputSize, const char *buildOptions, const char *internalBuildOptions,
+                                                         const ze_module_constants_t *pConstants) {
     const auto &neoDevice = device->getNEODevice();
     auto compilerInterface = neoDevice->getCompilerInterface();
     const auto driverHandle = static_cast<DriverHandleImp *>(device->getDriverHandle());
@@ -283,7 +283,7 @@ ze_result_t ModuleTranslationUnit::buildFromSpirV(const char *input, uint32_t in
         }
     }
 
-    NEO::TranslationInput inputArgs = {IGC::CodeType::spirV, IGC::CodeType::oclGenBin};
+    NEO::TranslationInput inputArgs = {intermediateType, IGC::CodeType::oclGenBin};
 
     inputArgs.src = ArrayRef<const char>(input, inputSize);
     inputArgs.apiOptions = ArrayRef<const char>(this->options.c_str(), this->options.length());
@@ -750,7 +750,14 @@ inline ze_result_t ModuleImp::initializeTranslationUnit(const ze_module_desc_t *
                                                          internalBuildOptions.c_str(),
                                                          desc->pConstants);
         } else {
-            return ZE_RESULT_ERROR_INVALID_ENUMERATION;
+            this->precompiled = false;
+            this->isFunctionSymbolExportEnabled = true;
+            this->isGlobalSymbolExportEnabled = true;
+            return this->translationUnit->buildExt(desc->format,
+                                                   reinterpret_cast<const char *>(desc->pInputModule),
+                                                   static_cast<uint32_t>(desc->inputSize),
+                                                   buildOptions.c_str(),
+                                                   internalBuildOptions.c_str());
         }
     }
 }

level_zero/core/source/module/module_imp.h

@@ -50,7 +50,14 @@ struct ModuleTranslationUnit {
     ModuleTranslationUnit(L0::Device *device);
     virtual ~ModuleTranslationUnit();
     MOCKABLE_VIRTUAL ze_result_t buildFromSpirV(const char *input, uint32_t inputSize, const char *buildOptions, const char *internalBuildOptions,
-                                                const ze_module_constants_t *pConstants);
+                                                const ze_module_constants_t *pConstants) {
+        return buildFromIntermediate(IGC::CodeType::spirV, input, inputSize, buildOptions, internalBuildOptions, pConstants);
+    }
+
+    MOCKABLE_VIRTUAL ze_result_t buildExt(ze_module_format_t inputFormat, const char *input, uint32_t inputSize, const char *buildOptions, const char *internalBuildOptions);
+
+    MOCKABLE_VIRTUAL ze_result_t buildFromIntermediate(IGC::CodeType::CodeType_t intermediateType, const char *input, uint32_t inputSize, const char *buildOptions, const char *internalBuildOptions,
+                                                       const ze_module_constants_t *pConstants);
     MOCKABLE_VIRTUAL ze_result_t staticLinkSpirV(std::vector<const char *> inputSpirVs, std::vector<uint32_t> inputModuleSizes, const char *buildOptions, const char *internalBuildOptions,
                                                  std::vector<const ze_module_constants_t *> specConstants);
     MOCKABLE_VIRTUAL ze_result_t createFromNativeBinary(const char *input, size_t inputSize, const char *internalBuildOptions);

level_zero/core/source/module/translation_unit_ext.cpp

@@ -0,0 +1,16 @@
+/*
+ * Copyright (C) 2025 Intel Corporation
+ *
+ * SPDX-License-Identifier: MIT
+ *
+ */
+
+#include "level_zero/core/source/module/module_imp.h"
+
+namespace L0 {
+
+ze_result_t ModuleTranslationUnit::buildExt(ze_module_format_t inputFormat, const char *input, uint32_t inputSize, const char *buildOptions, const char *internalBuildOptions) {
+    return ZE_RESULT_ERROR_INVALID_ENUMERATION;
+}
+
+} // namespace L0