Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
DeviceType.ixx File Reference

Device type definitions and conversion utilities for compute devices. More...

#include <stdexcept>
#include <algorithm>
#include <string>

Namespaces

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

Enumerations

enum class  Mila::Dnn::Compute::DeviceType { Mila::Dnn::Compute::Cpu , Mila::Dnn::Compute::Cuda }
 Enumeration of supported compute device types. More...
 

Functions

std::string Mila::Dnn::Compute::deviceToString (DeviceType device_type)
 Converts a DeviceType to its string representation.
 
DeviceType Mila::Dnn::Compute::toDeviceType (std::string device_type)
 Converts a string to the corresponding DeviceType.
 

Detailed Description

Device type definitions and conversion utilities for compute devices.