From 66bad80a98307246f94f8b69d2a62f4649e71455 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Tue, 18 Jan 2022 14:48:35 +0000 Subject: Fix for LOGICAL_LEFT/RIGHT_SHIFT shift values Added missing reference model REQUIRE check for shift value (0-31) Make sure result of LOGICAL_SHIFT_LEFT is masked to input size Fixed test generation to produce shift values in that range Signed-off-by: Jeremy Johnson Change-Id: Id511de0d989ea954fc1afd18dc2051341bce2cd0 --- verif/generator/tosa_test_gen.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'verif/generator') diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index 239a64e..cb97acb 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -5952,6 +5952,20 @@ class TosaTestGen: self.buildPlaceholderTensors(shapeList[0:count], dtypeList[0:count]) ) tens.extend(self.buildConstTensors(shapeList[count:], dtypeList[count:])) + elif op["op"] == Op.LOGICAL_LEFT_SHIFT or op["op"] == Op.LOGICAL_RIGHT_SHIFT: + assert ( + pCount == 2 and cCount == 0 + ), "Op.LOGICAL_LEFT_SHIFT or Op.LOGICAL_RIGHT_SHIFT must have 2 placeholders, 0 consts" + values_arr = self.getRandTensor(shapeList[0], dtypeList[0]) + shift_arr = np.int32(self.rng.integers(low=0, high=32, size=shapeList[1])) + placeholders = [] + placeholders.append( + self.ser.addPlaceholder(shapeList[0], dtypeList[0], values_arr) + ) + placeholders.append( + self.ser.addPlaceholder(shapeList[1], dtypeList[1], shift_arr) + ) + tens.extend(placeholders) else: tens.extend( self.buildPlaceholderTensors(shapeList[0:pCount], dtypeList[0:pCount]) -- cgit v1.2.1