Fix

apache · Jun 3, 2019 · 0a6a274 · 0a6a274
1 parent 395408f
commit 0a6a274
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/topi/python/topi/x86/conv2d.py b/topi/python/topi/x86/conv2d.py
@@ -419,13 +419,14 @@ def _alter_conv2d_layout(attrs, inputs, tinfo, F):
     # only optimize for NCHW
     if layout != 'NCHW':
         return None
-    if groups != 1 and not is_depthwise:
-        return None
 
     assert attrs["kernel_layout"] == "OIHW"
     kshape = get_const_tuple(kernel.shape)
     is_depthwise = groups == kshape[0] and kshape[1] == 1
 
+    if groups != 1 and not is_depthwise:
+        return None
+
     dispatch_ctx = autotvm.task.DispatchContext.current
     target = tvm.target.current_target()
     # query schedule and fallback if necessary