Tianqi Chen
302aaf9f96
Rename the reflected local `Var` field from `name_hint` to `name` and update its typed C++ consumers. Preserve distinct named-node APIs and the Python constructor keyword compatibility path, while making `.name` the sole stored Var property. Upgrade legacy compact JSON records for current and pre-unification Var schemas. Validation: full runtime/compiler build, focused C++ Var copy-helper test, focused Python IR/Relax/TIRx/script tests, Vulkan codegen syntax build, touched-file pre-commit checks, and `git diff --check`.
102 lines
3.1 KiB
Python
102 lines
3.1 KiB
Python
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
"""Tool to upgrade json from historical versions."""
|
|
|
|
import json
|
|
|
|
|
|
def get_version(jgraph):
|
|
"""
|
|
Get the tvm version from the json graph.
|
|
|
|
Parameters
|
|
----------
|
|
jgraph : dict
|
|
The json graph.
|
|
"""
|
|
return jgraph["metadata"]["tvm_version"]
|
|
|
|
|
|
def create_updater(node_map, from_ver, to_ver):
|
|
"""Create an updater to update json loaded data.
|
|
|
|
Parameters
|
|
----------
|
|
node_map : Map[str, Function]
|
|
Map from type_key to updating function
|
|
|
|
from_ver : str
|
|
Prefix of version that we can accept,
|
|
|
|
to_ver : str
|
|
The target version.
|
|
|
|
Returns
|
|
-------
|
|
fupdater : function
|
|
The updater function
|
|
"""
|
|
|
|
def _updater(data):
|
|
assert get_version(data).startswith(from_ver)
|
|
nodes = data["nodes"]
|
|
for idx, item in enumerate(nodes):
|
|
f = node_map.get(item["type"], None)
|
|
if isinstance(f, list):
|
|
for fpass in f:
|
|
item = fpass(item, nodes)
|
|
elif f:
|
|
item = f(item, nodes)
|
|
nodes[idx] = item
|
|
data["metadata"]["tvm_version"] = to_ver
|
|
return data
|
|
|
|
return _updater
|
|
|
|
|
|
def upgrade_json(json_str):
|
|
"""Update json from a historical version.
|
|
|
|
Parameters
|
|
----------
|
|
json_str : str
|
|
A historical json file.
|
|
|
|
Returns
|
|
-------
|
|
updated_json : str
|
|
The updated version.
|
|
"""
|
|
data = json.loads(json_str)
|
|
if "metadata" not in data and "attrs" in data:
|
|
raise ValueError("Legacy json graph format detected, we don't support it anymore.")
|
|
|
|
# `ir.Var` is the sole runtime variable node. Keep `tvm.ir.load_json`
|
|
# compatible with the pre-unification Relax/TIRx schemas and with graphs
|
|
# written before the canonical Var field was renamed to `name`. Rewriting
|
|
# nodes in place preserves node indices and shared references.
|
|
for node in data.get("nodes", []):
|
|
if node.get("type") == "relax.expr.Var":
|
|
node["type"] = "ir.Var"
|
|
elif node.get("type") == "tirx.Var":
|
|
node["type"] = "ir.Var"
|
|
if node.get("type") in ("ir.Var", "relax.expr.DataflowVar"):
|
|
fields = node.get("data", {})
|
|
if "name_hint" in fields and "name" not in fields:
|
|
fields["name"] = fields.pop("name_hint")
|
|
return json.dumps(data, indent=2)
|