| 210 | } |
| 211 | |
| 212 | FailureOr<::oneflow::DataType> FromMLIRTypeToOFDataType(Type mlir_type) { |
| 213 | if (mlir_type.dyn_cast<InvalidElementType>()) { return ::oneflow::DataType::kInvalidDataType; } |
| 214 | if (mlir_type.dyn_cast<CharElementType>()) { return ::oneflow::DataType::kChar; } |
| 215 | if (mlir_type.dyn_cast<OFRecordElementType>()) { return ::oneflow::DataType::kOFRecord; } |
| 216 | if (mlir_type.dyn_cast<TensorBufferElementType>()) { return ::oneflow::DataType::kTensorBuffer; } |
| 217 | if (mlir_type.isF16()) { return ::oneflow::DataType::kFloat16; } |
| 218 | if (mlir_type.isF32()) { return ::oneflow::DataType::kFloat; } |
| 219 | if (mlir_type.isF64()) { return ::oneflow::DataType::kDouble; } |
| 220 | |
| 221 | if (mlir_type.isSignlessInteger(8)) { return ::oneflow::DataType::kBool; } |
| 222 | if (mlir_type.isSignlessInteger(16)) { return ::oneflow::DataType::kUInt16; } |
| 223 | if (mlir_type.isSignlessInteger(32)) { return ::oneflow::DataType::kUInt32; } |
| 224 | if (mlir_type.isSignlessInteger(64)) { return ::oneflow::DataType::kUInt64; } |
| 225 | if (mlir_type.isSignlessInteger(128)) { return ::oneflow::DataType::kUInt128; } |
| 226 | |
| 227 | if (mlir_type.isSignedInteger(8)) { return ::oneflow::DataType::kInt8; } |
| 228 | if (mlir_type.isSignedInteger(16)) { return ::oneflow::DataType::kInt16; } |
| 229 | if (mlir_type.isSignedInteger(32)) { return ::oneflow::DataType::kInt32; } |
| 230 | if (mlir_type.isSignedInteger(64)) { return ::oneflow::DataType::kInt64; } |
| 231 | if (mlir_type.isSignedInteger(128)) { return ::oneflow::DataType::kInt128; } |
| 232 | llvm::errs() << "Unsupported data type: " << mlir_type << "\n"; |
| 233 | return failure(); |
| 234 | } |
| 235 | |
| 236 | FailureOr<::oneflow::DataType> FromMLIRDataTypeToOFDataType(::mlir::oneflow::DataType data_type) { |
| 237 | switch (data_type) { |
no outgoing calls
no test coverage detected