Frontend WIP (apache#47)

* incorporate changes * incorporate changes * incorporate changes * incorporate changes * incorporate changes * incorporate changes * parse keyword arguments * improves BoolLitNode pretty printing
jroesch · Aug 16, 2018 · c2f3d78 · c2f3d78
1 parent 6a92c76
commit c2f3d78
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/relay/python/relay/relay.py b/relay/python/relay/relay.py
@@ -141,8 +141,7 @@ def visit_Call(self, call_node):
         """Transform a Python call into a Relay call"""
         func = call_node.func
         args = call_node.args
-        # keywords = call_node.keywords
-
+        keywords = call_node.keywords
         if isinstance(func, ast.Attribute):
             if func.value.id == 'relay':
                 relay_func = IntrinsicId(func.attr)
@@ -151,13 +150,12 @@ def visit_Call(self, call_node):
                     "only supported namespace is relay right now") # improve me
         else:
             raise Exception("unsupported calls")
-
         # TODO(joshpoll): Handle args
         relay_args = [self.visit(arg) for arg in args]
 
-        # relay_kwds = {LocalId(kwd.arg): self.visit(kwd.value) for kwd in keywords}
+        relay_attrs = Attributes({LocalId(kwd.arg): self.visit(kwd.value) for kwd in keywords})
 
-        return Call(relay_func, relay_args)
+        return Call(relay_func, relay_args, relay_attrs)
 
     def visit_Assign(self, assign_node):
         targets = assign_node.targets

diff --git a/relay/src/relay/node.cc b/relay/src/relay/node.cc
@@ -65,7 +65,8 @@ TVM_REGISTER_API("nnvm.make.BoolLit")
 
 TVM_STATIC_IR_FUNCTOR_REGISTER(IRPrinter, vtable)
     .set_dispatch<BoolLitNode>([](const BoolLitNode *node, tvm::IRPrinter *p) {
-       p->stream << "BoolLitNode(" << node->value << ")";
+       std::string pybool = (node->value == 0) ? "False" : "True";
+       p->stream << "BoolLitNode(" << pybool << ")";
      });
 
 TensorLit TensorLitNode::make(tvm::Array<Expr> value) {