Expand description
Kernel module
Modules
- Convolution kernels
- Matmul kernels
- Pooling kernels
- Pseudo-random number generator kernels
- Reduction algorithms
Macros
- Creates a binary elementwise kernel.
- Creates a binary elementwise inplace kernel.
- Generates kernel source code by replacing some information using templating.
- Creates a unary kernel.
- Creates a unary inplace kernel.
- Creates a unary scalar kernel.
- Creates a unary scalar inplace kernel.
Structs
- Generated kernel from wgsl file.
- Generated kernel from wgsl file.
- 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.
- Generated kernel from wgsl file.
- Generated kernel from wgsl file.
- Generated kernel from wgsl file.
- Generated kernel from wgsl file.
Traits
- Dynamic wgpu kernel to create a source template.
- Static wgpu kernel to create a source template.
Functions
- Execute a binary kernel using the provided WORKGROUP.
- Execute a binary kernel using the default settings.
- Execute a binary inplace kernel using the provided WORKGROUP.
- Execute a binary inplace kernel using the default settings.
- 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.
- Execute a unary kernel using the provided WORKGROUP.
- Execute a unary kernel using the default settings.
- Execute a unary inplace kernel using the provided WORKGROUP.
- Execute a unary inplace kernel using the default settings.
- Execute a unary scalar kernel using the provided WORKGROUP.
- Execute a unary scalar kernel using the default settings.
- Execute a unary scalar inplace kernel using the provided WORKGROUP.
- Execute a unary scalar inplace kernel using the default settings.