#[repr(i32)]pub enum CudaGraphStrategy {
kDISABLED = 0,
kWHOLE_GRAPH_CAPTURE = 1,
}Expand description
CudaGraphStrategy
Strategies available for CUDA graphs optimizations for JIT (Just-In-Time) inference.
See [IRuntimeConfig::setCudaGraphStrategy()], IRuntimeConfig::getCudaGraphStrategy()
Variants§
kDISABLED = 0
Do not use CUDA graphs optimizations for inference.
kWHOLE_GRAPH_CAPTURE = 1
Capture CUDA graphs for the entire model execution as a single graph. This strategy captures the complete computational graph of the model and executes it atomically on the GPU stream. For dynamic shapes inference, this strategy captures the CUDA graph after the shape-specialized kernels are compiled for a given shape.
Trait Implementations§
Source§impl Clone for CudaGraphStrategy
impl Clone for CudaGraphStrategy
Source§fn clone(&self) -> CudaGraphStrategy
fn clone(&self) -> CudaGraphStrategy
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl ExternType for CudaGraphStrategy
impl ExternType for CudaGraphStrategy
Source§impl From<CudaGraphStrategy> for CudaGraphStrategy
impl From<CudaGraphStrategy> for CudaGraphStrategy
Source§fn from(value: CudaGraphStrategy) -> Self
fn from(value: CudaGraphStrategy) -> Self
Converts to this type from the input type.
Source§impl Hash for CudaGraphStrategy
impl Hash for CudaGraphStrategy
Source§impl Into<CudaGraphStrategy> for CudaGraphStrategy
impl Into<CudaGraphStrategy> for CudaGraphStrategy
Source§fn into(self) -> CudaGraphStrategy
fn into(self) -> CudaGraphStrategy
Converts this type into the (usually inferred) input type.
Source§impl PartialEq for CudaGraphStrategy
impl PartialEq for CudaGraphStrategy
Source§fn eq(&self, other: &CudaGraphStrategy) -> bool
fn eq(&self, other: &CudaGraphStrategy) -> bool
Tests for
self and other values to be equal, and is used by ==.impl Eq for CudaGraphStrategy
impl StructuralPartialEq for CudaGraphStrategy
impl UniquePtrTarget for CudaGraphStrategy
impl VectorElement for CudaGraphStrategy
impl WeakPtrTarget for CudaGraphStrategy
Auto Trait Implementations§
impl Freeze for CudaGraphStrategy
impl RefUnwindSafe for CudaGraphStrategy
impl Send for CudaGraphStrategy
impl Sync for CudaGraphStrategy
impl Unpin for CudaGraphStrategy
impl UnsafeUnpin for CudaGraphStrategy
impl UnwindSafe for CudaGraphStrategy
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