Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Cuda.Helpers Module Reference

Functions

constexpr int Mila::Dnn::Compute::Cuda::ceil_div (int M, int N)
 Calculates ceiling division for kernel grid/block dimensions.
int Mila::Dnn::Compute::Cuda::checkDevice (int deviceId)
 Validates that a device ID is valid and available.
int Mila::Dnn::Compute::Cuda::findCudaDevice (int deviceId=-1, bool preferMemory=false)
 Finds the most appropriate CUDA device for computation.
int Mila::Dnn::Compute::Cuda::getBestDeviceId (bool preferMemory=false)
 Identifies the best CUDA device based on performance characteristics.
int Mila::Dnn::Compute::Cuda::getDeviceCount ()
 Gets the number of available CUDA devices.
int Mila::Dnn::Compute::Cuda::getDriverVersion ()
 Gets the installed CUDA driver version.
int Mila::Dnn::Compute::Cuda::getRuntimeVersion ()
 Gets the installed CUDA runtime version.
void Mila::Dnn::Compute::Cuda::setCurrentDevice (int device_id)
 Sets the current CUDA device with thread-local caching.

Files

file  /__w/Mila/Mila/Mila/Src/Dnn/Compute/Devices/Cuda/Helpers/CudaHelpers.ixx
 CUDA utility functions for device management and kernel execution.