commit 718f347a2d886381de19420b5b5b99db8f2b7338
author Jeremy Johnson <jeremy.johnson@arm.com> Thu Nov 30 14:18:19 2023 +0000
committer Jeremy Johnson <jeremy.johnson@arm.com> Mon Dec 04 10:02:15 2023 +0000
tree 87f6ab932029654b4e0704938dbe6ab7135da27d
parent fe79accba2c220036c7b5ea0aa27bff5ef74ec73

Main Compliance FP16 support - generate and verify.

FP16 support for all existing operators for compliance:
* DOT_PRODUCT
* ULP
* EXACT
* ABS_ERROR

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I8d25448a793375b53880da3787d8f839767f02cf

verif/generator/tosa_utils.py

diff --git a/verif/generator/tosa_utils.py b/verif/generator/tosa_utils.py
index 318f296..3d733f4 100644
--- a/verif/generator/tosa_utils.py
+++ b/verif/generator/tosa_utils.py
@@ -55,7 +55,7 @@
     """Types supported by the new data generation and compliance flow."""
     if isinstance(dtype, list) or isinstance(dtype, tuple):
         dtype = dtype[0]
-    return dtype in (DType.FP32,)
+    return dtype in (DType.FP32, DType.FP16)
 
 
 def getOpNameFromOpListName(opName):