Mila.ixx File Reference

#include <cuda_fp16.h>
#include <string>
#include <iostream>
#include <memory>
#include "Version.h"
import Compute.DeviceRegistrar;
import Compute.OperationsRegistrar;
import Compute.CpuDevice;
import Compute.DeviceHelpers;
import Utils.DefaultLogger;
import Compute.CpuMemoryResource;
import Compute.CpuLinearOp;
import Compute.DeviceRegistry;
import Dnn.Modules.Gelu;
import Dnn.Gpt2.DatasetReader;
import Compute.Precision;
import Dnn.TensorHelpers;
import Dnn.TensorBuffer;
import Dnn.Tensor;
import Compute.CudaMHAOp;
import Dnn.ComponentConfig;
import Compute.CpuResidualOp;
import Dnn.CompositeModule;
import Dnn.Modules.Softmax;
import Dnn.Module;
import Mila.Version;
import Serialization.ModelArchive;
import Cuda.Error;
import Compute.CudaDevice;
import Compute.BinaryOperation;
import Compute.CudaLayerNormOp;
import Core.RandomGenerator;
import Data.DataLoader;
import Compute.CudaEncoderOp;
import Dnn.TensorTraits;
import Dnn.Model;
import Compute.OperationBase;
import Compute.OperationType;
import Compute.ComputeDevice;
import Dnn.Modules.Attention;
import Dnn.ActivationType;
import Cuda.Helpers;
import Compute.MemoryResource;
import Compute.OperationAttributes;
import Compute.CpuGeluOp;
import Compute.CudaMemoryResource;
import Compute.CpuLayerNormOp;
import Compute.CudaManagedMemoryResource;
import Dnn.Blocks.TransformerBlock;
import Compute.CudaSoftmaxOp;
import Compute.CudaPinnedMemoryResource;
import Compute.DynamicMemoryResource;
import Compute.OperationRegistry;
import Compute.CpuEncoderOp;
import Utils.Logger;
import Compute.UnaryOperation;
import Compute.CudaLinearOp;
import Compute.CpuSoftmaxOp;
import Compute.CudaGeluOp;
import Dnn.Modules.Linear;
import Compute.CudaResidualOp;
import Compute.DeviceContext;
import Dnn.Modules.LayerNorm;
import Dnn.Modules.Encoder;
import Compute.DeviceType;
import Dnn.Modules.Residual;
import Dnn.Blocks.MLP;

Namespaces
namespace	Mila
namespace	Mila::detail

Functions
Version	Mila::getAPIVersion ()
	Gets the current Mila API version.
bool	Mila::initialize (unsigned int randomSeed=0)
	Initializes the Mila framework.
void	Mila::initializeLogger (Utils::LogLevel level=Utils::LogLevel::Info)
void	Mila::shutdown ()

Variables
std::shared_ptr< Utils::DefaultLogger >	Mila::detail::g_defaultLogger