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

//! WebGPU runtime tensor operation implementations
//!
//! This module loads all operation implementations from the src/ops/wgpu/ directory.
//! Each operation type (type conversion, complex, normalization, etc.) is implemented in its own module.

use super::super::{WgpuClient, WgpuRuntime};
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<WgpuRuntime> for WgpuClient {}

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

#[path = "../../../ops/wgpu/fp8_matmul.rs"]
mod fp8_matmul;

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