Refactor copy and fill tests with events

Related-To: NEO-7434

Signed-off-by: Zbigniew Zdanowicz <[email protected]>

Author: zzdanowicz

level_zero/core/test/unit_tests/fixtures/cmdlist_fixture.cpp

@@ -137,5 +137,36 @@ void ImmediateCmdListSharedHeapsFixture::setUp() {
     ModuleMutableCommandListFixture::setUp();
 }
 
+bool AppendFillFixture::MockDriverFillHandle::findAllocationDataForRange(const void *buffer,
+                                                                         size_t size,
+                                                                         NEO::SvmAllocationData **allocData) {
+    mockAllocation.reset(new NEO::MockGraphicsAllocation(const_cast<void *>(buffer), size));
+    data.gpuAllocations.addAllocation(mockAllocation.get());
+    if (allocData) {
+        *allocData = &data;
+    }
+    return true;
+}
+
+void AppendFillFixture::setUp() {
+    dstPtr = new uint8_t[allocSize];
+    immediateDstPtr = new uint8_t[allocSize];
+
+    neoDevice = NEO::MockDevice::createWithNewExecutionEnvironment<NEO::MockDevice>(NEO::defaultHwInfo.get());
+    auto mockBuiltIns = new MockBuiltins();
+    neoDevice->executionEnvironment->rootDeviceEnvironments[0]->builtins.reset(mockBuiltIns);
+    NEO::DeviceVector devices;
+    devices.push_back(std::unique_ptr<NEO::Device>(neoDevice));
+    driverHandle = std::make_unique<Mock<MockDriverFillHandle>>();
+    driverHandle->initialize(std::move(devices));
+    device = driverHandle->devices[0];
+    neoDevice->deviceInfo.maxWorkGroupSize = 256;
+}
+
+void AppendFillFixture::tearDown() {
+    delete[] immediateDstPtr;
+    delete[] dstPtr;
+}
+
 } // namespace ult
 } // namespace L0

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

@@ -124,5 +124,72 @@ struct ImmediateCmdListSharedHeapsFixture : public ModuleMutableCommandListFixtu
     DebugManagerStateRestore restorer;
 };
 
+class AppendFillFixture : public DeviceFixture {
+  public:
+    class MockDriverFillHandle : public L0::DriverHandleImp {
+      public:
+        bool findAllocationDataForRange(const void *buffer,
+                                        size_t size,
+                                        NEO::SvmAllocationData **allocData) override;
+
+        const uint32_t rootDeviceIndex = 0u;
+        std::unique_ptr<NEO::GraphicsAllocation> mockAllocation;
+        NEO::SvmAllocationData data{rootDeviceIndex};
+    };
+
+    template <GFXCORE_FAMILY gfxCoreFamily>
+    class MockCommandList : public WhiteBox<::L0::CommandListCoreFamily<gfxCoreFamily>> {
+      public:
+        MockCommandList() : WhiteBox<::L0::CommandListCoreFamily<gfxCoreFamily>>() {}
+
+        ze_result_t appendLaunchKernelWithParams(Kernel *kernel,
+                                                 const ze_group_count_t *pThreadGroupDimensions,
+                                                 Event *event,
+                                                 const CmdListKernelLaunchParams &launchParams) override {
+            if (numberOfCallsToAppendLaunchKernelWithParams == thresholdOfCallsToAppendLaunchKernelWithParamsToFail) {
+                return ZE_RESULT_ERROR_UNKNOWN;
+            }
+            if (numberOfCallsToAppendLaunchKernelWithParams < 3) {
+                threadGroupDimensions[numberOfCallsToAppendLaunchKernelWithParams] = *pThreadGroupDimensions;
+                xGroupSizes[numberOfCallsToAppendLaunchKernelWithParams] = kernel->getGroupSize()[0];
+            }
+            numberOfCallsToAppendLaunchKernelWithParams++;
+            return CommandListCoreFamily<gfxCoreFamily>::appendLaunchKernelWithParams(kernel,
+                                                                                      pThreadGroupDimensions,
+                                                                                      event,
+                                                                                      launchParams);
+        }
+        ze_group_count_t threadGroupDimensions[3];
+        uint32_t xGroupSizes[3];
+        uint32_t thresholdOfCallsToAppendLaunchKernelWithParamsToFail = std::numeric_limits<uint32_t>::max();
+        uint32_t numberOfCallsToAppendLaunchKernelWithParams = 0;
+    };
+
+    void setUp();
+    void tearDown();
+
+    DebugManagerStateRestore restorer;
+
+    std::unique_ptr<Mock<MockDriverFillHandle>> driverHandle;
+    NEO::MockDevice *neoDevice = nullptr;
+    L0::Device *device = nullptr;
+    static constexpr size_t allocSize = 70;
+    static constexpr size_t patternSize = 8;
+    uint8_t *dstPtr = nullptr;
+    uint8_t pattern[patternSize] = {1, 2, 3, 4};
+
+    static constexpr size_t immediateAllocSize = 106;
+    uint8_t immediatePattern = 4;
+    uint8_t *immediateDstPtr = nullptr;
+};
+
+struct TestExpectedValues {
+    uint32_t expectedPacketsInUse = 0;
+    uint32_t expectedKernelCount = 0;
+    uint32_t expectedWalkerPostSyncOp = 0;
+    uint32_t expectedPostSyncPipeControls = 0;
+    bool postSyncAddressZero = false;
+};
+
 } // namespace ult
 } // namespace L0

level_zero/core/test/unit_tests/sources/cmdlist/CMakeLists.txt

@@ -20,16 +20,18 @@ target_sources(${TARGET_NAME} PRIVATE
                ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_append_launch_kernel_2.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_append_launch_kernel_3.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_append_memory.cpp
+               ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_append_multipartition_prologue.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_append_signal_event.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_append_wait_on_events.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_blit.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_fill.cpp
-               ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_append_multipartition_prologue.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_memory_extension.cpp
 )
 
 if(TESTS_XEHP_AND_LATER)
   target_sources(${TARGET_NAME} PRIVATE
+                 ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_copy_event_xehp_and_later.cpp
+                 ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_fill_event_xehp_and_later.cpp
                  ${CMAKE_CURRENT_SOURCE_DIR}/test_cmdlist_xehp_and_later.cpp
   )
 endif()