Fixed AutoScheduler bug for NHWC case

apache · Mar 18, 2021 · 03f4458 · 03f4458
1 parent 9d9529b
commit 03f4458
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/python/tvm/relay/op/strategy/x86.py b/python/tvm/relay/op/strategy/x86.py
@@ -214,11 +214,11 @@ def conv2d_strategy_cpu(attrs, inputs, out_type, target):
             assert kernel_layout == "HWIO"
             if not is_auto_scheduler_enabled():
                 logger.warning("group_conv2d is not optimized for x86 with autotvm.")
-                strategy.add_implementation(
-                    wrap_compute_conv2d(topi.nn.group_conv2d_nhwc, has_groups=True),
-                    wrap_topi_schedule(topi.generic.schedule_group_conv2d_nhwc),
-                    name="group_conv2d_nhwc.generic",
-                )
+            strategy.add_implementation(
+                wrap_compute_conv2d(topi.nn.group_conv2d_nhwc, has_groups=True),
+                wrap_topi_schedule(topi.generic.schedule_group_conv2d_nhwc),
+                name="group_conv2d_nhwc.generic",
+            )
         else:
             raise RuntimeError("Unsupported group_conv2d layout {}".format(layout))
     return strategy