cubek-reduce 0.3.0-pre.2

CubeK: Reduce Kernels
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

use cubecl::config::autotune::AutotuneLevel;
use cubek_test_utils::CatalogEntry;

use crate::launch::{ReduceStrategy, RoutineStrategy, VectorizationStrategy};
use crate::routines::{
    BlueprintStrategy, cube::CubeStrategy, plane::PlaneStrategy, unit::UnitStrategy,
};

pub fn strategies() -> Vec<CatalogEntry<ReduceStrategy>> {
    let unit = || RoutineStrategy::Unit(BlueprintStrategy::Inferred(UnitStrategy));
    let plane = || {
        RoutineStrategy::Plane(BlueprintStrategy::Inferred(PlaneStrategy {
            independent: true,
        }))
    };
    let cube = || {
        RoutineStrategy::Cube(BlueprintStrategy::Inferred(CubeStrategy {
            use_planes: true,
        }))
    };
    let serial = VectorizationStrategy {
        parallel_output_vectorization: false,
    };
    let parallel = VectorizationStrategy {
        parallel_output_vectorization: true,
    };
    vec![
        CatalogEntry::new(
            "unit_serial",
            "Unit (serial)",
            ReduceStrategy {
                autotune_level: AutotuneLevel::Full,
                routine: unit(),
                vectorization: serial,
            },
        ),
        CatalogEntry::new(
            "unit_parallel",
            "Unit (parallel)",
            ReduceStrategy {
                autotune_level: AutotuneLevel::Full,
                routine: unit(),
                vectorization: parallel,
            },
        ),
        CatalogEntry::new(
            "plane_serial",
            "Plane independent (serial)",
            ReduceStrategy {
                autotune_level: AutotuneLevel::Full,
                routine: plane(),
                vectorization: serial,
            },
        ),
        CatalogEntry::new(
            "plane_parallel",
            "Plane independent (parallel)",
            ReduceStrategy {
                autotune_level: AutotuneLevel::Full,
                routine: plane(),
                vectorization: parallel,
            },
        ),
        CatalogEntry::new(
            "cube_serial",
            "Cube use_planes (serial)",
            ReduceStrategy {
                autotune_level: AutotuneLevel::Full,
                routine: cube(),
                vectorization: serial,
            },
        ),
        CatalogEntry::new(
            "cube_parallel",
            "Cube use_planes (parallel)",
            ReduceStrategy {
                autotune_level: AutotuneLevel::Full,
                routine: cube(),
                vectorization: parallel,
            },
        ),
    ]
}