From d69e2838b2e2f46401c5da19f662c9e0d5a5df06 Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Wed, 5 Jul 2023 21:54:07 +0000 Subject: Enable framework generator to save numpy files in native dtypes With the following patch merged, the tosa serialization_lib is able to read u/int8, u/int16 in their native dtypes: https://review.mlplatform.org/c/tosa/serialization_lib/+/9871 Signed-off-by: Jerry Ge Change-Id: I5e63fb70b055cc5ee33e7dc26237eef9bedeac30 --- verif/frameworks/tosa_verif_framework_generator.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/verif/frameworks/tosa_verif_framework_generator.py b/verif/frameworks/tosa_verif_framework_generator.py index 12fff68..94a1a15 100755 --- a/verif/frameworks/tosa_verif_framework_generator.py +++ b/verif/frameworks/tosa_verif_framework_generator.py @@ -1130,17 +1130,15 @@ def run_unit_test( name="gen_quant_npy", ) - quant_val = np.round(fakequant_val / scale).astype(np.int32) + zeropoint + quant_val = np.round(fakequant_val / scale) + zeropoint # very few unit tests after TF hash may/2020, this quantized # value for some reason exceed [0, 255] range saved_val = np.clip(quant_val, qmin, qmax).astype(numpy_dtype[idx]) - # saved all quantized tensor as np.int32 - # since TOSA numpy Cpp API only supports int32 np.save( os.path.join(test_dir, placeholder_npy_filenames[idx]), - saved_val.astype(np.int32), + saved_val, False, ) -- cgit v1.2.1