From e4f40d9cd4167b6781ea3e9b8e57ec43469b5af6 Mon Sep 17 00:00:00 2001 From: Jared Roesch Date: Thu, 6 Dec 2018 22:12:50 -0800 Subject: [PATCH] Hacking (#35) Repair layout issue --- src/relay/backend/interpreter.cc | 1 + src/relay/op/nn/convolution.cc | 1 + tests/python/relay/test_ir_nodes.py | 3 +++ 3 files changed, 5 insertions(+) diff --git a/src/relay/backend/interpreter.cc b/src/relay/backend/interpreter.cc index 3128d2a71159..d464f099355f 100644 --- a/src/relay/backend/interpreter.cc +++ b/src/relay/backend/interpreter.cc @@ -10,6 +10,7 @@ #include #include #include +#include #include "compile_engine.h" namespace tvm { diff --git a/src/relay/op/nn/convolution.cc b/src/relay/op/nn/convolution.cc index b53f57464e81..9533ad69ed6d 100644 --- a/src/relay/op/nn/convolution.cc +++ b/src/relay/op/nn/convolution.cc @@ -64,6 +64,7 @@ bool Conv2DRel(const Array& 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 index e041acca0e0c..c797ee0b4340 100644 --- 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")