pub struct GpuGraph { /* private fields */ }Expand description
A captured CUDA graph that can be replayed with minimal dispatch overhead.
Created via gpu_graph_capture or manually with new() + capture_begin()
/ capture_end().
Implementations§
Source§impl GpuGraph
impl GpuGraph
Sourcepub fn capture_begin(
&mut self,
pool: Option<MemPoolId>,
mode: CaptureMode,
) -> Result<()>
pub fn capture_begin( &mut self, pool: Option<MemPoolId>, mode: CaptureMode, ) -> Result<()>
Begin capturing GPU operations into this graph.
All CUDA kernel launches on the current stream after this call
will be recorded until capture_end.
Sourcepub fn capture_end(&mut self) -> Result<()>
pub fn capture_end(&mut self) -> Result<()>
End capture and finalize the graph.
Trait Implementations§
Auto Trait Implementations§
impl !Send for GpuGraph
impl !Sync for GpuGraph
impl Freeze for GpuGraph
impl RefUnwindSafe for GpuGraph
impl Unpin for GpuGraph
impl UnsafeUnpin for GpuGraph
impl UnwindSafe for GpuGraph
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