-
Notifications
You must be signed in to change notification settings - Fork 116
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #2170 from igchor/enqueueWaitTest
[CTS] extend tests for urEnqueueEventsWait
- Loading branch information
Showing
5 changed files
with
288 additions
and
49 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
218 changes: 218 additions & 0 deletions
218
test/conformance/enqueue/urEnqueueEventsWaitMultiDevice.cpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,218 @@ | ||
| // Copyright (C) 2024 Intel Corporation | ||
| // Part of the Unified-Runtime Project, under the Apache License v2.0 with LLVM Exceptions. | ||
| // See LICENSE.TXT | ||
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | ||
|
|
||
| #include "helpers.h" | ||
|
|
||
| #include <thread> | ||
|
|
||
| #include <uur/fixtures.h> | ||
| #include <uur/raii.h> | ||
|
|
||
| struct urEnqueueEventsWaitMultiDeviceTest : uur::urMultiQueueMultiDeviceTest { | ||
| void SetUp() override { SetUp(2); /* we need at least 2 devices */ } | ||
|
|
||
| void SetUp(size_t numDuplicateDevices) { | ||
| UUR_RETURN_ON_FATAL_FAILURE(uur::urMultiQueueMultiDeviceTest::SetUp( | ||
| uur::KernelsEnvironment::instance->devices, numDuplicateDevices)); | ||
|
|
||
| for (auto device : devices) { | ||
| ur_device_usm_access_capability_flags_t shared_usm_single = 0; | ||
| EXPECT_SUCCESS(uur::GetDeviceUSMSingleSharedSupport( | ||
| device, shared_usm_single)); | ||
| if (!shared_usm_single) { | ||
| GTEST_SKIP() << "Shared USM is not supported by the device."; | ||
| } | ||
| } | ||
|
|
||
| ptrs.resize(devices.size()); | ||
| for (size_t i = 0; i < devices.size(); i++) { | ||
| EXPECT_SUCCESS(urUSMSharedAlloc(context, devices[i], nullptr, | ||
| nullptr, size, &ptrs[i])); | ||
| } | ||
| } | ||
|
|
||
| void TearDown() override { | ||
| for (auto ptr : ptrs) { | ||
| if (ptr) { | ||
| EXPECT_SUCCESS(urUSMFree(context, ptr)); | ||
| } | ||
| } | ||
| UUR_RETURN_ON_FATAL_FAILURE( | ||
| uur::urMultiQueueMultiDeviceTest::TearDown()); | ||
| } | ||
|
|
||
| void initData() { | ||
| EXPECT_SUCCESS(urEnqueueUSMFill(queues[0], ptrs[0], sizeof(pattern), | ||
| &pattern, size, 0, nullptr, nullptr)); | ||
| EXPECT_SUCCESS(urQueueFinish(queues[0])); | ||
| } | ||
|
|
||
| void verifyData(void *ptr, uint32_t pattern) { | ||
| for (size_t i = 0; i < count; i++) { | ||
| ASSERT_EQ(reinterpret_cast<uint32_t *>(ptr)[i], pattern); | ||
| } | ||
| } | ||
|
|
||
| uint32_t pattern = 42; | ||
| const size_t count = 1024; | ||
| const size_t size = sizeof(uint32_t) * count; | ||
|
|
||
| std::vector<void *> ptrs; | ||
| }; | ||
|
|
||
| TEST_F(urEnqueueEventsWaitMultiDeviceTest, EmptyWaitList) { | ||
| initData(); | ||
|
|
||
| ASSERT_SUCCESS(urEnqueueUSMMemcpy(queues[0], false, ptrs[1], ptrs[0], size, | ||
| 0, nullptr, nullptr)); | ||
| ASSERT_SUCCESS(urEnqueueEventsWait(queues[0], 0, nullptr, nullptr)); | ||
| ASSERT_SUCCESS(urQueueFinish(queues[0])); | ||
|
|
||
| verifyData(ptrs[1], pattern); | ||
| } | ||
|
|
||
| TEST_F(urEnqueueEventsWaitMultiDeviceTest, EmptyWaitListWithEvent) { | ||
| initData(); | ||
|
|
||
| ASSERT_SUCCESS(urEnqueueUSMMemcpy(queues[0], false, ptrs[1], ptrs[0], size, | ||
| 0, nullptr, nullptr)); | ||
|
|
||
| uur::raii::Event event; | ||
| ASSERT_SUCCESS(urEnqueueEventsWait(queues[0], 0, nullptr, event.ptr())); | ||
| ASSERT_SUCCESS(urEventWait(1, event.ptr())); | ||
|
|
||
| verifyData(ptrs[1], pattern); | ||
| } | ||
|
|
||
| TEST_F(urEnqueueEventsWaitMultiDeviceTest, EnqueueWaitOnADifferentQueue) { | ||
| initData(); | ||
|
|
||
| uur::raii::Event event; | ||
| ASSERT_SUCCESS(urEnqueueUSMMemcpy(queues[0], false, ptrs[1], ptrs[0], size, | ||
| 0, nullptr, event.ptr())); | ||
| ASSERT_SUCCESS(urEnqueueEventsWait(queues[0], 1, event.ptr(), nullptr)); | ||
| ASSERT_SUCCESS(urQueueFinish(queues[0])); | ||
|
|
||
| verifyData(ptrs[1], pattern); | ||
| } | ||
|
|
||
| struct urEnqueueEventsWaitMultiDeviceMTTest | ||
| : urEnqueueEventsWaitMultiDeviceTest, | ||
| testing::WithParamInterface<uur::BoolTestParam> { | ||
| void doComputation(std::function<void(size_t)> work) { | ||
| auto multiThread = GetParam().value; | ||
| std::vector<std::thread> threads; | ||
| for (size_t i = 0; i < devices.size(); i++) { | ||
| if (multiThread) { | ||
| threads.emplace_back(work, i); | ||
| } else { | ||
| work(i); | ||
| } | ||
| } | ||
| for (auto &thread : threads) { | ||
| thread.join(); | ||
| } | ||
| } | ||
|
|
||
| void SetUp() override { | ||
| const size_t numDuplicateDevices = 8; | ||
| UUR_RETURN_ON_FATAL_FAILURE( | ||
| urEnqueueEventsWaitMultiDeviceTest::SetUp(numDuplicateDevices)); | ||
| } | ||
|
|
||
| void TearDown() override { urEnqueueEventsWaitMultiDeviceTest::TearDown(); } | ||
| }; | ||
|
|
||
| template <typename T> | ||
| inline std::string | ||
| printParams(const testing::TestParamInfo<typename T::ParamType> &info) { | ||
| std::stringstream ss; | ||
|
|
||
| auto param1 = info.param; | ||
| ss << (param1.value ? "" : "No") << param1.name; | ||
|
|
||
| return ss.str(); | ||
| } | ||
|
|
||
| INSTANTIATE_TEST_SUITE_P( | ||
| , urEnqueueEventsWaitMultiDeviceMTTest, | ||
| testing::ValuesIn(uur::BoolTestParam::makeBoolParam("MultiThread")), | ||
| printParams<urEnqueueEventsWaitMultiDeviceMTTest>); | ||
|
|
||
| TEST_P(urEnqueueEventsWaitMultiDeviceMTTest, EnqueueWaitSingleQueueMultiOps) { | ||
| std::vector<uint32_t> data(count, pattern); | ||
|
|
||
| auto work = [this, &data](size_t i) { | ||
| ASSERT_SUCCESS(urEnqueueUSMMemcpy( | ||
| queues[0], false, ptrs[i], data.data(), size, 0, nullptr, nullptr)); | ||
| }; | ||
|
|
||
| doComputation(work); | ||
|
|
||
| auto verify = [this](size_t i) { | ||
| uur::raii::Event event; | ||
| ASSERT_SUCCESS(urEnqueueEventsWait(queues[0], 0, nullptr, event.ptr())); | ||
| ASSERT_SUCCESS(urEventWait(1, event.ptr())); | ||
|
|
||
| verifyData(ptrs[i], pattern); | ||
| }; | ||
|
|
||
| doComputation(verify); | ||
| } | ||
|
|
||
| TEST_P(urEnqueueEventsWaitMultiDeviceMTTest, EnqueueWaitOnAllQueues) { | ||
| std::vector<uur::raii::Event> eventsRaii(devices.size()); | ||
| std::vector<ur_event_handle_t> events(devices.size()); | ||
| auto work = [this, &events, &eventsRaii](size_t i) { | ||
| ASSERT_SUCCESS(urEnqueueUSMFill(queues[i], ptrs[i], sizeof(pattern), | ||
| &pattern, size, 0, nullptr, | ||
| eventsRaii[i].ptr())); | ||
| events[i] = eventsRaii[i].get(); | ||
| }; | ||
|
|
||
| doComputation(work); | ||
|
|
||
| uur::raii::Event gatherEvent; | ||
| ASSERT_SUCCESS(urEnqueueEventsWait(queues[0], devices.size(), events.data(), | ||
| gatherEvent.ptr())); | ||
| ASSERT_SUCCESS(urEventWait(1, gatherEvent.ptr())); | ||
|
|
||
| for (size_t i = 0; i < devices.size(); i++) { | ||
| verifyData(ptrs[i], pattern); | ||
| } | ||
| } | ||
|
|
||
| TEST_P(urEnqueueEventsWaitMultiDeviceMTTest, | ||
| EnqueueWaitOnAllQueuesCommonDependency) { | ||
| uur::raii::Event event; | ||
| ASSERT_SUCCESS(urEnqueueUSMFill(queues[0], ptrs[0], sizeof(pattern), | ||
| &pattern, size, 0, nullptr, event.ptr())); | ||
|
|
||
| std::vector<uur::raii::Event> perQueueEvents(devices.size()); | ||
| std::vector<ur_event_handle_t> eventHandles(devices.size()); | ||
| auto work = [this, &event, &perQueueEvents, &eventHandles](size_t i) { | ||
| ASSERT_SUCCESS(urEnqueueEventsWait(queues[i], 1, event.ptr(), | ||
| perQueueEvents[i].ptr())); | ||
| eventHandles[i] = perQueueEvents[i].get(); | ||
| }; | ||
|
|
||
| doComputation(work); | ||
|
|
||
| uur::raii::Event hGatherEvent; | ||
| ASSERT_SUCCESS(urEnqueueEventsWait(queues[0], eventHandles.size(), | ||
| eventHandles.data(), | ||
| hGatherEvent.ptr())); | ||
| ASSERT_SUCCESS(urEventWait(1, hGatherEvent.ptr())); | ||
|
|
||
| for (auto &event : eventHandles) { | ||
| ur_event_status_t status; | ||
| ASSERT_SUCCESS( | ||
| urEventGetInfo(event, UR_EVENT_INFO_COMMAND_EXECUTION_STATUS, | ||
| sizeof(ur_event_status_t), &status, nullptr)); | ||
| ASSERT_EQ(status, UR_EVENT_STATUS_COMPLETE); | ||
| } | ||
|
|
||
| verifyData(ptrs[0], pattern); | ||
| } |
Oops, something went wrong.