pub struct CompiledNode {
pub ptx_path: String,
pub entry_point: String,
pub output_shape: Shape,
pub output_dtype: DtypeRepr,
pub runtime_op: Option<Arc<dyn RuntimeOp>>,
pub backward_ptx_path: Option<String>,
pub backward_entry_point: String,
}Expand description
One PTX-compiled graph node: where its compiled kernel lives on disk, its output shape/dtype, and how to dispatch it at runtime.
Fields§
§ptx_path: StringPath to the compiled .o PTX file. Empty for Input placeholder nodes.
entry_point: StringThe kernel’s entry point symbol name.
output_shape: ShapeThis node’s output shape.
output_dtype: DtypeReprThis node’s output dtype.
runtime_op: Option<Arc<dyn RuntimeOp>>Runtime dispatch: arg-packing + grid computation. None for Input nodes.
backward_ptx_path: Option<String>Path to compiled backward PTX. None if no backward kernel for this op.
backward_entry_point: StringEntry point name for the backward kernel.
Auto Trait Implementations§
impl Freeze for CompiledNode
impl !RefUnwindSafe for CompiledNode
impl Send for CompiledNode
impl Sync for CompiledNode
impl Unpin for CompiledNode
impl UnsafeUnpin for CompiledNode
impl !UnwindSafe for CompiledNode
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