cubecl_matmul/components/global/read/reader/
fill_reader.rs

1use std::marker::PhantomData;
2
3use cubecl_core as cubecl;
4use cubecl_core::{CubeType, prelude::*};
5
6use crate::components::{MatrixPrecision, stage::FilledStage};
7
8#[derive(CubeType)]
9/// Accumulator reader that zeros the accumulator
10pub struct ZeroGlobalReader<IP: MatrixPrecision> {
11    #[cube(comptime)]
12    _ty: PhantomData<IP>,
13}
14
15#[cube]
16impl<IP: MatrixPrecision> ZeroGlobalReader<IP> {
17    #[allow(clippy::new_without_default)]
18    pub fn new() -> Self {
19        ZeroGlobalReader::<IP> { _ty: PhantomData }
20    }
21
22    /// Give a reader to the loaded data.
23    pub fn stage(&self) -> FilledStage<IP::Stage> {
24        FilledStage::new(IP::Stage::from_int(0))
25    }
26}