commit 8872710e27f2f93fe5df11306645c15427a7c06d
author Andrew Woloszyn <andrew.woloszyn@gmail.com> Fri Sep 27 10:33:05 2024 -0400
committer GitHub <noreply@github.com> Fri Sep 27 10:33:05 2024 -0400
tree aea0a23463d28e1464a6550563c2678d8aad9c59
parent 14728a7732939b92f664b8163dce4a9fb1ef6126

Revert "Removed the iree_thread_join in the cleanup of deferred_work_queue.c" (#18616)

Reverts iree-org/iree#18605

runtime/src/iree/hal/utils/deferred_work_queue.c

diff --git a/runtime/src/iree/hal/utils/deferred_work_queue.c b/runtime/src/iree/hal/utils/deferred_work_queue.c
index a0214b5..22c229b 100644
--- a/runtime/src/iree/hal/utils/deferred_work_queue.c
+++ b/runtime/src/iree/hal/utils/deferred_work_queue.c
@@ -598,7 +598,10 @@
   // 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(