aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_utils.py
diff options
context:
space:
mode:
authorTai Ly <tai.ly@arm.com>2024-03-08 22:19:41 +0000
committerTai Ly <tai.ly@arm.com>2024-03-17 19:56:21 -0700
commit60dc48c4ddf30f2a76d4cfcf1b40ca57b6f3bf95 (patch)
treee3d229a2d596e1a0788dfd75d77b996263055496 /verif/generator/tosa_utils.py
parente67115ef82bcba0718dcbd75cc8411985001b7cc (diff)
downloadreference_model-60dc48c4ddf30f2a76d4cfcf1b40ca57b6f3bf95.tar.gz
[ref model] Change Clamp and Pad attribute fields
This implements changes due to ClampAttribute and PadAttribute field changes. Signed-off-by: Tai Ly <tai.ly@arm.com> Change-Id: Ide01e2a27fe3c1ea7794e7a4b6780b7eae436caf
Diffstat (limited to 'verif/generator/tosa_utils.py')
-rw-r--r--verif/generator/tosa_utils.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py
index 6558bf8..cfe7cc6 100644
--- a/verif/generator/tosa_utils.py
+++ b/verif/generator/tosa_utils.py
@@ -64,6 +64,11 @@ def dtypeWidth(dtype):
raise Exception(f"Unknown dtype, cannot determine width: {dtype}")
+def dtypeIsFloat(dtype):
+ """Is floating point data type"""
+ return dtype in (DType.BF16, DType.FP16, DType.FP32, DType.FP8E4M3, DType.FP8E5M2)
+
+
def dtypeIsSupportedByCompliance(dtype):
"""Types supported by the new data generation and compliance flow."""
if isinstance(dtype, list) or isinstance(dtype, tuple):