commit d69e2838b2e2f46401c5da19f662c9e0d5a5df06
author Jerry Ge <jerry.ge@arm.com> Wed Jul 05 21:54:07 2023 +0000
committer Eric Kunze <eric.kunze@arm.com> Thu Jul 06 21:19:46 2023 +0000
tree b8ff5f60fac9032da6dbca9ab766a7a7372ad4a5
parent e2325d12ba6eebeb59d50e7a7ce578a8a32a03ed

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 <jerry.ge@arm.com>
Change-Id: I5e63fb70b055cc5ee33e7dc26237eef9bedeac30

verif/frameworks/tosa_verif_framework_generator.py

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 @@
                     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,
                 )