Fix Native Device Init

Signed-off-by: Neil R. Spruit <[email protected]>

Author: nrspruit

source/adapters/level_zero/program.cpp

@@ -84,8 +84,6 @@ UR_APIEXPORT ur_result_t UR_APICALL urProgramCreateWithBinary(
     ur_program_handle_t
         *Program ///< [out] pointer to handle of Program object created.
 ) {
-  std::ignore = Device;
-  std::ignore = Properties;
   // In OpenCL, clCreateProgramWithBinary() can be used to load any of the
   // following: "program executable", "compiled program", or "library of
   // compiled programs".  In addition, the loaded program can be either
@@ -99,7 +97,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urProgramCreateWithBinary(
 
   try {
     ur_program_handle_t_ *UrProgram = new ur_program_handle_t_(
-        ur_program_handle_t_::Native, Context, Binary, Size);
+        ur_program_handle_t_::Native, Context, Device, Properties, Binary, Size);
     *Program = reinterpret_cast<ur_program_handle_t>(UrProgram);
   } catch (const std::bad_alloc &) {
     return UR_RESULT_ERROR_OUT_OF_HOST_MEMORY;

source/adapters/level_zero/program.hpp

@@ -65,7 +65,7 @@ struct ur_program_handle_t_ : _ur_object {
     ze_module_constants_t ZeSpecConstants;
   };
 
-  // Construct a program in IL or Native state.
+  // Construct a program in IL.
   ur_program_handle_t_(state St, ur_context_handle_t Context, const void *Input,
                        size_t Length)
       : Context{Context},
@@ -74,6 +74,15 @@ struct ur_program_handle_t_ : _ur_object {
     std::memcpy(Code.get(), Input, Length);
   }
 
+  // Construct a program in NATIVE.
+  ur_program_handle_t_(state St, ur_context_handle_t Context, ur_device_handle_t Device, const ur_program_properties_t *Properties, const void *Input,
+                       size_t Length)
+      : Context{Context}, NativeDevice(Device), NativeProperties(Properties),
+        OwnZeModule{true}, State{St}, Code{new uint8_t[Length]},
+        CodeLength{Length}, ZeModule{nullptr}, ZeBuildLog{nullptr} {
+    std::memcpy(Code.get(), Input, Length);
+  }
+
   // Construct a program in Exe or Invalid state.
   ur_program_handle_t_(state St, ur_context_handle_t Context,
                        ze_module_handle_t ZeModule,
@@ -108,6 +117,13 @@ struct ur_program_handle_t_ : _ur_object {
 
   const ur_context_handle_t Context; // Context of the program.
 
+
+  // Device Handle used for the Native Build
+  ur_device_handle_t NativeDevice;
+
+  // Properties used for the Native Build
+  const ur_program_properties_t *NativeProperties;
+
   // Indicates if we own the ZeModule or it came from interop that
   // asked to not transfer the ownership to SYCL RT.
   const bool OwnZeModule;