Skip to content

Commit 91ff351

Browse files
committed
Adding support for E5M3 type
1 parent 13cddfe commit 91ff351

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

  • onnxruntime/core/providers/openvino/ov_versions

onnxruntime/core/providers/openvino/ov_versions/data_ops.cc

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -270,6 +270,8 @@ void DataOps::populate_types_supported() {
270270
std::make_pair(V_2024_4, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_UINT4));
271271
supported_types_initializer_.insert(
272272
std::make_pair(V_2026_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT8E4M3FN));
273+
supported_types_initializer_.insert(
274+
std::make_pair(V_2026_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT8E5M2));
273275

274276
supported_types_npu_.insert(
275277
std::make_pair(V_2020_4, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_BOOL));
@@ -291,6 +293,8 @@ void DataOps::populate_types_supported() {
291293
std::make_pair(V_2021_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT16));
292294
supported_types_npu_.insert(
293295
std::make_pair(V_2026_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT8E4M3FN));
296+
supported_types_npu_.insert(
297+
std::make_pair(V_2026_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT8E5M2));
294298
supported_types_npu_.insert(
295299
std::make_pair(V_2024_4, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_INT4));
296300
supported_types_npu_.insert(
@@ -322,6 +326,8 @@ void DataOps::populate_types_supported() {
322326
std::make_pair(V_2024_4, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_UINT4));
323327
supported_types_cpu_.insert(
324328
std::make_pair(V_2026_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT8E4M3FN));
329+
supported_types_cpu_.insert(
330+
std::make_pair(V_2026_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT8E5M2));
325331

326332
supported_types_gpu_.insert(
327333
std::make_pair(V_2020_4, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT));
@@ -343,6 +349,8 @@ void DataOps::populate_types_supported() {
343349
std::make_pair(V_2024_4, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_UINT4));
344350
supported_types_gpu_.insert(
345351
std::make_pair(V_2026_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT8E4M3FN));
352+
supported_types_gpu_.insert(
353+
std::make_pair(V_2026_1, ONNX_NAMESPACE::TensorProto_DataType::TensorProto_DataType_FLOAT8E5M2));
346354
}
347355

348356
void DataOps::populate_op_mode_supported() {

0 commit comments

Comments
 (0)