Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[CONV3D] remove incorrect schedule for conv3d_ndhwc #4659

Closed
wants to merge 2 commits into from
Closed

[CONV3D] remove incorrect schedule for conv3d_ndhwc #4659

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
e904a23
remove incorrect schedule for conv3d_ndhwc
optima2005 Jan 8, 2020
6c9345d
fix lint
optima2005 Jan 8, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
34 changes: 0 additions & 34 deletions topi/python/topi/cuda/conv3d.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -139,37 +139,3 @@ def _callback(op):

traverse_inline(s, outs[0].op, _callback)
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