Add task_id for DTask

jpsamaroo · jpsamaroo · commit 3638866714f3 · 2024-12-09T05:33:58.000-06:00
diff --git a/src/sch/eager.jl b/src/sch/eager.jl
@@ -134,3 +134,6 @@ function _find_thunk(e::Dagger.DTask)
         unwrap_weak_checked(EAGER_STATE[].thunk_dict[tid])
     end
 end
+Dagger.task_id(t::Dagger.DTask) = lock(EAGER_ID_MAP) do id_map
+    id_map[t.uid]
+end
