numr/runtime/cpu/
ops.rs

1//! CPU runtime tensor operation implementations
2//!
3//! This module loads all operation implementations from the src/ops/cpu/ directory.
4//! Each operation type (unary, binary, reduce, etc.) is implemented in its own module.
5
6use super::{CpuClient, CpuRuntime};
7use crate::ops::TensorOps;
8
9// TensorOps is a supertrait that composes all individual operation traits.
10// Since we implement all the component traits in separate files, we just need
11// this empty impl to satisfy the supertrait requirement.
12impl TensorOps<CpuRuntime> for CpuClient {}
13
14// Load all CPU operation implementations from src/ops/cpu/
15#[path = "../../ops/cpu/type_conversion.rs"]
16mod type_conversion;
17
18#[path = "../../ops/cpu/complex.rs"]
19mod complex;
20
21#[path = "../../ops/cpu/normalization.rs"]
22mod normalization;
23
24#[path = "../../ops/cpu/matmul.rs"]
25mod matmul;
26
27#[path = "../../ops/cpu/conv.rs"]
28mod conv;
29
30#[path = "../../ops/cpu/cumulative.rs"]
31mod cumulative;
32
33#[path = "../../ops/cpu/activation.rs"]
34mod activation;
35
36#[path = "../../ops/cpu/binary.rs"]
37mod binary;
38
39#[path = "../../ops/cpu/unary.rs"]
40mod unary;
41
42#[path = "../../ops/cpu/linalg.rs"]
43mod linalg;
44
45#[path = "../../ops/cpu/statistics.rs"]
46mod statistics;
47
48#[path = "../../ops/cpu/random.rs"]
49mod random;
50
51#[path = "../../ops/cpu/advanced_random.rs"]
52mod advanced_random;
53
54#[path = "../../ops/cpu/quasirandom.rs"]
55mod quasirandom;
56
57#[path = "../../ops/cpu/reduce.rs"]
58mod reduce;
59
60#[path = "../../ops/cpu/sorting.rs"]
61mod sorting;
62
63#[path = "../../ops/cpu/conditional.rs"]
64mod conditional;
65
66#[path = "../../ops/cpu/utility.rs"]
67mod utility;
68
69#[path = "../../ops/cpu/scalar.rs"]
70mod scalar;
71
72#[path = "../../ops/cpu/compare.rs"]
73mod compare;
74
75#[path = "../../ops/cpu/logical.rs"]
76mod logical;
77
78#[path = "../../ops/cpu/indexing.rs"]
79mod indexing;
80
81#[path = "../../ops/cpu/shape.rs"]
82mod shape;
83
84#[path = "../../ops/cpu/distance.rs"]
85mod distance;
86
87#[path = "../../ops/cpu/multivariate.rs"]
88mod multivariate;
89
90#[path = "../../ops/cpu/semiring_matmul.rs"]
91mod semiring_matmul;
92
93#[path = "../../ops/cpu/einsum.rs"]
94mod einsum;
95
96#[path = "../../ops/cpu/gemm_epilogue.rs"]
97mod gemm_epilogue;
98
99#[cfg(feature = "fp8")]
100#[path = "../../ops/cpu/fp8_matmul.rs"]
101mod fp8_matmul;
102
103#[cfg(feature = "sparse")]
104#[path = "../../ops/cpu/sparse_24.rs"]
105mod sparse_24;
numr/runtime/cpu/ops.rs

numr/runtime/cpu/
ops.rs
