Module nncombinator::cuda

Expand description

Function to wrap and handle cuda kernel

Modules

FFI Module

Implementation of function calls to cuda kernel functions

This is a memory-related module for CUDA.

Wrapper to handle cuda host memory

Cuda memory object allocated from the memory pool

Wrapper to handle cuda device memory

Trait defining the conversion to an immutable pointer type passed to the cuda kernel

Trait defining the conversion to an smutable pointer type passed to the cuda kernel

Obtaining an mutable pointer

Obtaining an mutable void pointer

Obtaining an immutable pointer

Obtaining an immutable void pointer

Trait to associate a type with a cudnn type

Trait defining cuda kernel functions

Trait that defines arguments passed to cuda kernel functions

Trait defining cuda’s synchronous memory operations

Trait defining cuda’s asynchronous memory operations