Commit 831b32e7 by mingwayzhang Committed by Thierry Moreau

Fix (2/2) [TOPI] conv2d schedule code (#3648) (#3717)

* Fix the tile_rx and tile_ry issue.

    Note that this patch depends on pull request #9 in tvm-distro.
parent e797a4e1
......@@ -39,7 +39,7 @@ PACKAGE_VERSION = {
'arm_cpu': "v0.04",
'llvm': "v0.03",
'cuda': "v0.04",
'cuda': "v0.05",
'rocm': "v0.02",
'opencl': "v0.02",
'mali': "v0.05",
......
......@@ -90,8 +90,8 @@ def schedule_direct_cuda(cfg, s, conv):
n, f, y, x = s[OL].op.axis
rc, ry, rx = s[OL].op.reduce_axis
rco, rci = cfg['tile_rc'].apply(s, OL, rc)
ryo, ryi = cfg['tile_rx'].apply(s, OL, ry)
rxo, rxi = cfg['tile_ry'].apply(s, OL, rx)
ryo, ryi = cfg['tile_ry'].apply(s, OL, ry)
rxo, rxi = cfg['tile_rx'].apply(s, OL, rx)
s[OL].reorder(rco, ryo, rxo, rci, ryi, rxi, n, f, y, x)
s[AA].compute_at(s[OL], rxo)
......
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