cubecl-cpp 0.11.0-pre.4

CPP transpiler for CubeCL
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

use crate::shared::Architecture;

pub use cubecl_core::ir::amd::{AMDArchitecture, AmdWmma};

impl Architecture for AMDArchitecture {
    fn warp_size(&self) -> u32 {
        // Zero for an architecture the table does not know, which the HIP runtime checks the
        // driver's own answer against before anything is compiled for it.
        self.plane_dim().unwrap_or(0)
    }

    fn is_wmma_capable(&self) -> bool {
        matches!(
            self,
            AMDArchitecture::GFX103 | AMDArchitecture::GFX11 | AMDArchitecture::GFX12
        )
    }

    fn is_mfma_capable(&self) -> bool {
        matches!(
            self,
            AMDArchitecture::GFX908 | AMDArchitecture::GFX90A | AMDArchitecture::GFX94
        )
    }
}