cubek-convolution 0.3.0-pre.1

CubeK: Convolution Kernels
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
use cubek_test_utils::CatalogEntry;

use crate::{AcceleratedTileKind, ConvAlgorithm, Strategy};

pub fn strategies() -> Vec<CatalogEntry<Strategy>> {
    vec![CatalogEntry::new(
        "simple_sync_cyclic_cmma",
        "SimpleSyncCyclic / Cmma (inferred)",
        Strategy::Inferred {
            algorithm: ConvAlgorithm::SimpleSyncCyclic,
            tile_kind: AcceleratedTileKind::Cmma,
        },
    )]
}