[RELAY] Fix serialization of OpNode

apache · Dec 10, 2018 · ac6b3d4 · ac6b3d4
1 parent 0b64a9f
commit ac6b3d4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/relay/ir/op.cc b/src/relay/ir/op.cc
@@ -139,8 +139,8 @@ TVM_REGISTER_API("relay.op._Register")
   });
 
 NodePtr<Node> CreateOp(const std::string& name) {
-  auto op = Op::Get(name);
-  CHECK(!op.defined()) << "Cannot find op \'" << name << '\'';
+  const Op& op = Op::Get(name);
+  CHECK(op.defined()) << "Cannot find op \'" << name << '\'';
   return op.node_;
 }
 

diff --git a/tests/python/relay/test_ir_nodes.py b/tests/python/relay/test_ir_nodes.py
@@ -18,6 +18,11 @@ def test_bad_constructor():
         pass
 
 
+def test_op():
+    x = relay.op.op.get("add")
+    check_json_roundtrip(x)
+
+
 # Span
 def test_span():
     span = relay.Span(None, 1, 1)
@@ -210,6 +215,7 @@ def test_tuple_get_item():
 
 
 if __name__ == "__main__":
+    test_op()
     test_bad_constructor()
     test_span()
     test_tensor_type()