expose lamellar_task to users + sync/send impls

pnnl · Dec 11, 2024 · e7b8d87 · e7b8d87
1 parent 70c3b4c
commit e7b8d87
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/src/active_messaging/prelude.rs b/src/active_messaging/prelude.rs
@@ -8,16 +8,14 @@ pub use crate::active_messaging::{
     am, local_am, typed_am_group, ActiveMessaging, AmData, AmDist, AmGroupData, AmHandle,
     AmLocalData, LamellarAM, LamellarSerde, RemoteActiveMessage, Serde,
 };
-
 pub use crate::async_trait;
-// pub use crate::futures_util::StreamExt;
 pub use crate::inventory;
 pub use crate::lamellar_arch::*;
-pub use crate::lamellar_team::LamellarTeam;
-//#[doc(hidden)]
 pub use crate::lamellar_team::IntoLamellarTeam;
+pub use crate::lamellar_team::LamellarTeam;
 pub use crate::lamellar_world::LamellarWorld;
 pub use crate::lamellar_world::LamellarWorldBuilder;
+pub use crate::scheduler::LamellarTask;
 pub use crate::LamellarEnv;
 pub use crate::LamellarTaskGroup;
 pub use crate::{

diff --git a/src/lib.rs b/src/lib.rs
@@ -287,6 +287,7 @@ pub use crate::lamellar_team::ArcLamellarTeam;
 pub(crate) use crate::lamellar_team::LamellarTeamRT;
 pub use crate::lamellar_world::*;
 pub use crate::scheduler::ExecutorType;
+pub use crate::scheduler::LamellarTask;
 
 extern crate lamellar_impl;
 // //#[doc(hidden)]

diff --git a/src/scheduler.rs b/src/scheduler.rs
@@ -102,6 +102,9 @@ pub struct LamellarTask<T> {
     executor: Arc<Executor>,
 }
 
+unsafe impl<T: Send> Send for LamellarTask<T> {}
+unsafe impl<T: Sync> Sync for LamellarTask<T> {}
+
 impl<T> LamellarTask<T> {
     pub fn block(self) -> T {
         RuntimeWarning::BlockingCall("LamellarTask::block", "<task>.await").print();
@@ -124,6 +127,9 @@ pub(crate) enum LamellarTaskInner<T> {
     TokioTask(tokio::task::JoinHandle<T>),
 }
 
+unsafe impl<T: Send> Send for LamellarTaskInner<T> {}
+unsafe impl<T: Sync> Sync for LamellarTaskInner<T> {}
+
 impl<T> Drop for LamellarTaskInner<T> {
     fn drop(self: &mut Self) {
         // let mut dropped = LamellarTaskInner::Dropped;