Skip to content

T
tic
Commit d364f358 by Wuwei Lin Committed by Tianqi Chen
Options

[TOPI] Raise exception group_conv2d_nchw not supported (#3195)

parent c4439a80
Showing with 2 additions and 0 deletions
topi/python/topi/cuda/group_conv2d_nchw.py
......@@ -346,6 +346,8 @@ def schedule_conv2d_nchw_cuda(cfg, outs):
def _callback(op):
if op.tag == "group_conv2d_NCHWc_int8":
schedule_group_conv2d_NCHWc_int8(cfg, s, op.output(0))
if op.tag == "group_conv2d_nchw":
raise tvm.error.OpNotImplemented("group_conv2d_nchw not supported")
traverse_inline(s, outs[0].op, _callback)
return s
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment