reverse changes in pr apache#4849 (apache#4853)

alexwong · Feb 26, 2020 · eb07246 · eb07246
1 parent 5fcb462
commit eb07246
Showing 1 changed file with 20 additions and 20 deletions.
diff --git a/topi/python/topi/intel_graphics/conv2d.py b/topi/python/topi/intel_graphics/conv2d.py
@@ -50,27 +50,27 @@ def _get_default_config(cfg, data, kernel, strides, padding, out_dtype, is_depth
             oc_bn = i
             break
 
-        if HSTR == 2:
-            if out_channel + hkernel == 515:
-                block_oh = 4
-                block_ow = 4
-            else:
-                block_oh = 4
-                block_ow = 5
-        elif hkernel == 3:
-            if out_channel == 512:
-                block_oh = 2
-                block_ow = 7
-            else:
-                block_oh = 2
-                block_ow = 14
+    if HSTR == 2:
+        if out_channel + hkernel == 515:
+            block_oh = 4
+            block_ow = 4
         else:
-            block_oh = 1
-            block_ow = 16
-        cfg["tile_ic"] = SplitEntity([in_channel // ic_bn, ic_bn])
-        cfg["tile_oc"] = SplitEntity([out_channel // oc_bn, oc_bn])
-        cfg["block_oh"] = OtherOptionEntity(block_oh)
-        cfg["block_ow"] = OtherOptionEntity(block_ow)
+            block_oh = 4
+            block_ow = 5
+    elif hkernel == 3:
+        if out_channel == 512:
+            block_oh = 2
+            block_ow = 7
+        else:
+            block_oh = 2
+            block_ow = 14
+    else:
+        block_oh = 1
+        block_ow = 16
+    cfg["tile_ic"] = SplitEntity([in_channel // ic_bn, ic_bn])
+    cfg["tile_oc"] = SplitEntity([out_channel // oc_bn, oc_bn])
+    cfg["block_oh"] = OtherOptionEntity(block_oh)
+    cfg["block_ow"] = OtherOptionEntity(block_ow)
 
 
 def _create_schedule_template(cfg, data, kernel, strides, padding, dilation, layout):