Skip to content

T
tic
Unverified Commit b7364b42 by Leyuan Wang Committed by GitHub
Options

reverse changes in pr #4849 (#4853)

parent 0dbe70c1
Showing with 20 additions and 20 deletions
topi/python/topi/intel_graphics/conv2d.py
...@@ -50,27 +50,27 @@ def _get_default_config(cfg, data, kernel, strides, padding, out_dtype, is_depth ...@@ -50,27 +50,27 @@ def _get_default_config(cfg, data, kernel, strides, padding, out_dtype, is_depth
oc_bn = i oc_bn = i
break break
if HSTR == 2: if HSTR == 2:
if out_channel + hkernel == 515: if out_channel + hkernel == 515:
block_oh = 4 block_oh = 4
block_ow = 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
else: else:
block_oh = 1 block_oh = 4
block_ow = 16 block_ow = 5
cfg["tile_ic"] = SplitEntity([in_channel // ic_bn, ic_bn]) elif hkernel == 3:
cfg["tile_oc"] = SplitEntity([out_channel // oc_bn, oc_bn]) if out_channel == 512:
cfg["block_oh"] = OtherOptionEntity(block_oh) block_oh = 2
cfg["block_ow"] = OtherOptionEntity(block_ow) 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): def _create_schedule_template(cfg, data, kernel, strides, padding, dilation, layout):
......
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