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

Functions

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

Files

file  /home/runner/work/Mila/Mila/Mila/Src/Dnn/Compute/Cuda/CudaHelpers.ixx
 CUDA utility functions for device management and kernel execution.