ComputePrecision.ixx File Reference

Defines precision policy for neural network operations. More...

#include <string_view>
#include <cuda_fp16.h>
#include <cuda_bf16.h>
#include <cuda_fp8.h>
#include <type_traits>
#include <string>
#include <sstream>
import Compute.DeviceType;
import Dnn.TensorTraits;

Classes
class	Mila::Dnn::Compute::ComputePrecision
	Controls automatic mixed precision behavior for neural network operations. More...

Namespaces
namespace	Mila
namespace	Mila::Dnn
namespace	Mila::Dnn::Compute

Detailed Description

Defines precision policy for neural network operations.