Struct pjrt_sys::protos::stream_executor::dnn::AlgorithmProto

pub struct AlgorithmProto {
    pub algo_id: i64,
    pub math_type: i32,
    pub tuning_knobs: HashMap<i64, i64>,
    pub is_cudnn_frontend: bool,
    pub workspace_size: Option<u64>,
}

Generic algorithm representation.

Fields

algo_id: i64

math_type: i32

tuning_knobs: HashMap<i64, i64>

is_cudnn_frontend: bool

Legacy algorithm enums and cuDNN Frontend engine numbers need to coexist in the same proto medium-term, until we can be confident of no longer needing the legacy cuDNN convolution API. Once the migration is complete, we can stop producing legacy algorithm enums and remove this field.

workspace_size: Option<u64>

For ROCm only, it’s impossible to re-query the required workspace size after running the algorithm search, so we must store the workspace size along with the choice of algorithm. For consistency and convenience, cuDNN uses this field in the same way, even though it would be possible to re-query the workspace size from cuDNN at each use.

Since this message is persisted in files, we need to be able to distinguish 0 workspace size from unknown workspace size in an old message, so this is a message field.

Implementations

impl AlgorithmProto

pub fn math_type(&self) -> MathType

Returns the enum value of math_type, or the default if the field is set to an invalid enum value.

pub fn set_math_type(&mut self, value: MathType)

Sets math_type to the provided enum value.

Trait Implementations

impl Clone for AlgorithmProto

fn clone(&self) -> AlgorithmProto

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for AlgorithmProto

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for AlgorithmProto

fn default() -> Self

Returns the “default value” for a type.

impl Message for AlgorithmProto

fn encoded_len(&self) -> usize

Returns the encoded length of the message without a length delimiter.

fn clear(&mut self)

Clears the message, resetting all fields to their default.

fn encode(&self, buf: &mut impl BufMut) -> Result<(), EncodeError>

where Self: Sized,

Encodes the message to a buffer.

fn encode_to_vec(&self) -> Vec<u8>

where Self: Sized,

Encodes the message to a newly allocated buffer.

fn encode_length_delimited( &self, buf: &mut impl BufMut, ) -> Result<(), EncodeError>

where Self: Sized,

Encodes the message with a length-delimiter to a buffer.

fn encode_length_delimited_to_vec(&self) -> Vec<u8>

where Self: Sized,

Encodes the message with a length-delimiter to a newly allocated buffer.

fn decode(buf: impl Buf) -> Result<Self, DecodeError>

where Self: Default,

Decodes an instance of the message from a buffer.

fn decode_length_delimited(buf: impl Buf) -> Result<Self, DecodeError>

where Self: Default,

Decodes a length-delimited instance of the message from the buffer.

fn merge(&mut self, buf: impl Buf) -> Result<(), DecodeError>

where Self: Sized,

Decodes an instance of the message from a buffer, and merges it into self.

fn merge_length_delimited(&mut self, buf: impl Buf) -> Result<(), DecodeError>

where Self: Sized,

Decodes a length-delimited instance of the message from buffer, and merges it into self.

impl PartialEq for AlgorithmProto

fn eq(&self, other: &AlgorithmProto) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for AlgorithmProto