Add WaitOnMemory and WriteToMemory experimental extensions

Signed-off-by: Aravind Gopalakrishnan <[email protected]>

Author: aravindksg

level_zero/core/source/cmdlist/cmdlist.h

@@ -141,6 +141,10 @@ struct CommandList : _ze_command_list_handle_t {
     virtual ze_result_t appendMIBBEnd() = 0;
     virtual ze_result_t appendMINoop() = 0;
     virtual ze_result_t appendPipeControl(void *dstPtr, uint64_t value) = 0;
+    virtual ze_result_t appendWaitOnMemory(void *desc, void *ptr,
+                                           uint32_t data, ze_event_handle_t hSignalEvent) = 0;
+    virtual ze_result_t appendWriteToMemory(void *desc, void *ptr,
+                                            uint64_t data) = 0;
 
     static CommandList *create(uint32_t productFamily, Device *device, NEO::EngineGroupType engineGroupType,
                                ze_result_t &resultValue);

level_zero/core/source/cmdlist/cmdlist_extended/cmdlist_extended.inl

@@ -59,4 +59,19 @@ ze_result_t CommandListCoreFamily<gfxCoreFamily>::appendPipeControl(void *dstPtr
     return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
 }
 
+template <GFXCORE_FAMILY gfxCoreFamily>
+ze_result_t CommandListCoreFamily<gfxCoreFamily>::appendWaitOnMemory(void *desc,
+                                                                     void *ptr,
+                                                                     uint32_t data,
+                                                                     ze_event_handle_t hSignalEvent) {
+    return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+}
+
+template <GFXCORE_FAMILY gfxCoreFamily>
+ze_result_t CommandListCoreFamily<gfxCoreFamily>::appendWriteToMemory(void *desc,
+                                                                      void *ptr,
+                                                                      uint64_t data) {
+    return ZE_RESULT_ERROR_UNSUPPORTED_FEATURE;
+}
+
 } // namespace L0

level_zero/core/source/cmdlist/cmdlist_hw.h

@@ -135,6 +135,10 @@ struct CommandListCoreFamily : CommandListImp {
     ze_result_t appendMIBBEnd() override;
     ze_result_t appendMINoop() override;
     ze_result_t appendPipeControl(void *dstPtr, uint64_t value) override;
+    ze_result_t appendWaitOnMemory(void *desc, void *ptr,
+                                   uint32_t data, ze_event_handle_t hSignalEvent) override;
+    ze_result_t appendWriteToMemory(void *desc, void *ptr,
+                                    uint64_t data) override;
 
     ze_result_t appendQueryKernelTimestamps(uint32_t numEvents, ze_event_handle_t *phEvents, void *dstptr,
                                             const size_t *pOffsets, ze_event_handle_t hSignalEvent,

level_zero/core/test/unit_tests/mocks/mock_cmdlist.h

@@ -319,6 +319,13 @@ struct MockCommandList : public CommandList {
     ADDMETHOD_NOBASE(appendPipeControl, ze_result_t, ZE_RESULT_SUCCESS,
                      (void *dstPtr,
                       uint64_t value));
+    ADDMETHOD_NOBASE(appendWaitOnMemory, ze_result_t, ZE_RESULT_SUCCESS,
+                     (void *desc, void *ptr,
+                      uint32_t data, ze_event_handle_t hSignalEvent));
+
+    ADDMETHOD_NOBASE(appendWriteToMemory, ze_result_t, ZE_RESULT_SUCCESS,
+                     (void *desc, void *ptr,
+                      uint64_t data));
 
     ADDMETHOD_NOBASE(executeCommandListImmediate, ze_result_t, ZE_RESULT_SUCCESS,
                      (bool perforMigration));