Hacking (apache#35)

Repair layout issue
tmoreau89 · Jan 2, 2019 · 9a52a0d · 9a52a0d
1 parent 45b4d77
commit 9a52a0d
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 1 deletion.
diff --git a/src/relay/backend/interpreter.cc b/src/relay/backend/interpreter.cc
@@ -9,6 +9,7 @@
 #include <tvm/relay/interpreter.h>
 #include <tvm/relay/pass.h>
 #include <tvm/relay/attrs/debug.h>
+#include <tvm/relay/logging.h>
 #include "compile_engine.h"
 
 namespace tvm {
@@ -192,6 +193,7 @@ class Interpreter :
   }
 
   Value Eval(const Expr& expr) {
+    RELAY_LOG(INFO) << "Eval: expr_ptr=" << expr.get() << std::endl;
     return (*this)(expr);
   }
 

diff --git a/src/relay/op/nn/convolution.cc b/src/relay/op/nn/convolution.cc
@@ -62,6 +62,7 @@ bool Conv2DRel(const Array<Type>& types,
     channels = param->channels;
     dilated_ksize_y = 1 + (param->kernel_size[0] - 1) * param->dilation[0];
     dilated_ksize_x = 1 + (param->kernel_size[1] - 1) * param->dilation[1];
+
     // assign result to reporter
     reporter->Assign(types[1], TensorTypeNode::make(wshape, data->dtype));
   } else {

diff --git a/src/relay/op/tensor/transform.cc b/src/relay/op/tensor/transform.cc
@@ -1728,7 +1728,7 @@ Expr MakeBitPack(Expr data, int lanes) {
 
 TVM_REGISTER_API("relay.op._make.bitpack")
 .set_body([](const TVMArgs& args, TVMRetValue* rv) {
-  runtime::detail::unpack_call<Expr, 4>(MakeBitPack, args, rv);
+  runtime::detail::unpack_call<Expr, 2>(MakeBitPack, args, rv);
 });
 
 RELAY_REGISTER_OP("bitpack")

diff --git a/tests/python/relay/test_ir_nodes.py b/tests/python/relay/test_ir_nodes.py
@@ -209,6 +209,9 @@ def test_tuple_get_item():
     str(get)
     check_json_roundtrip(get)
 
+def test_op():
+    add = op.op.get("add")
+    check_json_roundtrip(add)
 
 def test_op():
     add = op.op.get("add")