From ef386809c81e60a5b60d6757be90d964aa8032d3 Mon Sep 17 00:00:00 2001 From: Zhao Wu Date: Tue, 26 Mar 2019 19:17:47 +0800 Subject: [PATCH] [Relay] Fix Relay ARM CPU depthwise spatial pack schedule alter op layout issue. (#2861) * Fix Relay ARM CPU spatial pack depthwise alter op layout issue. * Update tune_relay_arm.py --- python/tvm/relay/op/nn/_nn.py | 1 - tutorials/autotvm/tune_relay_arm.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/python/tvm/relay/op/nn/_nn.py b/python/tvm/relay/op/nn/_nn.py index 958aabd2d45a..da58e9386c0f 100644 --- a/python/tvm/relay/op/nn/_nn.py +++ b/python/tvm/relay/op/nn/_nn.py @@ -85,7 +85,6 @@ def compute_conv2d(attrs, inputs, out_type, target): inputs[0], inputs[1], strides, padding, dilation, layout, out_dtype=out_dtype) elif layout == "NCHW" and \ - kernel_layout == "OIHW" and \ get_const_int(inputs[1].shape[0]) == groups and \ get_const_int(inputs[1].shape[1]) == 1: out = topi.nn.depthwise_conv2d_nchw( diff --git a/tutorials/autotvm/tune_relay_arm.py b/tutorials/autotvm/tune_relay_arm.py index 266a1532a8b6..b56cb1f1dd42 100644 --- a/tutorials/autotvm/tune_relay_arm.py +++ b/tutorials/autotvm/tune_relay_arm.py @@ -186,7 +186,7 @@ def get_network(name, batch_size): 'log_filename': log_file, 'tuner': 'xgb', - 'n_trial': 1000, + 'n_trial': 1500, 'early_stopping': 800, 'measure_option': autotvm.measure_option(