Expand description
GPU kernel library for common scientific computing operations
This module provides optimized GPU kernels for various operations used in scientific computing, with support for multiple GPU backends.
Modules§
- blas
- BLAS-like operations for GPU
- complex
- Complex number operations for GPU kernels
- elementwise
- Element-wise operation kernels for GPU
- ml
- Machine learning kernels for GPU
- reduction
- Reduction operations for GPU (sum, mean, max, min, etc.)
- transform
- Transform operations for GPU (FFT, DCT, convolution, etc.)
Structs§
- Base
Kernel - Base kernel implementation that can be used by specialized kernels
- Kernel
Metadata - Metadata for kernel execution
- Kernel
Params - Parameters for kernel specialization
- Kernel
Registry - Registry of available GPU kernels
Enums§
- Data
Type - Supported data types for GPU kernels
- Operation
Type - The type of operation performed by the kernel
Traits§
- GpuKernel
- GPU Kernel interface