From e3d06c949c5952f6eaa7a3084b235dcc03b6662f Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 31 Mar 2020 22:10:24 +0100
Subject: [PATCH] [Util] add prefix option to pack_innermost_dim_as_hex_str
---
src/finn/util/data_packing.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/src/finn/util/data_packing.py b/src/finn/util/data_packing.py
index 1c08ac4fc..6c810c226 100644
--- a/src/finn/util/data_packing.py
+++ b/src/finn/util/data_packing.py
@@ -119,7 +119,9 @@ def npbytearray2hexstring(npbytearray, prefix="0x"):
return prefix + binascii.hexlify(bytearray(npbytearray)).decode("utf-8")
-def pack_innermost_dim_as_hex_string(ndarray, dtype, pad_to_nbits, reverse_inner=False):
+def pack_innermost_dim_as_hex_string(
+ ndarray, dtype, pad_to_nbits, reverse_inner=False, prefix="0x"
+):
"""Pack the innermost dimension of the given numpy ndarray into hex
strings using array2hexstring.
@@ -143,7 +145,9 @@ def pack_innermost_dim_as_hex_string(ndarray, dtype, pad_to_nbits, reverse_inner
ndarray = np.asarray(ndarray, dtype=np.float32)
def fun(x):
- return array2hexstring(x, dtype, pad_to_nbits, reverse=reverse_inner)
+ return array2hexstring(
+ x, dtype, pad_to_nbits, reverse=reverse_inner, prefix=prefix
+ )
return np.apply_along_axis(fun, ndarray.ndim - 1, ndarray)
--
GitLab