Expand description
Kernel module
Modules§
- Convolution kernels
- Matmul kernels
- Pooling kernels
- Pseudo-random number generator kernels
- Reduction algorithms
Macros§
- Creates a binary kernel.
- Generates kernel source code by replacing some information using templating.
- Creates a unary kernel.
Structs§
- Generated kernel from wgsl file.
- Generated kernel from wgsl file.
- Generate kernel source code by replacing some information using templating.
- Generates kernel source code by replacing some information using templating.
- Kernel source code abstraction allowing for templating.
Traits§
- Dynamic wgpu kernel to create a source template.
- Static wgpu kernel to create a source template.
Functions§
- Launch an binary operation.
- Create a vector containing the dimension, strides and shape of tensors.
- Similar to build info but with dynamic rank.
- Cast a tensor to the given element type.
- Make a wgpu tensor contiguous.
- Similar to into contiguous but with dynamic rank.
- Launch an unary operation.