vyre 0.4.0

GPU compute intermediate representation with a standard operation library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13

//! Shared `DataType` shape constants and size limits for data movement ops.

use crate::ir::DataType;

pub(crate) const BYTES_BYTES_INPUTS: &[DataType] = &[DataType::Bytes, DataType::Bytes];
pub(crate) const BYTES_U32_INPUTS: &[DataType] = &[DataType::Bytes, DataType::U32];
pub(crate) const U32_U32_INPUTS: &[DataType] = &[DataType::U32, DataType::U32];

pub(crate) const BYTES_OUTPUTS: &[DataType] = &[DataType::Bytes];
pub(crate) const TWO_BYTES_OUTPUTS: &[DataType] = &[DataType::Bytes, DataType::Bytes];

/// Maximum byte output produced by a single data movement CPU reference call.
pub const MAX_OUTPUT_BYTES: usize = 64 * 1024 * 1024;