diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-03-12 11:46:50 +0000 |
---|---|---|
committer | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-03-12 11:46:50 +0000 |
commit | 80fd9b8bf8d6def0a4ce6a3c59bdc598fecbd1d1 (patch) | |
tree | 8e98100881ddb091f4947853500f11a327a5fec9 /verif | |
parent | 1adff8301ac7e4933382549ad22b3b47cdb01eda (diff) | |
download | reference_model-80fd9b8bf8d6def0a4ce6a3c59bdc598fecbd1d1.tar.gz |
Fix CONV2D WrongOutputType ERRORIF test gen
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I663a92001bf902cdc5327f386b2e896052a816ff
Diffstat (limited to 'verif')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index 4309024..0f68999 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -5259,7 +5259,7 @@ class OutputShaper: if error_name == ErrorIf.WrongOutputType: if ifm.dtype == DType.FP16: excludes = [DType.FP16, DType.FP32] - if ifm.dtype in [DType.FP8E4M3, DType.FP8E5M2]: + elif ifm.dtype in [DType.FP8E4M3, DType.FP8E5M2]: excludes = [DType.FP16] else: excludes = [out_dtype] |