Expand description
§SIMD Optimizations Module
Vectorized operations using AVX2/SSE instructions for high-performance tensor operations.
§Features
- Vectorized dot products
- SIMD matrix multiplication
- Element-wise operations (add, mul, relu)
- Reduction operations (sum, max, min)
§Platform Support
- x86_64 with AVX2: Full SIMD acceleration
- Fallback: Pure Rust implementation for other platforms
Functions§
- add_
elementwise_ simd - Adição elemento-wise com SIMD
- dot_
product_ simd - Dot product vetorizado usando AVX2
- mul_
elementwise_ simd - Multiplicação elemento-wise com SIMD
- mul_
scalar_ simd - Multiplicação escalar com SIMD
- relu_
simd - ReLU com SIMD (max(0, x))
- sum_
simd - Soma de todos os elementos com SIMD