xmos-ai-tools 1.3.2.dev90__py3-none-macosx_10_15_universal2.whl → 1.3.2.dev108__py3-none-macosx_10_15_universal2.whl

xmos_ai_tools/xformer/__init__.py

@@ -5,7 +5,8 @@ from typing import Union, List, Optional
 from .flash import generate_flash
 import re
 
-__compilation_output = ""
+__compilation_stdout = ""
+__compilation_stderr = ""
 __arena_size = 0
 
 
@@ -29,20 +30,22 @@ def convert(
 
     args.append(str(filename))
 
-    process_call: subprocess.CompletedProcess = subprocess.run(
-        [arg for arg in args],
-        stdout=subprocess.PIPE,
-        stderr=subprocess.STDOUT,
-        check=True,
-    )
-
-    global __compilation_output, __arena_size
-    __compilation_output = process_call.stdout.decode("utf-8")
-    size_str = re.sub("((.|\n|\r)*)Tensor arena size :", "", __compilation_output)
-    size_str = re.sub("(\n|\r)((.|\n|\r)*)", "", size_str)
-    __arena_size = int(size_str.strip())
-
-    return process_call.returncode
+    try:
+        process_call: subprocess.CompletedProcess = subprocess.run(
+            [arg for arg in args],
+            check=True, text=True, capture_output=True,
+        )
+        global __compilation_stdout, __compilation_stderr, __arena_size
+        __compilation_stdout = process_call.stdout
+        __compilation_stderr = process_call.stderr
+        size_str = re.sub("((.|\n|\r)*)Tensor arena size :", "", __compilation_stdout)
+        size_str = re.sub("(\n|\r)((.|\n|\r)*)", "", size_str)
+        __arena_size = int(size_str.strip())
+        return process_call.returncode
+    except subprocess.CalledProcessError as e:
+        print(e)
+        print("Return code:", e.returncode)
+        print("Error output:", e.stderr)
 
 
 def tensor_arena_size() -> int:
@@ -50,7 +53,8 @@ def tensor_arena_size() -> int:
 
 
 def print_optimization_report():
-    print(__compilation_output)
+    print(__compilation_stderr)
+    print(__compilation_stdout)
 
 
 def print_help(show_hidden: Optional[bool] = False) -> int:

xmos_ai_tools/xinterpreters/host_interpreter.py

@@ -242,7 +242,6 @@ class TFLMHostInterpreter:
         if self.obj:
             lib.delete_interpreter(self.obj)
             self.obj = None
-            print(self.obj)
 
     def tensor_arena_size(self) -> int:
         """! Read the size of the tensor arena required.

{xmos_ai_tools-1.3.2.dev90.dist-info → xmos_ai_tools-1.3.2.dev108.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: xmos_ai_tools
-Version: 1.3.2.dev90
+Version: 1.3.2.dev108
 Summary: XMOS AI Tools
 Home-page: https://github.com/xmos/ai_tools
 Author: XMOS
@@ -30,4 +30,4 @@ Requires-Dist: tflite>=2.4.0
 Documentation
 -------------
 
-Click [here](https://github.com/xmos/ai_tools/blob/ba134f5c7797a561414d85cbaa98648f7dee8b9f/README.md) for documentation on using xmos-ai-tools to deploy AI models on xcore.
+Click [here](https://github.com/xmos/ai_tools/blob/ded91c91e745c004e545e25b7d5f1b5cd6b966dd/README.md) for documentation on using xmos-ai-tools to deploy AI models on xcore.

{xmos_ai_tools-1.3.2.dev90.dist-info → xmos_ai_tools-1.3.2.dev108.dist-info}/RECORD

@@ -378,17 +378,17 @@ xmos_ai_tools/runtime/include/tensorflow/lite/micro/tools/benchmarking/metrics.h
 xmos_ai_tools/runtime/include/tensorflow/lite/micro/tools/benchmarking/op_resolver.h,sha256=g0dl9tzUqngiINvjBlqDclFqvkC85MC4kbU13vE4OkY,6071
 xmos_ai_tools/runtime/include/tensorflow/lite/schema/schema_generated.h,sha256=OzGCvswrW_FArm-HxD70BExn0PVtBbHAFPDO6ZZOFtc,1093177
 xmos_ai_tools/runtime/include/tensorflow/lite/schema/schema_utils.h,sha256=tkHMDPARjIqppYCVInIowwdHxjNP3pfSS9O7vx-ODeo,1333
-xmos_ai_tools/runtime/lib/libhost_xtflitemicro.a,sha256=02QNMG8z6Gc6O1hi6WhPoyxNaU7uX278J95a1fSLqUU,2431256
-xmos_ai_tools/runtime/lib/libxtflitemicro.a,sha256=mU05QFt_TIbBrNiqmLw3_dpMXQvQwFfBAnxeoB70Vn8,70440318
-xmos_ai_tools/xformer/__init__.py,sha256=jA0xba6ZitK9HwWvDJewM8AMU4IZnx_iB1GSBrRUGvU,1627
+xmos_ai_tools/runtime/lib/libhost_xtflitemicro.a,sha256=sgtO2Cb3dJ3Dps2c0Ly3ZQjkzmfALm6GdyQcTo53w4M,2431256
+xmos_ai_tools/runtime/lib/libxtflitemicro.a,sha256=lvrVMYJaYvmS9p5q7gM4NExtleEPkKO-563f4Xia8G8,70440318
+xmos_ai_tools/xformer/__init__.py,sha256=AxzDorvqJ0FiTc4CLecHPwbMkSuWzXGZT2-u72Aau5s,1904
 xmos_ai_tools/xformer/flash.py,sha256=MG4coi_Lvvg-oQmw1pomJD8eeOH4gAMjixjBFvO2BCk,6376
 xmos_ai_tools/xinterpreters/__init__.py,sha256=PFRB9VxOLKaA--j2ZvWGcmesv2C6uNYqJ_kBam68aUI,50
 xmos_ai_tools/xinterpreters/exceptions.py,sha256=HOjADxHYMPI9mN0YIbWxtw9hSeL2B6XWWwqtGtyJdVs,577
-xmos_ai_tools/xinterpreters/host_interpreter.py,sha256=No-g29_twbIaJ4S4fzU5TGpKd4WPVEE0737HtL877sw,25692
+xmos_ai_tools/xinterpreters/host_interpreter.py,sha256=rqoTcSWK7O9eboMsg0o2UYWapWTE5xiki7XTJZ_JbwQ,25664
 xmos_ai_tools/xinterpreters/libs/macos/xtflm_python.1.0.1.dylib,sha256=uwA1dNlFvRyw-lyg7msQ2P_vi6syL4OGWWZj8pa3fWE,2321592
 xmos_ai_tools/xinterpreters/libs/macos/xtflm_python.dylib,sha256=uwA1dNlFvRyw-lyg7msQ2P_vi6syL4OGWWZj8pa3fWE,2321592
-xmos_ai_tools-1.3.2.dev90.data/data/bin/xcore-opt,sha256=s02tefnJ7Sm7TdbJPz0qB8SK6052cMXq2uL4uwB0BJE,283234872
-xmos_ai_tools-1.3.2.dev90.dist-info/METADATA,sha256=84FC4gHUEBTSb51pvaDCXRvn0z5zY717Q_nj9WrSQ8Q,1341
-xmos_ai_tools-1.3.2.dev90.dist-info/WHEEL,sha256=cblCMhPsi-5aS-1Fj5LRCkSbCUQwa2nrUvkiKN0vRpQ,113
-xmos_ai_tools-1.3.2.dev90.dist-info/top_level.txt,sha256=YWegea73ll3tMlRWRdHJemUy2VOuEYDdOIaffxu_eF0,14
-xmos_ai_tools-1.3.2.dev90.dist-info/RECORD,,
+xmos_ai_tools-1.3.2.dev108.data/data/bin/xcore-opt,sha256=lJwRhCr3bwJVK_KXrVEDVJJ0YWt44SMkiSMyOqO4EBE,283342392
+xmos_ai_tools-1.3.2.dev108.dist-info/METADATA,sha256=VGPXQn2iVk0xtn3nQX4zcb5LNLYZrX-aWpt72YRRerc,1342
+xmos_ai_tools-1.3.2.dev108.dist-info/WHEEL,sha256=cblCMhPsi-5aS-1Fj5LRCkSbCUQwa2nrUvkiKN0vRpQ,113
+xmos_ai_tools-1.3.2.dev108.dist-info/top_level.txt,sha256=YWegea73ll3tMlRWRdHJemUy2VOuEYDdOIaffxu_eF0,14
+xmos_ai_tools-1.3.2.dev108.dist-info/RECORD,,