monarch_tensor_worker::stream

Trait StreamMessageClient

Source 
pub trait StreamMessageClient: Send + Sync {

Show 23 methods    // Required methods
    fn call_function<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        arg0: CallFunctionParams,
        arg1: HashMap<Ref, DeviceMesh>,
        arg2: HashMap<Ref, (DeviceMesh, Vec<String>, Arc<ActorHandle<NcclCommActor>>)>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn borrow_create<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        borrow: u64,
        tensor: Ref,
        first_use_sender: PortHandle<(Option<Event>, TensorCellResult)>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn borrow_first_use<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        borrow: u64,
        result: Ref,
        first_use_receiver: Arc<Mutex<PortReceiver<(Option<Event>, TensorCellResult)>>>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn borrow_last_use<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        borrow: u64,
        result: Ref,
        last_use_sender: PortHandle<(Option<Event>, TensorCellResult)>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn borrow_drop<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        borrow: u64,
        last_use_receiver: Arc<Mutex<PortReceiver<(Option<Event>, TensorCellResult)>>>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn delete_refs<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        arg0: Vec<Ref>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn request_status<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort),
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn init_comm<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        arg0: ActorHandle<NcclCommActor>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn reduce<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        comm: Arc<ActorHandle<NcclCommActor>>,
        dim_size: i64,
        result: Ref,
        local_tensor: Ref,
        factory: Factory,
        reduction: Reduction,
        scatter: bool,
        in_place: bool,
        out: Option<Ref>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn send_tensor<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        result: Ref,
        from_rank: Option<usize>,
        to_rank: Option<usize>,
        tensor: Ref,
        factory: Factory,
        comm: Arc<ActorHandle<NcclCommActor>>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn send_value<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        seq: Seq,
        worker_actor_id: ActorId,
        mutates: Vec<Ref>,
        function: Option<ResolvableFunction>,
        args: Vec<WireValue>,
        kwargs: HashMap<String, WireValue>,
        device_meshes: HashMap<Ref, DeviceMesh>,
        pipe: Option<PortHandle<PipeMessage>>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn set_value<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        seq: Seq,
        results: Vec<Option<Ref>>,
        pipe: PortHandle<PipeMessage>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn define_recording<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        recording: Ref,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn finalize_recording<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        recording: Ref,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn call_recording<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        seq: Seq,
        recording: Ref,
        results: Vec<Ref>,
        actuals: Vec<Ref>,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn recording_formal<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        result: Ref,
        argument_index: usize,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn recording_result<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        result: Ref,
        output_index: usize,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn set_ref_unit_tests_only<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        arg0: Ref,
        arg1: WireValue,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn set_tensor_ref_unit_tests_only<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        arg0: Ref,
        arg1: TensorCellResult,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn get_ref_unit_tests_only<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort),
        arg0: Ref,
    ) -> Pin<Box<dyn Future<Output = Result<Option<Result<WireValue, String>>, Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn get_tensor_ref_unit_tests_only<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort),
        arg0: Ref,
    ) -> Pin<Box<dyn Future<Output = Result<Option<TensorCellResult>, Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn send_result_of_actor_call<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        arg0: ActorId,
        arg1: ActorCallParams,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;
    fn call_actor_method<'life0, 'life1, 'async_trait>(
        &'life0 self,
        caps: &'life1 (impl 'async_trait + CanSend),
        arg0: ActorMethodParams,
    ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
       where Self: 'async_trait,
             'life0: 'async_trait,
             'life1: 'async_trait;

}
Expand description

The custom client trait for this message type.

Required Methods§

Source

fn call_function<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: CallFunctionParams, arg1: HashMap<Ref, DeviceMesh>, arg2: HashMap<Ref, (DeviceMesh, Vec<String>, Arc<ActorHandle<NcclCommActor>>)>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn borrow_create<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), borrow: u64, tensor: Ref, first_use_sender: PortHandle<(Option<Event>, TensorCellResult)>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn borrow_first_use<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), borrow: u64, result: Ref, first_use_receiver: Arc<Mutex<PortReceiver<(Option<Event>, TensorCellResult)>>>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn borrow_last_use<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), borrow: u64, result: Ref, last_use_sender: PortHandle<(Option<Event>, TensorCellResult)>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn borrow_drop<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), borrow: u64, last_use_receiver: Arc<Mutex<PortReceiver<(Option<Event>, TensorCellResult)>>>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn delete_refs<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: Vec<Ref>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn request_status<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort), ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn init_comm<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: ActorHandle<NcclCommActor>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn reduce<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), comm: Arc<ActorHandle<NcclCommActor>>, dim_size: i64, result: Ref, local_tensor: Ref, factory: Factory, reduction: Reduction, scatter: bool, in_place: bool, out: Option<Ref>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn send_tensor<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), result: Ref, from_rank: Option<usize>, to_rank: Option<usize>, tensor: Ref, factory: Factory, comm: Arc<ActorHandle<NcclCommActor>>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn send_value<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), seq: Seq, worker_actor_id: ActorId, mutates: Vec<Ref>, function: Option<ResolvableFunction>, args: Vec<WireValue>, kwargs: HashMap<String, WireValue>, device_meshes: HashMap<Ref, DeviceMesh>, pipe: Option<PortHandle<PipeMessage>>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn set_value<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), seq: Seq, results: Vec<Option<Ref>>, pipe: PortHandle<PipeMessage>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn define_recording<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), recording: Ref, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn finalize_recording<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), recording: Ref, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn call_recording<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), seq: Seq, recording: Ref, results: Vec<Ref>, actuals: Vec<Ref>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn recording_formal<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), result: Ref, argument_index: usize, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn recording_result<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), result: Ref, output_index: usize, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn set_ref_unit_tests_only<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: Ref, arg1: WireValue, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn set_tensor_ref_unit_tests_only<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: Ref, arg1: TensorCellResult, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn get_ref_unit_tests_only<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort), arg0: Ref, ) -> Pin<Box<dyn Future<Output = Result<Option<Result<WireValue, String>>, Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn get_tensor_ref_unit_tests_only<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort), arg0: Ref, ) -> Pin<Box<dyn Future<Output = Result<Option<TensorCellResult>, Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn send_result_of_actor_call<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: ActorId, arg1: ActorCallParams, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Source

fn call_actor_method<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: ActorMethodParams, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

The generated client method for this enum variant.

Dyn Compatibility§

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementations on Foreign Types§

Source§

impl<A> StreamMessageClient for ActorHandle<A>
where A: Handler<StreamMessage>,

Source§

fn call_function<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: CallFunctionParams, arg1: HashMap<Ref, DeviceMesh>, arg2: HashMap<Ref, (DeviceMesh, Vec<String>, Arc<ActorHandle<NcclCommActor>>)>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn borrow_create<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), borrow: u64, tensor: Ref, first_use_sender: PortHandle<(Option<Event>, TensorCellResult)>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn borrow_first_use<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), borrow: u64, result: Ref, first_use_receiver: Arc<Mutex<PortReceiver<(Option<Event>, TensorCellResult)>>>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn borrow_last_use<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), borrow: u64, result: Ref, last_use_sender: PortHandle<(Option<Event>, TensorCellResult)>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn borrow_drop<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), borrow: u64, last_use_receiver: Arc<Mutex<PortReceiver<(Option<Event>, TensorCellResult)>>>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn delete_refs<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: Vec<Ref>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn request_status<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort), ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn init_comm<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: ActorHandle<NcclCommActor>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn reduce<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), comm: Arc<ActorHandle<NcclCommActor>>, dim_size: i64, result: Ref, local_tensor: Ref, factory: Factory, reduction: Reduction, scatter: bool, in_place: bool, out: Option<Ref>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn send_tensor<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), result: Ref, from_rank: Option<usize>, to_rank: Option<usize>, tensor: Ref, factory: Factory, comm: Arc<ActorHandle<NcclCommActor>>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn send_value<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), seq: Seq, worker_actor_id: ActorId, mutates: Vec<Ref>, function: Option<ResolvableFunction>, args: Vec<WireValue>, kwargs: HashMap<String, WireValue>, device_meshes: HashMap<Ref, DeviceMesh>, pipe: Option<PortHandle<PipeMessage>>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn set_value<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), seq: Seq, results: Vec<Option<Ref>>, pipe: PortHandle<PipeMessage>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn define_recording<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), recording: Ref, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn finalize_recording<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), recording: Ref, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn call_recording<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), seq: Seq, recording: Ref, results: Vec<Ref>, actuals: Vec<Ref>, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn recording_formal<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), result: Ref, argument_index: usize, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn recording_result<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), result: Ref, output_index: usize, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn set_ref_unit_tests_only<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: Ref, arg1: WireValue, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn set_tensor_ref_unit_tests_only<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: Ref, arg1: TensorCellResult, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn get_ref_unit_tests_only<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort), arg0: Ref, ) -> Pin<Box<dyn Future<Output = Result<Option<Result<WireValue, String>>, Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn get_tensor_ref_unit_tests_only<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend + CanOpenPort), arg0: Ref, ) -> Pin<Box<dyn Future<Output = Result<Option<TensorCellResult>, Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn send_result_of_actor_call<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: ActorId, arg1: ActorCallParams, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Source§

fn call_actor_method<'life0, 'life1, 'async_trait>( &'life0 self, caps: &'life1 (impl 'async_trait + CanSend), arg0: ActorMethodParams, ) -> Pin<Box<dyn Future<Output = Result<(), Error>> + Send + 'async_trait>>
where Self: 'async_trait, 'life0: 'async_trait, 'life1: 'async_trait,

Implementors§