[CLN] Remove init_context function as unused (#3941)

Signed-off-by: Anatoly Myachev <[email protected]>

Author: anmyachev

third_party/intel/backend/driver.c

@@ -9,7 +9,6 @@
 #include <cstddef>
 #include <iostream>
 #include <string>
-#include <unordered_map>
 #include <utility>
 #include <vector>
 
@@ -25,8 +24,6 @@
 #include <Python.h>
 #include <numpy/arrayobject.h>
 
-static SyclQueueMap g_sycl_queue_map;
-
 static std::vector<ze_device_handle_t> g_devices;
 static std::vector<std::pair<sycl::device, ze_device_handle_t>>
     g_sycl_l0_device_list;
@@ -297,27 +294,6 @@ extern "C" EXPORT_FUNC PyObject *load_binary(PyObject *args) {
   }
 }
 
-extern "C" EXPORT_FUNC PyObject *init_context(PyObject *cap) {
-  void *queue = NULL;
-  if (!(queue = PyLong_AsVoidPtr(cap)))
-    return NULL;
-  sycl::queue *sycl_queue = static_cast<sycl::queue *>(queue);
-  if (g_sycl_queue_map.find(*sycl_queue) == g_sycl_queue_map.end()) {
-    const auto updated_sycl_devices = update(*sycl_queue, g_sycl_queue_map);
-    if (!updated_sycl_devices.empty()) {
-      // Update global data
-      const uint32_t deviceCount =
-          std::min(updated_sycl_devices.size(), g_devices.size());
-      for (uint32_t i = 0; i < deviceCount; ++i) {
-        g_devices[i] = sycl::get_native<sycl::backend::ext_oneapi_level_zero>(
-            updated_sycl_devices[i]);
-      }
-    }
-  }
-  auto context = g_sycl_queue_map[*sycl_queue].context;
-  return Py_BuildValue("(K)", (uint64_t)context);
-}
-
 extern "C" EXPORT_FUNC PyObject *init_devices(PyObject *cap) {
   void *queue = NULL;
   if (!(queue = PyLong_AsVoidPtr(cap)))

third_party/intel/backend/driver.py

@@ -186,15 +186,15 @@ class SpirvUtils:
 
     def __init__(self, cache_path: str):
         self.shared_library = ctypes.PyDLL(cache_path)
-        methods = ("init_context", "init_devices", "load_binary", "wait_on_sycl_queue")
+        methods = ("init_devices", "load_binary", "wait_on_sycl_queue")
         for method in methods:
             getattr(self.shared_library, method).restype = ctypes.py_object
             getattr(self.shared_library, method).argtypes = (ctypes.py_object, )
         self.shared_library.get_device_properties.restype = ctypes.py_object
         self.shared_library.get_device_properties.argtypes = (ctypes.c_int, )
 
     def __getattribute__(self, name):
-        if name in ("get_device_properties", "init_context", "init_devices", "wait_on_sycl_queue"):
+        if name in ("get_device_properties", "init_devices", "wait_on_sycl_queue"):
             shared_library = super().__getattribute__("shared_library")
             return getattr(shared_library, name)
 
@@ -306,7 +306,6 @@ def __init__(self):
         self.mod = compile_module_from_src(Path(os.path.join(dirname, "driver.c")).read_text(), "spirv_utils")
         self.load_binary = self.mod.load_binary
         self.get_device_properties = self.mod.get_device_properties
-        self.context = self.mod.init_context(self.get_sycl_queue())
         self.device_count = self.mod.init_devices(self.get_sycl_queue())
         self.wait_on_sycl_queue = self.mod.wait_on_sycl_queue
 

third_party/intel/backend/include/sycl_functions.h

@@ -9,8 +9,6 @@
 #include <sstream>
 #include <string>
 #include <string_view>
-#include <unordered_map>
-#include <variant>
 
 #include <level_zero/ze_api.h>
 #include <sycl/sycl.hpp>
@@ -22,8 +20,6 @@ typedef struct l0_resc_handles {
   ze_command_list_handle_t cmd_list;
 } l0_resc_handles;
 
-using SyclQueueMap = std::unordered_map<sycl::queue, l0_resc_handles>;
-
 // Create an exception handler for asynchronous SYCL exceptions
 auto exception_handler = [](sycl::exception_list e_list) {
   for (std::exception_ptr const &e : e_list) {
@@ -147,36 +143,4 @@ void printModuleKernelName(ze_module_handle_t hModule) {
   }
 }
 
-std::vector<sycl::device> update(sycl::queue sycl_queue,
-                                 SyclQueueMap &sycl_queue_map) {
-  // Get l0-context
-  auto sycl_context = sycl_queue.get_context();
-  ze_context_handle_t hCtxt =
-      sycl::get_native<sycl::backend::ext_oneapi_level_zero>(sycl_context);
-  // Get l0-device
-  const std::vector<sycl::device> &sycl_devices = sycl_context.get_devices();
-  assert(sycl_devices.size() > 0);
-  ze_device_handle_t hDev =
-      sycl::get_native<sycl::backend::ext_oneapi_level_zero>(
-          sycl_devices.front());
-  // Get l0-queue
-  bool immediate_cmd_list = false;
-  std::variant<ze_command_queue_handle_t, ze_command_list_handle_t> queue_var =
-      sycl::get_native<sycl::backend::ext_oneapi_level_zero>(sycl_queue);
-  auto l0_queue = std::get_if<ze_command_queue_handle_t>(&queue_var);
-  if (l0_queue == nullptr) {
-    auto imm_cmd_list = std::get_if<ze_command_list_handle_t>(&queue_var);
-    if (imm_cmd_list == nullptr) {
-      return {};
-    }
-    immediate_cmd_list = true;
-    sycl_queue_map[sycl_queue].cmd_list = *imm_cmd_list;
-  }
-  sycl_queue_map[sycl_queue].context = hCtxt;
-  sycl_queue_map[sycl_queue].device = hDev;
-  sycl_queue_map[sycl_queue].queue = immediate_cmd_list ? 0 : *l0_queue;
-
-  return sycl_devices;
-}
-
 #endif // SYCL_FUNCTIONS_INCLUDE_H_

utils/SPIRVRunner/SPIRVRunner.cpp

@@ -4,14 +4,12 @@
 
 #include "llvm_parser.h"
 #include "sycl_functions.h"
-#include <algorithm>
 #include <cstdlib>
 #include <filesystem>
 #include <fstream>
 #include <ios>
 #include <iostream>
 #include <nlohmann/json.hpp>
-#include <regex>
 #include <string>
 #include <vector>
 using json = nlohmann::json;
@@ -108,8 +106,6 @@ struct KernelArguments {
 };
 
 /** SYCL Globals **/
-SyclQueueMap g_sycl_queue_map;
-
 static std::vector<ze_device_handle_t> g_devices;
 static std::vector<std::pair<sycl::device, ze_device_handle_t>>
     g_sycl_l0_device_list;
@@ -188,23 +184,6 @@ loadBinary(const std::string &kernel_name, const std::string &build_flags,
   return std::make_tuple(mod, fun, n_regs, n_spills);
 }
 
-ze_context_handle_t initContext(sycl::queue *sycl_queue) {
-  if (g_sycl_queue_map.find(*sycl_queue) == g_sycl_queue_map.end()) {
-    const auto updated_sycl_devices = update(*sycl_queue, g_sycl_queue_map);
-    if (!updated_sycl_devices.empty()) {
-      // Update global data
-      const uint32_t deviceCount =
-          std::min(updated_sycl_devices.size(), g_devices.size());
-      for (uint32_t i = 0; i < deviceCount; ++i) {
-        g_devices[i] = sycl::get_native<sycl::backend::ext_oneapi_level_zero>(
-            updated_sycl_devices[i]);
-      }
-    }
-  }
-  auto context = g_sycl_queue_map[*sycl_queue].context;
-  return context;
-}
-
 size_t initDevices(sycl::queue *sycl_queue) {
   auto sycl_context = sycl_queue->get_context();
 
@@ -466,7 +445,6 @@ int main(int argc, char **argv) {
 
     std::cout << "Running on device: "
               << q.get_device().get_info<sycl::info::device::name>() << "\n";
-    initContext(&q);
     initDevices(&q);
 
     // Parse the JSON file and create argument dictionary