| // Copyright 2023 The Dawn Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| #include <atomic> |
| #include <cstdint> |
| #include <utility> |
| #include <vector> |
| |
| #include "dawn/common/FutureUtils.h" |
| #include "dawn/tests/DawnTest.h" |
| #include "dawn/webgpu.h" |
| |
| namespace dawn { |
| namespace { |
| |
| std::pair<wgpu::Instance, wgpu::Device> CreateExtraInstance(wgpu::InstanceDescriptor* desc) { |
| // IMPORTANT: DawnTest overrides RequestAdapter and RequestDevice and mixes |
| // up the two instances. We use these to bypass the override. |
| auto* requestAdapter = reinterpret_cast<WGPUProcInstanceRequestAdapter>( |
| wgpuGetProcAddress(nullptr, "wgpuInstanceRequestAdapter")); |
| auto* requestDevice = reinterpret_cast<WGPUProcAdapterRequestDevice>( |
| wgpuGetProcAddress(nullptr, "wgpuAdapterRequestDevice")); |
| |
| wgpu::Instance instance2 = wgpu::CreateInstance(desc); |
| |
| wgpu::Adapter adapter2; |
| requestAdapter( |
| instance2.Get(), nullptr, |
| [](WGPURequestAdapterStatus status, WGPUAdapter adapter, const char*, void* userdata) { |
| ASSERT_EQ(status, WGPURequestAdapterStatus_Success); |
| *reinterpret_cast<wgpu::Adapter*>(userdata) = wgpu::Adapter(adapter); |
| }, |
| &adapter2); |
| DAWN_ASSERT(adapter2); |
| |
| wgpu::Device device2; |
| requestDevice( |
| adapter2.Get(), nullptr, |
| [](WGPURequestDeviceStatus status, WGPUDevice device, const char*, void* userdata) { |
| ASSERT_EQ(status, WGPURequestDeviceStatus_Success); |
| *reinterpret_cast<wgpu::Device*>(userdata) = wgpu::Device(device); |
| }, |
| &device2); |
| DAWN_ASSERT(device2); |
| |
| return std::pair(std::move(instance2), std::move(device2)); |
| } |
| |
| // EventCompletionTests |
| |
| enum class WaitType { |
| TimedWaitAny, |
| SpinWaitAny, |
| SpinProcessEvents, |
| }; |
| |
| enum class WaitTypeAndCallbackMode { |
| TimedWaitAny_Future, |
| TimedWaitAny_FutureSpontaneous, |
| SpinWaitAny_Future, |
| SpinWaitAny_FutureSpontaneous, |
| SpinProcessEvents_ProcessEvents, |
| SpinProcessEvents_ProcessEventsSpontaneous, |
| }; |
| |
| std::ostream& operator<<(std::ostream& o, WaitTypeAndCallbackMode waitMode) { |
| switch (waitMode) { |
| case WaitTypeAndCallbackMode::TimedWaitAny_Future: |
| return o << "TimedWaitAny_Future"; |
| case WaitTypeAndCallbackMode::SpinWaitAny_Future: |
| return o << "SpinWaitAny_Future"; |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEvents: |
| return o << "SpinProcessEvents_ProcessEvents"; |
| case WaitTypeAndCallbackMode::TimedWaitAny_FutureSpontaneous: |
| return o << "TimedWaitAny_FutureSpontaneous"; |
| case WaitTypeAndCallbackMode::SpinWaitAny_FutureSpontaneous: |
| return o << "SpinWaitAny_FutureSpontaneous"; |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEventsSpontaneous: |
| return o << "SpinProcessEvents_ProcessEventsSpontaneous"; |
| } |
| } |
| |
| DAWN_TEST_PARAM_STRUCT(EventCompletionTestParams, WaitTypeAndCallbackMode); |
| |
| class EventCompletionTests : public DawnTestWithParams<EventCompletionTestParams> { |
| protected: |
| wgpu::Instance testInstance; |
| wgpu::Device testDevice; |
| wgpu::Queue testQueue; |
| std::vector<wgpu::FutureWaitInfo> mFutures; |
| std::atomic<uint64_t> mCallbacksCompletedCount = 0; |
| uint64_t mCallbacksIssuedCount = 0; |
| uint64_t mCallbacksWaitedCount = 0; |
| |
| void SetUp() override { |
| DawnTestWithParams::SetUp(); |
| WaitTypeAndCallbackMode mode = GetParam().mWaitTypeAndCallbackMode; |
| if (UsesWire()) { |
| DAWN_TEST_UNSUPPORTED_IF(mode == WaitTypeAndCallbackMode::TimedWaitAny_Future || |
| mode == |
| WaitTypeAndCallbackMode::TimedWaitAny_FutureSpontaneous); |
| } |
| testInstance = GetInstance(); |
| testDevice = device; |
| testQueue = queue; |
| // Make sure these aren't used accidentally (unfortunately can't do the same for instance): |
| device = nullptr; |
| queue = nullptr; |
| } |
| |
| void UseSecondInstance() { |
| wgpu::InstanceDescriptor desc; |
| desc.features.timedWaitAnyEnable = !UsesWire(); |
| std::tie(testInstance, testDevice) = CreateExtraInstance(&desc); |
| testQueue = testDevice.GetQueue(); |
| } |
| |
| void LoseTestDevice() { |
| EXPECT_CALL(mDeviceLostCallback, |
| Call(WGPUDeviceLostReason_Undefined, testing::_, testing::_)) |
| .Times(1); |
| testDevice.ForceLoss(wgpu::DeviceLostReason::Undefined, "Device lost for testing"); |
| testDevice.Tick(); |
| } |
| |
| void TrivialSubmit() { |
| wgpu::CommandBuffer cb = testDevice.CreateCommandEncoder().Finish(); |
| testQueue.Submit(1, &cb); |
| } |
| |
| wgpu::CallbackMode GetCallbackMode() { |
| switch (GetParam().mWaitTypeAndCallbackMode) { |
| case WaitTypeAndCallbackMode::TimedWaitAny_Future: |
| case WaitTypeAndCallbackMode::SpinWaitAny_Future: |
| return wgpu::CallbackMode::Future; |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEvents: |
| return wgpu::CallbackMode::ProcessEvents; |
| case WaitTypeAndCallbackMode::TimedWaitAny_FutureSpontaneous: |
| case WaitTypeAndCallbackMode::SpinWaitAny_FutureSpontaneous: |
| return wgpu::CallbackMode::Future | wgpu::CallbackMode::Spontaneous; |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEventsSpontaneous: |
| return wgpu::CallbackMode::ProcessEvents | wgpu::CallbackMode::Spontaneous; |
| } |
| } |
| |
| bool IsSpontaneous() { return GetCallbackMode() & wgpu::CallbackMode::Spontaneous; } |
| |
| void TrackForTest(wgpu::Future future) { |
| mCallbacksIssuedCount++; |
| |
| switch (GetParam().mWaitTypeAndCallbackMode) { |
| case WaitTypeAndCallbackMode::TimedWaitAny_Future: |
| case WaitTypeAndCallbackMode::TimedWaitAny_FutureSpontaneous: |
| case WaitTypeAndCallbackMode::SpinWaitAny_Future: |
| case WaitTypeAndCallbackMode::SpinWaitAny_FutureSpontaneous: |
| mFutures.push_back(wgpu::FutureWaitInfo{future, false}); |
| break; |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEvents: |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEventsSpontaneous: |
| ASSERT_EQ(future.id, 0ull); |
| break; |
| } |
| } |
| |
| wgpu::Future OnSubmittedWorkDone(WGPUQueueWorkDoneStatus expectedStatus) { |
| struct Userdata { |
| EventCompletionTests* self; |
| WGPUQueueWorkDoneStatus expectedStatus; |
| }; |
| Userdata* userdata = new Userdata{this, expectedStatus}; |
| |
| return testQueue.OnSubmittedWorkDoneF({ |
| nullptr, |
| GetCallbackMode(), |
| [](WGPUQueueWorkDoneStatus status, void* userdata) { |
| Userdata* u = reinterpret_cast<Userdata*>(userdata); |
| u->self->mCallbacksCompletedCount++; |
| ASSERT_EQ(status, u->expectedStatus); |
| delete u; |
| }, |
| userdata, |
| }); |
| } |
| |
| void TestWaitAll(bool loopOnlyOnce = false) { |
| switch (GetParam().mWaitTypeAndCallbackMode) { |
| case WaitTypeAndCallbackMode::TimedWaitAny_Future: |
| case WaitTypeAndCallbackMode::TimedWaitAny_FutureSpontaneous: |
| return TestWaitImpl(WaitType::TimedWaitAny); |
| case WaitTypeAndCallbackMode::SpinWaitAny_Future: |
| case WaitTypeAndCallbackMode::SpinWaitAny_FutureSpontaneous: |
| return TestWaitImpl(WaitType::SpinWaitAny); |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEvents: |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEventsSpontaneous: |
| return TestWaitImpl(WaitType::SpinProcessEvents); |
| } |
| } |
| |
| void TestWaitIncorrectly() { |
| switch (GetParam().mWaitTypeAndCallbackMode) { |
| case WaitTypeAndCallbackMode::TimedWaitAny_Future: |
| case WaitTypeAndCallbackMode::TimedWaitAny_FutureSpontaneous: |
| case WaitTypeAndCallbackMode::SpinWaitAny_Future: |
| case WaitTypeAndCallbackMode::SpinWaitAny_FutureSpontaneous: |
| return TestWaitImpl(WaitType::SpinProcessEvents); |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEvents: |
| case WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEventsSpontaneous: |
| return TestWaitImpl(WaitType::SpinWaitAny); |
| } |
| } |
| |
| private: |
| void TestWaitImpl(WaitType waitType, bool loopOnlyOnce = false) { |
| uint64_t oldCompletedCount = mCallbacksCompletedCount; |
| |
| const auto start = std::chrono::high_resolution_clock::now(); |
| auto testTimeExceeded = [=]() -> bool { |
| return std::chrono::high_resolution_clock::now() - start > std::chrono::seconds(5); |
| }; |
| |
| switch (waitType) { |
| case WaitType::TimedWaitAny: { |
| bool emptyWait = mFutures.size() == 0; |
| // Loop at least once so we can test it with 0 futures. |
| do { |
| ASSERT_FALSE(testTimeExceeded()); |
| DAWN_ASSERT(!UsesWire()); |
| wgpu::WaitStatus status; |
| |
| uint64_t oldCompletionCount = mCallbacksCompletedCount; |
| // Any futures should succeed within a few milliseconds at most. |
| status = testInstance.WaitAny(mFutures.size(), mFutures.data(), UINT64_MAX); |
| ASSERT_EQ(status, wgpu::WaitStatus::Success); |
| bool mayHaveCompletedEarly = IsSpontaneous(); |
| if (!mayHaveCompletedEarly && !emptyWait) { |
| ASSERT_GT(mCallbacksCompletedCount, oldCompletionCount); |
| } |
| |
| // Verify this succeeds instantly because some futures completed already. |
| status = testInstance.WaitAny(mFutures.size(), mFutures.data(), 0); |
| ASSERT_EQ(status, wgpu::WaitStatus::Success); |
| |
| RemoveCompletedFutures(); |
| if (loopOnlyOnce) { |
| break; |
| } |
| } while (mFutures.size() > 0); |
| } break; |
| case WaitType::SpinWaitAny: { |
| bool emptyWait = mFutures.size() == 0; |
| // Loop at least once so we can test it with 0 futures. |
| do { |
| ASSERT_FALSE(testTimeExceeded()); |
| |
| uint64_t oldCompletionCount = mCallbacksCompletedCount; |
| FlushWire(); |
| testDevice.Tick(); |
| auto status = testInstance.WaitAny(mFutures.size(), mFutures.data(), 0); |
| if (status == wgpu::WaitStatus::TimedOut) { |
| continue; |
| } |
| ASSERT_TRUE(status == wgpu::WaitStatus::Success); |
| bool mayHaveCompletedEarly = IsSpontaneous(); |
| if (!mayHaveCompletedEarly && !emptyWait) { |
| ASSERT_GT(mCallbacksCompletedCount, oldCompletionCount); |
| } |
| |
| RemoveCompletedFutures(); |
| if (loopOnlyOnce) { |
| break; |
| } |
| } while (mFutures.size() > 0); |
| } break; |
| case WaitType::SpinProcessEvents: { |
| do { |
| ASSERT_FALSE(testTimeExceeded()); |
| |
| FlushWire(); |
| testDevice.Tick(); |
| testInstance.ProcessEvents(); |
| |
| if (loopOnlyOnce) { |
| break; |
| } |
| } while (mCallbacksCompletedCount < mCallbacksIssuedCount); |
| } break; |
| } |
| |
| if (!IsSpontaneous()) { |
| ASSERT_EQ(mCallbacksCompletedCount - oldCompletedCount, |
| mCallbacksIssuedCount - mCallbacksWaitedCount); |
| } |
| ASSERT_EQ(mCallbacksCompletedCount, mCallbacksIssuedCount); |
| mCallbacksWaitedCount = mCallbacksCompletedCount; |
| } |
| |
| void RemoveCompletedFutures() { |
| size_t oldSize = mFutures.size(); |
| if (oldSize > 0) { |
| mFutures.erase( |
| std::remove_if(mFutures.begin(), mFutures.end(), |
| [](const wgpu::FutureWaitInfo& info) { return info.completed; }), |
| mFutures.end()); |
| ASSERT_LT(mFutures.size(), oldSize); |
| } |
| } |
| }; |
| |
| // Wait when no events have been requested. |
| TEST_P(EventCompletionTests, NoEvents) { |
| TestWaitAll(); |
| } |
| |
| // WorkDone event after submitting some trivial work. |
| TEST_P(EventCompletionTests, WorkDoneSimple) { |
| TrivialSubmit(); |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TestWaitAll(); |
| } |
| |
| // WorkDone event before device loss, wait afterward. |
| TEST_P(EventCompletionTests, WorkDoneAcrossDeviceLoss) { |
| TrivialSubmit(); |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TestWaitAll(); |
| } |
| |
| // WorkDone event after device loss. |
| TEST_P(EventCompletionTests, WorkDoneAfterDeviceLoss) { |
| TrivialSubmit(); |
| LoseTestDevice(); |
| ASSERT_DEVICE_ERROR_ON(testDevice, |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success))); |
| TestWaitAll(); |
| } |
| |
| // WorkDone event twice after submitting some trivial work. |
| TEST_P(EventCompletionTests, WorkDoneTwice) { |
| TrivialSubmit(); |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TestWaitAll(); |
| } |
| |
| // WorkDone event without ever having submitted any work. |
| TEST_P(EventCompletionTests, WorkDoneNoWork) { |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TestWaitAll(); |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TestWaitAll(); |
| } |
| |
| // WorkDone event after all work has completed already. |
| TEST_P(EventCompletionTests, WorkDoneAlreadyCompleted) { |
| TrivialSubmit(); |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TestWaitAll(); |
| TrackForTest(OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success)); |
| TestWaitAll(); |
| } |
| |
| // WorkDone events waited in reverse order. |
| TEST_P(EventCompletionTests, WorkDoneOutOfOrder) { |
| // With ProcessEvents or Spontaneous we can't control the order of completion. |
| DAWN_TEST_UNSUPPORTED_IF(GetCallbackMode() & |
| (wgpu::CallbackMode::ProcessEvents | wgpu::CallbackMode::Spontaneous)); |
| |
| TrivialSubmit(); |
| wgpu::Future f1 = OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success); |
| TrivialSubmit(); |
| wgpu::Future f2 = OnSubmittedWorkDone(WGPUQueueWorkDoneStatus_Success); |
| |
| // When using WaitAny, normally callback ordering guarantees would guarantee f1 completes before |
| // f2. But if we wait on f2 first, then f2 is allowed to complete first because f1 still hasn't |
| // had an opportunity to complete. |
| TrackForTest(f2); |
| TestWaitAll(); |
| TrackForTest(f1); |
| TestWaitAll(true); |
| } |
| |
| constexpr WGPUQueueWorkDoneStatus kStatusUninitialized = |
| static_cast<WGPUQueueWorkDoneStatus>(INT32_MAX); |
| |
| TEST_P(EventCompletionTests, WorkDoneDropInstanceBeforeEvent) { |
| // TODO(crbug.com/dawn/1987): Wire does not implement instance destruction correctly yet. |
| DAWN_TEST_UNSUPPORTED_IF(UsesWire()); |
| |
| UseSecondInstance(); |
| testInstance = nullptr; // Drop the last external ref to the instance. |
| |
| WGPUQueueWorkDoneStatus status = kStatusUninitialized; |
| testQueue.OnSubmittedWorkDoneF({nullptr, GetCallbackMode(), |
| [](WGPUQueueWorkDoneStatus status, void* userdata) { |
| *reinterpret_cast<WGPUQueueWorkDoneStatus*>(userdata) = |
| status; |
| }, |
| &status}); |
| |
| // Callback should have been called immediately because we leaked it since there's no way to |
| // call WaitAny or ProcessEvents anymore. |
| // |
| // TODO(crbug.com/dawn/2059): Once Spontaneous is implemented, this should no longer expect the |
| // callback to be cleaned up immediately (and should expect it to happen on a future Tick). |
| ASSERT_EQ(status, WGPUQueueWorkDoneStatus_Unknown); |
| } |
| |
| TEST_P(EventCompletionTests, WorkDoneDropInstanceAfterEvent) { |
| // TODO(crbug.com/dawn/1987): Wire does not implement instance destruction correctly yet. |
| DAWN_TEST_UNSUPPORTED_IF(UsesWire()); |
| |
| UseSecondInstance(); |
| |
| WGPUQueueWorkDoneStatus status = kStatusUninitialized; |
| testQueue.OnSubmittedWorkDoneF({nullptr, GetCallbackMode(), |
| [](WGPUQueueWorkDoneStatus status, void* userdata) { |
| *reinterpret_cast<WGPUQueueWorkDoneStatus*>(userdata) = |
| status; |
| }, |
| &status}); |
| |
| ASSERT_EQ(status, kStatusUninitialized); |
| |
| testInstance = nullptr; // Drop the last external ref to the instance. |
| |
| // Callback should have been called immediately because we leaked it since there's no way to |
| // call WaitAny or ProcessEvents anymore. |
| // |
| // TODO(crbug.com/dawn/2059): Once Spontaneous is implemented, this should no longer expect the |
| // callback to be cleaned up immediately (and should expect it to happen on a future Tick). |
| ASSERT_EQ(status, WGPUQueueWorkDoneStatus_Unknown); |
| } |
| |
| // TODO(crbug.com/dawn/1987): |
| // - Test any reentrancy guarantees (for ProcessEvents or WaitAny inside a callback), |
| // to make sure things don't blow up and we don't attempt to hold locks recursively. |
| // - Other tests? |
| |
| DAWN_INSTANTIATE_TEST_P(EventCompletionTests, |
| // TODO(crbug.com/dawn/2058): Enable tests for the rest of the backends. |
| {MetalBackend()}, |
| { |
| WaitTypeAndCallbackMode::TimedWaitAny_Future, |
| WaitTypeAndCallbackMode::TimedWaitAny_FutureSpontaneous, |
| WaitTypeAndCallbackMode::SpinWaitAny_Future, |
| WaitTypeAndCallbackMode::SpinWaitAny_FutureSpontaneous, |
| WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEvents, |
| WaitTypeAndCallbackMode::SpinProcessEvents_ProcessEventsSpontaneous, |
| |
| // TODO(crbug.com/dawn/2059): The cases with the Spontaneous flag |
| // enabled were added before we implemented all of the spontaneous |
| // completions. They might accidentally be overly strict. |
| |
| // TODO(crbug.com/dawn/2059): Make guarantees that Spontaneous callbacks |
| // get called (as long as you're hitting "checkpoints"), and add the |
| // corresponding tests, for example: |
| // - SpinProcessEvents_Spontaneous, |
| // - SpinSubmit_Spontaneous, |
| // - SpinCheckpoint_Spontaneous (if wgpuDeviceCheckpoint is added). |
| // - (Note we don't want to guarantee Tick will process events - we |
| // could even test that it doesn't, if we make that true.) |
| }); |
| |
| // WaitAnyTests |
| |
| class WaitAnyTests : public DawnTest {}; |
| |
| TEST_P(WaitAnyTests, UnsupportedTimeout) { |
| wgpu::Instance instance2; |
| wgpu::Device device2; |
| |
| if (UsesWire()) { |
| // The wire (currently) never supports timedWaitAnyEnable, so we can run this test on the |
| // default instance/device. |
| instance2 = GetInstance(); |
| device2 = device; |
| } else { |
| // When not using the wire, DawnTest will unconditionally set timedWaitAnyEnable since it's |
| // useful for other tests. For this test, we need it to be false to test validation. |
| wgpu::InstanceDescriptor desc; |
| desc.features.timedWaitAnyEnable = false; |
| std::tie(instance2, device2) = CreateExtraInstance(&desc); |
| } |
| |
| // UnsupportedTimeout is still validated if no futures are passed. |
| for (uint64_t timeout : {uint64_t(1), uint64_t(0), UINT64_MAX}) { |
| ASSERT_EQ(instance2.WaitAny(0, nullptr, timeout), |
| timeout > 0 ? wgpu::WaitStatus::UnsupportedTimeout : wgpu::WaitStatus::Success); |
| } |
| |
| for (uint64_t timeout : {uint64_t(1), uint64_t(0), UINT64_MAX}) { |
| wgpu::FutureWaitInfo info{device2.GetQueue().OnSubmittedWorkDoneF( |
| {nullptr, wgpu::CallbackMode::Future, [](WGPUQueueWorkDoneStatus, void*) {}, nullptr})}; |
| wgpu::WaitStatus status = instance2.WaitAny(1, &info, timeout); |
| if (timeout == 0) { |
| ASSERT_TRUE(status == wgpu::WaitStatus::Success || |
| status == wgpu::WaitStatus::TimedOut); |
| } else { |
| ASSERT_EQ(status, wgpu::WaitStatus::UnsupportedTimeout); |
| } |
| } |
| } |
| |
| TEST_P(WaitAnyTests, UnsupportedCount) { |
| for (uint64_t timeout : {uint64_t(0), uint64_t(1)}) { |
| // We don't support values higher than the default (64), and if you ask for lower than 64 |
| // you still get 64. DawnTest doesn't request anything (so requests 0) so gets 64. |
| for (size_t count : {kTimedWaitAnyMaxCountDefault, kTimedWaitAnyMaxCountDefault + 1}) { |
| std::vector<wgpu::FutureWaitInfo> infos; |
| for (size_t i = 0; i < count; ++i) { |
| infos.push_back( |
| {queue.OnSubmittedWorkDoneF({nullptr, wgpu::CallbackMode::Future, |
| [](WGPUQueueWorkDoneStatus, void*) {}, nullptr})}); |
| } |
| wgpu::WaitStatus status = GetInstance().WaitAny(infos.size(), infos.data(), timeout); |
| if (timeout == 0) { |
| ASSERT_TRUE(status == wgpu::WaitStatus::Success || |
| status == wgpu::WaitStatus::TimedOut); |
| } else if (UsesWire()) { |
| // Wire doesn't support timeouts at all. |
| ASSERT_EQ(status, wgpu::WaitStatus::UnsupportedTimeout); |
| } else if (count <= 64) { |
| ASSERT_EQ(status, wgpu::WaitStatus::Success); |
| } else { |
| ASSERT_EQ(status, wgpu::WaitStatus::UnsupportedCount); |
| } |
| } |
| } |
| } |
| |
| TEST_P(WaitAnyTests, UnsupportedMixedSources) { |
| wgpu::Device device2 = CreateDevice(); |
| wgpu::Queue queue2 = device2.GetQueue(); |
| for (uint64_t timeout : {uint64_t(0), uint64_t(1)}) { |
| std::vector<wgpu::FutureWaitInfo> infos{{ |
| {queue.OnSubmittedWorkDoneF({nullptr, wgpu::CallbackMode::Future, |
| [](WGPUQueueWorkDoneStatus, void*) {}, nullptr})}, |
| {queue2.OnSubmittedWorkDoneF({nullptr, wgpu::CallbackMode::Future, |
| [](WGPUQueueWorkDoneStatus, void*) {}, nullptr})}, |
| }}; |
| wgpu::WaitStatus status = GetInstance().WaitAny(infos.size(), infos.data(), timeout); |
| if (timeout == 0) { |
| ASSERT_TRUE(status == wgpu::WaitStatus::Success || |
| status == wgpu::WaitStatus::TimedOut); |
| } else if (UsesWire()) { |
| // Wire doesn't support timeouts at all. |
| ASSERT_EQ(status, wgpu::WaitStatus::UnsupportedTimeout); |
| } else { |
| ASSERT_EQ(status, wgpu::WaitStatus::UnsupportedMixedSources); |
| } |
| } |
| } |
| |
| DAWN_INSTANTIATE_TEST(WaitAnyTests, |
| // TODO(crbug.com/dawn/2058): Enable tests for the rest of the backends. |
| MetalBackend()); |
| |
| } // anonymous namespace |
| } // namespace dawn |
