numr 0.5.2

High-performance numerical computing with multi-backend GPU acceleration (CPU/CUDA/WebGPU)
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105

//! CPU runtime tensor operation implementations
//!
//! This module loads all operation implementations from the src/ops/cpu/ directory.
//! Each operation type (unary, binary, reduce, etc.) is implemented in its own module.

use super::{CpuClient, CpuRuntime};
use crate::ops::TensorOps;

// TensorOps is a supertrait that composes all individual operation traits.
// Since we implement all the component traits in separate files, we just need
// this empty impl to satisfy the supertrait requirement.
impl TensorOps<CpuRuntime> for CpuClient {}

// Load all CPU operation implementations from src/ops/cpu/
#[path = "../../ops/cpu/type_conversion.rs"]
mod type_conversion;

#[path = "../../ops/cpu/complex.rs"]
mod complex;

#[path = "../../ops/cpu/normalization.rs"]
mod normalization;

#[path = "../../ops/cpu/matmul.rs"]
mod matmul;

#[path = "../../ops/cpu/conv.rs"]
mod conv;

#[path = "../../ops/cpu/cumulative.rs"]
mod cumulative;

#[path = "../../ops/cpu/activation.rs"]
mod activation;

#[path = "../../ops/cpu/binary.rs"]
mod binary;

#[path = "../../ops/cpu/unary.rs"]
mod unary;

#[path = "../../ops/cpu/linalg.rs"]
mod linalg;

#[path = "../../ops/cpu/statistics.rs"]
mod statistics;

#[path = "../../ops/cpu/random.rs"]
mod random;

#[path = "../../ops/cpu/advanced_random.rs"]
mod advanced_random;

#[path = "../../ops/cpu/quasirandom.rs"]
mod quasirandom;

#[path = "../../ops/cpu/reduce.rs"]
mod reduce;

#[path = "../../ops/cpu/sorting.rs"]
mod sorting;

#[path = "../../ops/cpu/conditional.rs"]
mod conditional;

#[path = "../../ops/cpu/utility.rs"]
mod utility;

#[path = "../../ops/cpu/scalar.rs"]
mod scalar;

#[path = "../../ops/cpu/compare.rs"]
mod compare;

#[path = "../../ops/cpu/logical.rs"]
mod logical;

#[path = "../../ops/cpu/indexing.rs"]
mod indexing;

#[path = "../../ops/cpu/shape.rs"]
mod shape;

#[path = "../../ops/cpu/distance.rs"]
mod distance;

#[path = "../../ops/cpu/multivariate.rs"]
mod multivariate;

#[path = "../../ops/cpu/semiring_matmul.rs"]
mod semiring_matmul;

#[path = "../../ops/cpu/einsum.rs"]
mod einsum;

#[path = "../../ops/cpu/gemm_epilogue.rs"]
mod gemm_epilogue;

#[cfg(feature = "fp8")]
#[path = "../../ops/cpu/fp8_matmul.rs"]
mod fp8_matmul;

#[cfg(feature = "sparse")]
#[path = "../../ops/cpu/sparse_24.rs"]
mod sparse_24;