mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
01abb5af21
Follow up to #107586. Pull Request resolved: https://github.com/pytorch/pytorch/pull/115214 Approved by: https://github.com/peterbell10, https://github.com/malfet
26 lines
1.0 KiB
C++
26 lines
1.0 KiB
C++
#pragma once
|
|
|
|
#include <onnx/onnx_pb.h>
|
|
|
|
namespace torch::onnx {
|
|
|
|
// The following constants are defined here to avoid breaking Meta's internal
|
|
// usage of ONNX which pre-dates ONNX 1.14 and thus does not support FLOAT8:
|
|
// cf. https://github.com/pytorch/pytorch/pull/106379#issuecomment-1675189340
|
|
// -abock, 2023-08-25
|
|
//
|
|
// ::ONNX_NAMESPACE::TensorProto_DataType_FLOAT8E4M3FN
|
|
constexpr auto TensorProto_DataType_FLOAT8E4M3FN =
|
|
static_cast<::ONNX_NAMESPACE::TensorProto_DataType>(17);
|
|
// ::ONNX_NAMESPACE::TensorProto_DataType_FLOAT8E4M3FNUZ
|
|
constexpr auto TensorProto_DataType_FLOAT8E4M3FNUZ =
|
|
static_cast<::ONNX_NAMESPACE::TensorProto_DataType>(18);
|
|
// ::ONNX_NAMESPACE::TensorProto_DataType_FLOAT8E5M2
|
|
constexpr auto TensorProto_DataType_FLOAT8E5M2 =
|
|
static_cast<::ONNX_NAMESPACE::TensorProto_DataType>(19);
|
|
// ::ONNX_NAMESPACE::TensorProto_DataType_FLOAT8E5M2FNUZ
|
|
constexpr auto TensorProto_DataType_FLOAT8E5M2FNUZ =
|
|
static_cast<::ONNX_NAMESPACE::TensorProto_DataType>(20);
|
|
|
|
} // namespace torch::onnx
|