| 143 | } |
| 144 | |
| 145 | Maybe<DLDataType> ToDLDataType(DataType ofdtype) { |
| 146 | DLDataType dtype; |
| 147 | dtype.lanes = 1; |
| 148 | dtype.bits = GetSizeOfDataType(ofdtype) * 8; |
| 149 | switch (ofdtype) { |
| 150 | case DataType::kUInt8: dtype.code = DLDataTypeCode::kDLUInt; break; |
| 151 | case DataType::kInt8: dtype.code = DLDataTypeCode::kDLInt; break; |
| 152 | case DataType::kInt16: dtype.code = DLDataTypeCode::kDLInt; break; |
| 153 | case DataType::kInt32: dtype.code = DLDataTypeCode::kDLInt; break; |
| 154 | case DataType::kInt64: dtype.code = DLDataTypeCode::kDLInt; break; |
| 155 | case DataType::kFloat16: dtype.code = DLDataTypeCode::kDLFloat; break; |
| 156 | case DataType::kFloat: dtype.code = DLDataTypeCode::kDLFloat; break; |
| 157 | case DataType::kDouble: dtype.code = DLDataTypeCode::kDLFloat; break; |
| 158 | case DataType::kBFloat16: dtype.code = DLDataTypeCode::kDLBfloat; break; |
| 159 | default: UNIMPLEMENTED_THEN_RETURN() << "Unsupported data type: " << DataType_Name(ofdtype); |
| 160 | } |
| 161 | return dtype; |
| 162 | } |
| 163 | |
| 164 | // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init) |
| 165 | struct ATenDLMTensor { |
no test coverage detected