pub struct Team { /* private fields */ }Expand description
The process-wide team. Exists only when FTTS_INT8_THREADS requests more than one thread.
Implementations§
Source§impl Team
impl Team
Sourcepub fn linear_q8(
&self,
x_q: &[i8],
x_scales: &[f32],
weight: &QuantizedMatrix,
bias: Option<&[f32]>,
m: usize,
out: &mut [f32],
tier: Int8Tier,
)
pub fn linear_q8( &self, x_q: &[i8], x_scales: &[f32], weight: &QuantizedMatrix, bias: Option<&[f32]>, m: usize, out: &mut [f32], tier: Int8Tier, )
Runs one W8A8 linear across the team. Bit-identical to the serial path per element.
§Panics
Panics on shape mismatches, exactly as the serial kernel does.
Sourcepub fn gqa_attention(
&self,
queries: &[f32],
keys: &[f32],
values: &[f32],
mask: &[f32],
query_positions: usize,
key_positions: usize,
q_heads: usize,
kv_heads: usize,
head_dim: usize,
out: &mut [f32],
)
pub fn gqa_attention( &self, queries: &[f32], keys: &[f32], values: &[f32], mask: &[f32], query_positions: usize, key_positions: usize, q_heads: usize, kv_heads: usize, head_dim: usize, out: &mut [f32], )
Runs the default-arithmetic GQA attention across the team, partitioned over query
heads. Bit-identical to the serial f32ref::gqa_attention (same extracted loop).
§Panics
Panics on shape mismatches, exactly as the serial reference does.
Auto Trait Implementations§
impl !Freeze for Team
impl RefUnwindSafe for Team
impl Send for Team
impl Sync for Team
impl Unpin for Team
impl UnsafeUnpin for Team
impl UnwindSafe for Team
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