Skip to content

Commit

Permalink
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
tests
Browse files Browse the repository at this point in the history
electriclilies committed Jan 6, 2022

Verified

This commit was created on GitHub.com and signed with GitHub’s verified signature. The key has expired.
1 parent 592cf52 commit 84510b3
Showing 1 changed file with 11 additions and 2 deletions.
13 changes: 11 additions & 2 deletions tests/python/relay/test_json_compact.py
Original file line number Diff line number Diff line change
@@ -49,7 +49,13 @@ def test_var():
{"type_key": ""},
{
"type_key": "relay.Var",
"attrs": {"_checked_type_": "0", "span": "0", "type_annotation": "0", "vid": "2", "virtual_device_": "0"},
"attrs": {
"_checked_type_": "0",
"span": "0",
"type_annotation": "0",
"vid": "2",
"virtual_device_": "0",
},
},
{"type_key": "relay.Id", "attrs": {"name_hint": "a3"}},
{"type_key": "relay.TensorType", "attrs": {"dtype": "float32", "shape": "4", "span": "0"}},
@@ -130,7 +136,10 @@ def test_global_var():
assert isinstance(tvar, tvm.ir.GlobalVar)
nodes = [
{"type_key": ""},
{"type_key": "GlobalVar", "attrs": {"_checked_type_": "0", "name_hint": "x", "span": "0", "virtual_device_": "0"}},
{
"type_key": "GlobalVar",
"attrs": {"_checked_type_": "0", "name_hint": "x", "span": "0", "virtual_device_": "0"},
},
]
data = {
"root": 1,

0 comments on commit 84510b3

Please sign in to comment.