From 1620b831900baca06ecd5856d17a4fc36ed0a149 Mon Sep 17 00:00:00 2001
From: Julian P Samaroo <jpsamaroo@jpsamaroo.me>
Date: Sun, 23 Jun 2024 13:20:09 -0500
Subject: [PATCH] Sch: Halt on Julia exit

---
 src/sch/Sch.jl | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/sch/Sch.jl b/src/sch/Sch.jl
index cfb680bbd..54c8eab96 100644
--- a/src/sch/Sch.jl
+++ b/src/sch/Sch.jl
@@ -515,6 +515,11 @@ function scheduler_init(ctx, state::ComputeState, d::Thunk, options, deps)
         end
     end
 
+    # Halt scheduler on Julia exit
+    atexit() do
+        notify(state.halt)
+    end
+
     # Listen for new workers
     @async begin
         try