Revert "Removed the iree_thread_join in the cleanup of deferred_work_…

…queue.c" (iree-org#18616) Reverts iree-org#18605
nod-ai · Sep 27, 2024 · 8872710 · 8872710
1 parent 14728a7
commit 8872710
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/runtime/src/iree/hal/utils/deferred_work_queue.c b/runtime/src/iree/hal/utils/deferred_work_queue.c
@@ -598,7 +598,10 @@ void iree_hal_deferred_work_queue_destroy(
   // Request the workers to exit.
   iree_hal_deferred_work_queue_request_exit(work_queue);
 
+  iree_thread_join(work_queue->worker_thread);
   iree_thread_release(work_queue->worker_thread);
+
+  iree_thread_join(work_queue->completion_thread);
   iree_thread_release(work_queue->completion_thread);
 
   iree_hal_deferred_work_queue_working_area_deinitialize(