Expand description
CUDA kernel definitions
This module contains the Rust-side definitions for CUDA kernels. The actual CUDA code would be in .cu files compiled by nvcc.
Modules§
- attention
- Attention kernel signatures
- elementwise
- Elementwise kernel signatures
- matmul
- Matrix multiplication kernel signatures
- normalization
- Normalization kernel signatures
- reduction
- Reduction kernel signatures
Structs§
- Launch
Config - Kernel launch configuration
Type Aliases§
- Kernel
Fn - Kernel function pointer type