From e0ded5922496db4f65b8e3be397fa3b7603493c7 Mon Sep 17 00:00:00 2001 From: Jeremy Johnson Date: Mon, 15 Apr 2024 11:21:32 +0100 Subject: Fix ARITHMETIC_RIGHT_SHIFT shift tensor type for int 8 & 16 Signed-off-by: Jeremy Johnson Change-Id: I91f7bc956c3b141e1518098781bbf29577c3fbbc --- verif/generator/tosa_arg_gen.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py index 5957a33..26dd6f9 100644 --- a/verif/generator/tosa_arg_gen.py +++ b/verif/generator/tosa_arg_gen.py @@ -1080,9 +1080,9 @@ class TosaTensorValuesGen: for idx, shape in enumerate(shapeList[:]): if idx == 1: if dtypeList[idx] == DType.INT8: - arr = np.int32(rng.integers(low=0, high=8, size=shape)) + arr = np.int8(rng.integers(low=0, high=8, size=shape)) elif dtypeList[idx] == DType.INT16: - arr = np.int32(rng.integers(low=0, high=16, size=shape)) + arr = np.int16(rng.integers(low=0, high=16, size=shape)) elif dtypeList[idx] == DType.INT32: arr = np.int32(rng.integers(low=0, high=32, size=shape)) elif error_name == ErrorIf.WrongInputType: -- cgit v1.2.1