|
Mila
Deep Neural Network Library
|
#include <string>#include <sstream>#include <cuda_runtime.h>#include <cuda_bf16.h>#include <cuda_fp16.h>Namespaces | |
| namespace | Mila |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Utils |
| namespace | Mila::Dnn::Utils::Cuda |
Macros | |
| #define | PRECISION_MODE PRECISION_BF16 |
Typedefs | |
| typedef __nv_bfloat16 | Mila::Dnn::Utils::Cuda::floatX |
| #define PRECISION_MODE PRECISION_BF16 |