Skip to main content

Module cudnn

Module cudnn 

Source
Expand description

GraphOpRecord impls for crate::kernel::CudnnActor requests.

Each op holds the same payload as the matching CudnnMsg variant minus the reply channel, plus the cuDNN handle that the actor ordinarily owns. The graph script caller passes the handle in once, then submits a series of ConvForwardOps without reaching back into the actor.

For Phase 3 we keep these structs descriptor-shape-only: their record method validates the inputs and returns Unrecoverable on hosts where cuDNN isn’t loadable. A future sub-phase will wire the actual cuDNN-record path (cuDNN supports stream-capture as of v8.5).

Structs§

ActivationOp
Capture-mode op for CudnnMsg::Activation.
ConvForwardOp
Capture-mode op for CudnnMsg::ConvForward.
SoftmaxOp
Capture-mode op for CudnnMsg::Softmax.