Fixing flake-y host call CTS test.
diff --git a/runtime/src/iree/hal/cts/queue_host_call_test.h b/runtime/src/iree/hal/cts/queue_host_call_test.h
index fda974c..84527c1 100644
--- a/runtime/src/iree/hal/cts/queue_host_call_test.h
+++ b/runtime/src/iree/hal/cts/queue_host_call_test.h
@@ -367,6 +367,11 @@
IREE_HAL_WAIT_FLAG_DEFAULT)),
StatusIs(StatusCode::kAborted));
+ // NOTE: it's not guaranteed that all semaphores are updated atomically so
+ // it's possible for the wait to return and at least one of the semaphores to
+ // not be ready yet. We give a little time for the assignment to propagate.
+ std::this_thread::sleep_for(std::chrono::milliseconds(50));
+
// Query individual semaphores to verify they're in error state.
uint64_t value0 = 0;
EXPECT_THAT(Status(iree_hal_semaphore_query(
@@ -436,6 +441,11 @@
EXPECT_EQ(state.did_call, 1);
EXPECT_TRUE(state.wait_completed);
+ // NOTE: it's not guaranteed that all semaphores are updated atomically so
+ // it's possible for the wait to return and at least one of the semaphores to
+ // not be ready yet. We give a little time for the assignment to propagate.
+ std::this_thread::sleep_for(std::chrono::milliseconds(50));
+
// Query individual semaphores to verify they're in error state.
uint64_t value0 = 0;
EXPECT_THAT(Status(iree_hal_semaphore_query(
