diff options
| author | Les Bell <les.bell@arm.com> | 2021-10-11 12:50:31 +0100 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2021-10-12 16:13:08 +0000 |
| commit | 6040b4d7ba011c0c7e00212a1bc8104e23e838d2 (patch) | |
| tree | 8afa2ada8bff899495270d36949061d83aa9d7f6 | |
| parent | b6b59e3870085870337a4ea1d6d051178cbd6730 (diff) | |
| download | reference_model-6040b4d7ba011c0c7e00212a1bc8104e23e838d2.tar.gz | |
COND_IF int8, int16 tests
Change-Id: Id1c2d2a41bdfff6b3fe45f79bf99abd0b5beb704
Signed-off-by: Les Bell <les.bell@arm.com>
| -rw-r--r-- | verif/tosa_test_gen.py | 24 |
1 files changed, 13 insertions, 11 deletions
diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index 928ac0e..a03c66f 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -2934,17 +2934,19 @@ class TosaTestGen: op['op'], [cond_tens.name, a.name, b.name], [result_tens.name], attr ) - self.ser.startBasicBlock(then_block) - self.ser.addInputTensor(a) - self.ser.addInputTensor(b) - then_tens = self.ser.addOutput(a.shape, a.dtype) - self.ser.addOperator(Op.ADD, [a.name, b.name], [then_tens.name]) + if a.dtype in (DType.FLOAT, DType.INT32): + then_op, else_op = Op.ADD, Op.SUB + elif a.dtype in (DType.INT8, DType.INT16): + then_op, else_op = Op.LOGICAL_RIGHT_SHIFT, Op.LOGICAL_LEFT_SHIFT + else: + assert False, f"No tests for DType: {a.dtype}" - self.ser.startBasicBlock(else_block) - self.ser.addInputTensor(a) - self.ser.addInputTensor(b) - else_tens = self.ser.addOutput(a.shape, a.dtype) - self.ser.addOperator(Op.SUB, [a.name, b.name], [else_tens.name]) + for block, op in ((then_block, then_op), (else_block, else_op)): + self.ser.startBasicBlock(block) + self.ser.addInputTensor(a) + self.ser.addInputTensor(b) + tens = self.ser.addOutput(a.shape, a.dtype) + self.ser.addOperator(op, [a.name, b.name], [tens.name]) return result_tens @@ -4117,7 +4119,7 @@ class TosaTestGen: TosaTensorGen.tgBasic, TosaArgGen.agCondIf, ), - "types": TYPE_FI32, + "types": TYPE_INT_FP, }, # while_loop "while_loop": { |