From e904a23161568f23253464b9bb46edd57f5a5ebc Mon Sep 17 00:00:00 2001 From: optima2005 Date: Wed, 8 Jan 2020 20:12:02 +0800 Subject: [PATCH 1/2] remove incorrect schedule for conv3d_ndhwc --- topi/python/topi/cuda/conv3d.py | 32 -------------------------------- 1 file changed, 32 deletions(-) diff --git a/topi/python/topi/cuda/conv3d.py b/topi/python/topi/cuda/conv3d.py index 7d3c0b4afc1b..a0656ca72e8f 100644 --- a/topi/python/topi/cuda/conv3d.py +++ b/topi/python/topi/cuda/conv3d.py @@ -141,35 +141,3 @@ def _callback(op): return s -@autotvm.register_topi_schedule(generic.schedule_conv3d_ndhwc, ["cuda", "gpu"], - ["direct"]) -def schedule_conv3d_ndhwc_cuda(cfg, outs): - """TOPI schedule callback of conv3d for cuda gpu - - Parameters - ---------- - cfg: ConfigEntity - The config for this template - - outs: Array of Tensor - The computation graph description of conv2d - in the format of an array of tensors. - - Returns - ------- - s: Schedule - The computation schedule for conv2d. - """ - target = tvm.target.current_target() - if 'cudnn' in target.libs: - return generic.schedule_extern(outs) - - outs = [outs] if isinstance(outs, tvm.tensor.Tensor) else outs - s = tvm.create_schedule([x.op for x in outs]) - - def _callback(op): - if op.tag == 'conv3d_ndhwc': - schedule_direct_3d_cuda(cfg, s, op.output(0)) - - traverse_inline(s, outs[0].op, _callback) - return s From 6c9345d852a788372c93e40f87bb3d41e4bd9138 Mon Sep 17 00:00:00 2001 From: optima2005 Date: Wed, 8 Jan 2020 22:06:10 +0800 Subject: [PATCH 2/2] fix lint --- topi/python/topi/cuda/conv3d.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/topi/python/topi/cuda/conv3d.py b/topi/python/topi/cuda/conv3d.py index a0656ca72e8f..50b499ba79fb 100644 --- a/topi/python/topi/cuda/conv3d.py +++ b/topi/python/topi/cuda/conv3d.py @@ -139,5 +139,3 @@ def _callback(op): traverse_inline(s, outs[0].op, _callback) return s - -