Fix an OrderDict initilization bug.

The dict which is used to initilize OrderDict is not ordered, so metadata may not be at the end.
apache · Mar 20, 2019 · 34802fa · 34802fa
1 parent 89acfeb
commit 34802fa
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/python/tvm/relay/backend/graph_runtime_codegen.py b/python/tvm/relay/backend/graph_runtime_codegen.py
@@ -365,14 +365,14 @@ def nested_defaultdict():
             metadata['signatures']['default']['outputs'][node_name]['shape'] = shapes[node_id[0]]
 
         # Keep  'metadata' always at end
-        json_dict = OrderedDict({
-            "nodes": nodes,
-            "arg_nodes": arg_nodes,
-            "heads": heads,
-            "attrs": attrs,
-            "node_row_ptr":  node_row_ptr,
-            "metadata": metadata
-        })
+        json_dict = OrderedDict([
+            ("nodes", nodes),
+            ("arg_nodes", arg_nodes),
+            ("heads", heads),
+            ("attrs", attrs),
+            ("node_row_ptr", node_row_ptr),
+            ("metadata", metadata),
+        ])
 
         return json.dumps(json_dict, indent=2)