diff options
author | Won Jeon <won.jeon@arm.com> | 2023-08-10 10:33:01 +0000 |
---|---|---|
committer | Won Jeon <won.jeon@arm.com> | 2023-08-18 15:21:15 -0700 |
commit | a21b2e88d19d8cb11a9120d40bacbb594d600565 (patch) | |
tree | 3bc8a40db72a31c1e552a3bd6339627a1175686e /reference_model/include/dtype.h | |
parent | e0247481eb1f83f6eb7161d3f7ac2690b180952a (diff) | |
download | reference_model-a21b2e88d19d8cb11a9120d40bacbb594d600565.tar.gz |
Add DIM operator to reference model
Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: Iea11ee5d3d98773e9c5e9b827593c05afb41ce3b
Diffstat (limited to 'reference_model/include/dtype.h')
-rw-r--r-- | reference_model/include/dtype.h | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/reference_model/include/dtype.h b/reference_model/include/dtype.h index bde678a..1b01a0e 100644 --- a/reference_model/include/dtype.h +++ b/reference_model/include/dtype.h @@ -40,6 +40,7 @@ enum TOSA_REF_TYPE : uint32_t TOSA_REF_TYPE_UINT16 = 9, TOSA_REF_TYPE_FP16 = 10, TOSA_REF_TYPE_BF16 = 11, + TOSA_REF_TYPE_SHAPE = 12, TOSA_REF_TYPE_FP64 = 99, // FP64 is special: add new data types above }; @@ -71,6 +72,8 @@ inline const char* EnumNameTOSAREFTYPE(TOSA_REF_TYPE e) return EnumNameDType(DType_FP16); case TOSA_REF_TYPE_BF16: return EnumNameDType(DType_BF16); + case TOSA_REF_TYPE_SHAPE: + return EnumNameDType(DType_SHAPE); case TOSA_REF_TYPE_FP64: return "FP64"; default: @@ -82,7 +85,7 @@ inline const char* EnumNameTOSAREFTYPE(TOSA_REF_TYPE e) // return corresponding TOSA_REF_TYPE for DType inline TOSA_REF_TYPE ConvertDType(const DType dtype) { - assert(DType_MAX == DType_BF16); // must update whenever DType_MAX changes + assert(DType_MAX == DType_SHAPE); // must update whenever DType_MAX changes if (g_func_config.precise_mode) { @@ -122,6 +125,8 @@ inline TOSA_REF_TYPE ConvertDType(const DType dtype) return TOSA_REF_TYPE_FP16; case DType_BF16: return TOSA_REF_TYPE_BF16; + case DType_SHAPE: + return TOSA_REF_TYPE_SHAPE; default: break; } |