[#65152] optimizers: tvm: Minify model JSON

Signed-off-by: Maciej Torhan <[email protected]>
antmicro · Sep 4, 2024 · 224e9d3 · 224e9d3
1 parent 630a23c
commit 224e9d3
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/kenning/optimizers/tvm.py b/kenning/optimizers/tvm.py
@@ -6,6 +6,7 @@
 Wrapper for TVM deep learning compiler.
 """
 
+import json
 import re
 from typing import Dict, List, Literal, Optional, Tuple, Union
 
@@ -658,7 +659,13 @@ def generator():
                 )
 
             if self.target_microtvm_board:
-                graph_json = lib.get_graph_json().encode()
+                graph_json = lib.get_graph_json()
+                # minify JSON
+                graph_json = json.dumps(
+                    json.loads(graph_json), separators=(",", ":")
+                )
+                graph_json = graph_json.encode()
+
                 params = tvm.runtime.params.save_param_dict(lib.get_params())
 
                 graph_data = b""