Mila
Deep Neural Network Library
|
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. | |