pub struct PipelineExecutorN { /* private fields */ }Expand description
N-stage heterogeneous executor.
Implementations§
Source§impl PipelineExecutorN
impl PipelineExecutorN
pub fn new() -> Self
pub fn push<S: PipelineStage>(self, stage: S) -> Self
Sourcepub async fn run<I, O>(
&mut self,
streams: &[Arc<CudaStream>],
completion: &Arc<dyn CompletionStrategy>,
input: I,
) -> Result<O, GpuError>
pub async fn run<I, O>( &mut self, streams: &[Arc<CudaStream>], completion: &Arc<dyn CompletionStrategy>, input: I, ) -> Result<O, GpuError>
Run the chain across streams (one per stage). On success
returns the tail stage’s output. On any stage failure, the
error short-circuits the chain.
pub fn stage_count(&self) -> usize
Trait Implementations§
Auto Trait Implementations§
impl Freeze for PipelineExecutorN
impl !RefUnwindSafe for PipelineExecutorN
impl Send for PipelineExecutorN
impl !Sync for PipelineExecutorN
impl Unpin for PipelineExecutorN
impl UnsafeUnpin for PipelineExecutorN
impl !UnwindSafe for PipelineExecutorN
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more