aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_arg_gen.py
diff options
context:
space:
mode:
authorJames Ward <james.ward@arm.com>2022-11-17 11:15:14 +0000
committerJames Ward <james.ward@arm.com>2022-11-21 12:09:35 +0000
commitf089099f2d87c7801f72ce1801daace1db83d154 (patch)
tree98c6d473ef880246fbbbf7c1f390950259c5f794 /verif/generator/tosa_arg_gen.py
parent990cc4be0b2c33f470e42ce7761e06a5b072afe6 (diff)
downloadreference_model-f089099f2d87c7801f72ce1801daace1db83d154.tar.gz
Create MI tests for Data Layout: CONCAT, PAD, RESHAPE, REVERSE, SLICE, TILE, TRANSPOSE
Signed-off-by: James Ward <james.ward@arm.com> Change-Id: I626d9ece0290c5104c9ac42d20be2f9db1e55fa7
Diffstat (limited to 'verif/generator/tosa_arg_gen.py')
-rw-r--r--verif/generator/tosa_arg_gen.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py
index 932ad55..4e15b06 100644
--- a/verif/generator/tosa_arg_gen.py
+++ b/verif/generator/tosa_arg_gen.py
@@ -1247,7 +1247,7 @@ class TosaArgGen:
if dtype in [DType.BOOL, DType.INT8, DType.INT16, DType.INT32]:
pad_const_int = testGen.getRandNumberDType(dtype)
pad_const_fp = 0
- elif dtype in (DType.FP16, DType.FP32):
+ elif dtype in (DType.FP16, DType.BF16, DType.FP32):
pad_const_int = 0
pad_const_fp = testGen.getRandNumberDType(dtype)
else: