Skip to content

Commit a98b713

Browse files
jroeschtmoreau89
authored andcommitted
Hacking (apache#35)
Repair layout issue
1 parent d16b039 commit a98b713

File tree

3 files changed

+5
-0
lines changed

3 files changed

+5
-0
lines changed

src/relay/backend/interpreter.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
#include <tvm/relay/interpreter.h>
1111
#include <tvm/relay/pass.h>
1212
#include <tvm/relay/attrs/debug.h>
13+
#include <tvm/relay/logging.h>
1314
#include "compile_engine.h"
1415

1516
namespace tvm {

src/relay/op/nn/convolution.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@ bool Conv2DRel(const Array<Type>& types,
6464
channels = param->channels;
6565
dilated_ksize_y = 1 + (param->kernel_size[0] - 1) * param->dilation[0];
6666
dilated_ksize_x = 1 + (param->kernel_size[1] - 1) * param->dilation[1];
67+
6768
// assign result to reporter
6869
reporter->Assign(types[1], TensorTypeNode::make(wshape, data->dtype));
6970
} else {

tests/python/relay/test_ir_nodes.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,9 @@ def test_tuple_get_item():
209209
str(get)
210210
check_json_roundtrip(get)
211211

212+
def test_op():
213+
add = op.op.get("add")
214+
check_json_roundtrip(add)
212215

213216
def test_op():
214217
add = op.op.get("add")

0 commit comments

Comments
 (0)