update the json compact function

apache · May 17, 2020 · 467a324 · 467a324
1 parent f327e25
commit 467a324
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/tvm/ir/json_compact.py b/python/tvm/ir/json_compact.py
@@ -129,8 +129,8 @@ def _convert(item, nodes):
         "relay.ModulePass": _rename("transform.ModulePass"),
         "relay.Sequential": _rename("transform.Sequential"),
         # TIR
-        "Variable": _update_tir_var("tir.Var"),
-        "SizeVar": _update_tir_var("tir.SizeVar"),
+        "Variable": [_update_tir_var("tir.Var"), _update_from_std_str("name_hint")],
+        "SizeVar": [_update_tir_var("tir.SizeVar"), _update_from_std_str("name_hint")],
     }
     return create_updater(node_map, "0.6", "0.7")