Struct Fragment 

pub struct Fragment {
    pub data: Vec<f32>,
    pub rows: usize,
    pub cols: usize,
}

Matrix fragment — a tile of a matrix stored in registers. On GPU, these map to warp-distributed register fragments.

Fields

data: Vec<f32>

Data stored as f32 (accumulator format).

rows: usize

Number of rows.

cols: usize

Number of columns.

Implementations

impl Fragment

pub fn zeros(rows: usize, cols: usize) -> Self

Create a zero-initialized fragment.

pub fn from_f32(data: &[f32], rows: usize, cols: usize) -> Result<Self>

Create from f32 data.

pub fn from_half(data: &[Half], rows: usize, cols: usize) -> Result<Self>

Load from fp16 data (converting to f32 accumulator format).

pub fn from_bf16(data: &[BFloat16], rows: usize, cols: usize) -> Result<Self>

Load from bf16 data.

pub fn get(&self, row: usize, col: usize) -> f32

Get element at (row, col).

pub fn set(&mut self, row: usize, col: usize, val: f32)

Set element at (row, col).

pub fn to_half(&self) -> Vec<Half>

Store to fp16.

pub fn to_bf16(&self) -> Vec<BFloat16>

Store to bf16.

Trait Implementations

impl Clone for Fragment

fn clone(&self) -> Fragment

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Fragment

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

Formats the value using the given formatter.