diff options
author | Jerry Ge <jerry.ge@arm.com> | 2023-07-05 21:54:07 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-07-06 21:19:46 +0000 |
commit | d69e2838b2e2f46401c5da19f662c9e0d5a5df06 (patch) | |
tree | b8ff5f60fac9032da6dbca9ab766a7a7372ad4a5 /verif/frameworks/tosa_verif_framework_generator.py | |
parent | e2325d12ba6eebeb59d50e7a7ce578a8a32a03ed (diff) | |
download | reference_model-d69e2838b2e2f46401c5da19f662c9e0d5a5df06.tar.gz |
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
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_generator.py')
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_generator.py | 6 |
1 files 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, ) |