wait for task group to complete before returning from forEach

Liam Keeganrequested to merge
fix_tsan_error into feature/dune-assembler/main-multithreaded

Merge request reports