InfiniOps
Operator Library for Accelerators
Loading...
Searching...
No Matches
device.h File Reference
#include <infini/rt.h>
#include "common/traits.h"

Go to the source code of this file.

Classes

struct  infini::ops::ActiveDevicesImpl< typename >
 
struct  infini::ops::ActiveDevicesImpl< typename >::Filter
 

Namespaces

namespace  infini
 
namespace  infini::ops
 

Typedefs

using infini::ops::Device = infini::rt::Device
 
template<Device::Type device_type>
using infini::ops::DeviceEnabled = infini::rt::DeviceEnabled< device_type >
 
using infini::ops::AllDeviceTypes = List< Device::Type::kCpu, Device::Type::kNvidia, Device::Type::kCambricon, Device::Type::kAscend, Device::Type::kMetax, Device::Type::kMoore, Device::Type::kIluvatar, Device::Type::kHygon, Device::Type::kMars, Device::Type::kThead >
 
template<typename T >
using infini::ops::ActiveDevices = typename ActiveDevicesImpl< T >::type