Module dopt.cuda
This is the main interface for the dopt CUDA backend.
The APIs in this module allow users to evaluate operation graphs on GPUs through the use of CUDA. There is also functionality to register CUDA implementations of custom operations.
In future, this module will also have an interface allowing the user to register their own optimisation passes to be called when constructing a plan.
Functions
| Name | Description |
|---|---|
deregisterCUDAKernel(opType) | Deregisters a kernel constructor associated with the given operation type. |
evaluateCUDA(ops, args) | Used for performing a one-off evaluation of a set of operations. |
evaluateCUDA(op, args) | A convenience overload that evaluates a single operation and returns a single Buffer.
|
listCUDAOperations() | Provides a list of all operation types supported by the CUDA backend. |
registerCUDAKernel(opName, kernelCtr) | Registers a CUDA kernel constructor for a given operation type. |
Interfaces
| Name | Description |
|---|---|
CUDAKernel | Provides a common interface for CUDA kernels. |
Classes
| Name | Description |
|---|---|
CUDABuffer | A class that encapsulates the CUDA memory allocation/deallocation process. |
CUDAPlan | A Plan stores all the resources (preallocated buffers, custom CUDA kernels) required to evaluate nodes from the Operation graph. |