From 6040b4d7ba011c0c7e00212a1bc8104e23e838d2 Mon Sep 17 00:00:00 2001 From: Les Bell Date: Mon, 11 Oct 2021 12:50:31 +0100 Subject: COND_IF int8, int16 tests Change-Id: Id1c2d2a41bdfff6b3fe45f79bf99abd0b5beb704 Signed-off-by: Les Bell --- verif/tosa_test_gen.py | 24 +++++++++++++----------- 1 file 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": { -- cgit v1.2.1