refactor: Adding support for extending L0 context object

Related-To: NEO-13406

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

Author: jchodor

level_zero/core/source/context/CMakeLists.txt

@@ -1,5 +1,5 @@
 #
-# Copyright (C) 2023-2024 Intel Corporation
+# Copyright (C) 2023-2025 Intel Corporation
 #
 # SPDX-License-Identifier: MIT
 #
@@ -13,4 +13,7 @@ target_sources(${L0_STATIC_LIB_NAME}
                ${CMAKE_CURRENT_SOURCE_DIR}/context_imp.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/context_imp.h
                ${CMAKE_CURRENT_SOURCE_DIR}/context.h
+               ${CMAKE_CURRENT_SOURCE_DIR}${BRANCH_DIR_SUFFIX}/context_ext.cpp
 )
+
+add_subdirectories()

level_zero/core/source/context/context.h

@@ -28,6 +28,8 @@ namespace L0 {
 struct DriverHandle;
 struct Image;
 
+class ContextExt;
+
 struct Context : _ze_context_handle_t {
     inline static ze_memory_type_t parseUSMType(InternalMemoryType memoryType) {
         switch (memoryType) {
@@ -178,6 +180,8 @@ struct Context : _ze_context_handle_t {
 
     static Context *fromHandle(ze_context_handle_t handle) { return static_cast<Context *>(handle); }
     inline ze_context_handle_t toHandle() { return this; }
+
+    virtual ContextExt *getContextExt() = 0;
 };
 
 } // namespace L0

level_zero/core/source/context/context_ext.cpp

@@ -0,0 +1,21 @@
+/*
+ * Copyright (C) 2025 Intel Corporation
+ *
+ * SPDX-License-Identifier: MIT
+ *
+ */
+
+namespace L0 {
+
+class ContextExt;
+struct DriverHandle;
+
+ContextExt *createContextExt(DriverHandle *driverHandle) {
+    return nullptr;
+}
+
+void destroyContextExt(ContextExt *ctxExt) {
+    return;
+}
+
+} // namespace L0

level_zero/core/source/context/context_imp.cpp

@@ -61,6 +61,11 @@ DriverHandle *ContextImp::getDriverHandle() {
 
 ContextImp::ContextImp(DriverHandle *driverHandle) {
     this->driverHandle = static_cast<DriverHandleImp *>(driverHandle);
+    this->contextExt = createContextExt(driverHandle);
+}
+
+ContextImp::~ContextImp() {
+    destroyContextExt(this->contextExt);
 }
 
 ze_result_t ContextImp::allocHostMem(const ze_host_mem_alloc_desc_t *hostDesc,

level_zero/core/source/context/context_imp.h

@@ -1,5 +1,5 @@
 /*
- * Copyright (C) 2020-2024 Intel Corporation
+ * Copyright (C) 2020-2025 Intel Corporation
  *
  * SPDX-License-Identifier: MIT
  *
@@ -22,9 +22,12 @@ struct DriverHandleImp;
 struct Device;
 struct IpcCounterBasedEventData;
 
+ContextExt *createContextExt(DriverHandle *driverHandle);
+void destroyContextExt(ContextExt *ctxExt);
+
 struct ContextImp : Context {
     ContextImp(DriverHandle *driverHandle);
-    ~ContextImp() override = default;
+    ~ContextImp() override;
     ze_result_t destroy() override;
     ze_result_t getStatus() override;
     DriverHandle *getDriverHandle() override;
@@ -187,6 +190,10 @@ struct ContextImp : Context {
         unsigned int elementSizeInBytes,
         size_t *rowPitch) override;
 
+    ContextExt *getContextExt() override {
+        return contextExt;
+    }
+
   protected:
     ze_result_t getIpcMemHandlesImpl(const void *ptr, uint32_t *numIpcHandles, ze_ipc_mem_handle_t *pIpcHandles);
     void setIPCHandleData(NEO::GraphicsAllocation *graphicsAllocation, uint64_t handle, IpcMemoryData &ipcData, uint64_t ptrAddress, uint8_t type);
@@ -197,6 +204,7 @@ struct ContextImp : Context {
     std::vector<ze_device_handle_t> deviceHandles;
     DriverHandleImp *driverHandle = nullptr;
     uint32_t numDevices = 0;
+    ContextExt *contextExt = nullptr;
 };
 
 } // namespace L0

level_zero/core/source/memory_manager/CMakeLists.txt

@@ -1,5 +1,5 @@
 #
-# Copyright (C) 2023 Intel Corporation
+# Copyright (C) 2023-2025 Intel Corporation
 #
 # SPDX-License-Identifier: MIT
 #
@@ -9,3 +9,5 @@ target_sources(${L0_STATIC_LIB_NAME}
                ${CMAKE_CURRENT_SOURCE_DIR}/CMakeLists.txt
                ${CMAKE_CURRENT_SOURCE_DIR}/compression_selector_l0.cpp
 )
+
+add_subdirectories()

level_zero/core/test/unit_tests/fixtures/device_fixture.h

@@ -58,14 +58,16 @@ template <typename T>
 struct DeviceFixtureWithCustomMemoryManager : public DeviceFixture {
     void setUp() {
         auto executionEnvironment = NEO::MockDevice::prepareExecutionEnvironment(defaultHwInfo.get(), 0u);
-        auto memoryManager = new T(*executionEnvironment);
+        memoryManager = new T(*executionEnvironment);
         executionEnvironment->memoryManager.reset(memoryManager);
         DeviceFixture::setupWithExecutionEnvironment(*executionEnvironment);
     }
 
     void tearDown() {
         DeviceFixture::tearDown();
     }
+
+    T *memoryManager = nullptr;
 };
 
 struct DriverHandleGetMemHandlePtrMock : public L0::DriverHandleImp {