From 22a7af37740490975ebd4bee8fe1ae1a2dee75e6 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Thu, 16 May 2019 23:39:13 +0800 Subject: [PATCH] [TOPI] Raise exception group_conv2d_nchw not supported (#3195) --- topi/python/topi/cuda/group_conv2d_nchw.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/topi/python/topi/cuda/group_conv2d_nchw.py b/topi/python/topi/cuda/group_conv2d_nchw.py index be4ae3554e33..cbdb3cb8031d 100644 --- a/topi/python/topi/cuda/group_conv2d_nchw.py +++ b/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