From 33d837e3235d709596cd10d43db18399281b436a Mon Sep 17 00:00:00 2001 From: Les Bell Date: Tue, 10 Aug 2021 08:34:43 +0100 Subject: fix output type for resize in test gen Signed-off-by: Les Bell Change-Id: I9daf8fd691af74500819718519caac30c966a1f2 --- verif/tosa_test_gen.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py index c458538..99dc5f8 100644 --- a/verif/tosa_test_gen.py +++ b/verif/tosa_test_gen.py @@ -733,11 +733,11 @@ class TosaArgGen: # Exclude illegal {mode, type} configurations. Pick legal output types if m == ResizeMode.NEAREST and dtype == DType.INT8: - outputDTypeList = [DType.INT32] + outputDTypeList = [DType.INT8] elif m == ResizeMode.NEAREST and dtype == DType.INT16: outputDTypeList = [DType.INT16] elif m == ResizeMode.BILINEAR and dtype == DType.INT8: - outputDTypeList = [DType.INT8] + outputDTypeList = [DType.INT32] elif m == ResizeMode.BILINEAR and dtype == DType.INT16: outputDTypeList = [DType.INT48] elif dtype == DType.FLOAT: @@ -770,7 +770,7 @@ class TosaArgGen: arg_list.append( ( "mode{}_odim{}x{}_out{}_st{:.2f}x{:.2f}_off{:.2f}x{:.2f}".format( - m, + "N" if m == ResizeMode.NEAREST else "B", output_dims[0], output_dims[1], testGen.typeStr(outputDType), @@ -824,7 +824,7 @@ class TosaArgGen: arg_list.append( ( "mode{}_shift{}_odim{}x{}_out{}_st{}x{}_off{}x{}".format( - m, + "N" if m == ResizeMode.NEAREST else "B", shift, output_dims[0], output_dims[1], -- cgit v1.2.1