Hacking (apache#35)

Repair layout issue
tmoreau89 · Mar 21, 2019 · a98b713 · a98b713
1 parent d16b039
commit a98b713
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 0 deletions.
diff --git a/src/relay/backend/interpreter.cc b/src/relay/backend/interpreter.cc
@@ -10,6 +10,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 {

diff --git a/src/relay/op/nn/convolution.cc b/src/relay/op/nn/convolution.cc
@@ -64,6 +64,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/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")