Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Dnn.Utils.Cuda.Common.ixx File Reference
#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
 

Macro Definition Documentation

◆ PRECISION_MODE

#define PRECISION_MODE   PRECISION_BF16