diff options
Diffstat (limited to 'reference_model/src/generate/generate_utils.cc')
-rw-r--r-- | reference_model/src/generate/generate_utils.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/reference_model/src/generate/generate_utils.cc b/reference_model/src/generate/generate_utils.cc index 8b16e97..271b7f5 100644 --- a/reference_model/src/generate/generate_utils.cc +++ b/reference_model/src/generate/generate_utils.cc @@ -34,6 +34,8 @@ NLOHMANN_JSON_SERIALIZE_ENUM(DType, { DType::DType_BF16, "BF16" }, { DType::DType_FP32, "FP32" }, { DType::DType_SHAPE, "SHAPE" }, + { DType::DType_FP8E4M3, "FP8E4M3" }, + { DType::DType_FP8E5M2, "FP8E5M2" }, }) NLOHMANN_JSON_SERIALIZE_ENUM(Op, @@ -225,6 +227,8 @@ size_t elementSizeFromType(DType type) case DType::DType_BOOL: case DType::DType_UINT8: case DType::DType_INT8: + case DType::DType_FP8E4M3: + case DType::DType_FP8E5M2: return 1; case DType::DType_UINT16: case DType::DType_INT16: |