data_type.h File Reference

#include <infini/rt.h>
#include "common/traits.h"
#include "device.h"

Go to the source code of this file.

Namespaces
namespace	infini
namespace	infini::ops

Typedefs
template<Device::Type dev, DataType dtype>
using	infini::ops::TypeMap = infini::rt::TypeMap< dev, dtype >
template<Device::Type dev, DataType dtype>
using	infini::ops::TypeMapType = infini::rt::TypeMapType< dev, dtype >
using	infini::ops::FloatTypes = List< DataType::kFloat32, DataType::kFloat64 >
using	infini::ops::ReducedFloatTypes = List< DataType::kFloat16, DataType::kBFloat16 >
using	infini::ops::IntTypes = List< DataType::kInt8, DataType::kInt16, DataType::kInt32, DataType::kInt64 >
using	infini::ops::UIntTypes = List< DataType::kUInt8, DataType::kUInt16, DataType::kUInt32, DataType::kUInt64 >
using	infini::ops::BitTypes8 = List< DataType::kInt8, DataType::kUInt8 >
using	infini::ops::BitTypes16 = List< DataType::kInt16, DataType::kUInt16, DataType::kFloat16, DataType::kBFloat16 >
using	infini::ops::BitTypes32 = List< DataType::kInt32, DataType::kUInt32, DataType::kFloat32 >
using	infini::ops::BitTypes64 = List< DataType::kInt64, DataType::kUInt64, DataType::kFloat64 >
using	infini::ops::AllFloatTypes = ConcatType< FloatTypes, ReducedFloatTypes >
using	infini::ops::AllIntTypes = ConcatType< IntTypes, UIntTypes >
using	infini::ops::AllTypes = ConcatType< AllFloatTypes, AllIntTypes >

Variables
template<Device::Type dev, typename T >
constexpr bool	infini::ops::IsBFloat16 = infini::rt::IsBFloat16<dev, T>
template<Device::Type dev, typename T >
constexpr bool	infini::ops::IsFP16 = infini::rt::IsFP16<dev, T>