cubek-std 0.3.0-pre.1

CubeK: Standard Library
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14

use cubecl::prelude::VectorSize;
use cubek_test_utils::CatalogEntry;

pub struct UnaryStrategy {
    pub vectorization: VectorSize,
}

pub fn strategies() -> Vec<CatalogEntry<UnaryStrategy>> {
    vec![
        CatalogEntry::new("vec1", "Vec1", UnaryStrategy { vectorization: 1 }),
        CatalogEntry::new("vec4", "Vec4", UnaryStrategy { vectorization: 4 }),
        CatalogEntry::new("vec8", "Vec8", UnaryStrategy { vectorization: 8 }),
    ]
}