singe-kernel 0.1.0-alpha.4

Reusable CPU and GPU 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

#[cfg(all(feature = "dtype-f16", feature = "kernel-activation"))]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f16/activation.rs"]
pub mod activation;
#[cfg(all(
    any(feature = "dtype-f16", feature = "dtype-bf16"),
    feature = "kernel-cast"
))]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f16/cast.rs"]
pub mod cast;
#[cfg(all(feature = "dtype-f16", feature = "kernel-elementwise"))]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f16/elementwise.rs"]
pub mod elementwise;
#[cfg(all(feature = "dtype-f16", feature = "kernel-quantization"))]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f16/quantization.rs"]
pub mod quantization;
#[cfg(all(feature = "dtype-f16", feature = "kernel-reduction"))]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f16/reduction.rs"]
pub mod reduction;
#[cfg(all(feature = "dtype-f16", feature = "kernel-scalar"))]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f16/scalar.rs"]
pub mod scalar;
#[cfg(all(feature = "dtype-f16", feature = "kernel-softmax"))]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f16/softmax.rs"]
pub mod softmax;
#[cfg(all(feature = "dtype-f16", feature = "kernel-unary"))]
#[doc(hidden)]
#[path = "../../../../-generated/cutile/f16/unary.rs"]
pub mod unary;

#[cfg(all(feature = "dtype-f16", feature = "kernel-embedding"))]
pub mod embedding;
#[cfg(all(feature = "dtype-f16", feature = "kernel-gather"))]
pub mod gather;
#[cfg(feature = "kernel-normalization")]
pub mod normalization;
#[cfg(all(feature = "dtype-f16", feature = "kernel-pooling"))]
pub mod pooling;
#[cfg(all(feature = "dtype-f16", feature = "kernel-positional"))]
pub mod positional;
#[cfg(all(feature = "dtype-f16", feature = "kernel-shape"))]
pub mod shape;

#[cfg(any(
    feature = "kernel-activation",
    feature = "kernel-cast",
    feature = "kernel-elementwise",
    feature = "kernel-quantization",
    feature = "kernel-reduction",
    feature = "kernel-scalar",
    feature = "kernel-softmax",
    feature = "kernel-unary",
))]
#[doc(hidden)]
pub mod utility;