c3511c5e2c2f903606209c9eb6d56c2221570a24,tests/python/unittest/test_tir_transform_storage_rewrite.py,,test_inplace_rule2,#Any#Any#,291

Before Change


    Ab = tvm.tir.decl_buffer(A.shape, A.dtype, name="A")
    Bb = tvm.tir.decl_buffer(B.shape, B.dtype, name="B")
    Cc = tvm.tir.decl_buffer(C.shape, B.dtype, name="C")
    Dd = tvm.tir.decl_buffer(D.shape, B.dtype, name="D")
    stmt = tvm.tir.ir_pass.StorageFlatten(stmt, {A: Ab, B: Bb, C: Cc, D:Dd}, 64)

    mod = tvm.IRModule.from_expr(tvm.tir.PrimFunc([Ab, Bb, Cc, Dd], stmt))
    mod = tvm.tir.transform.Simplify()(mod)
    mod = tvm.tir.transform.StorageRewrite()(mod)

After Change


    assert isinstance(bounds, tvm.container.Map)
    stmt = tvm.te.schedule.ScheduleOps(s, bounds)

    func = tvm.te.schedule.SchedulePostProcToPrimFunc([A, B, C, D], stmt, None)
    mod = tvm.IRModule.from_expr(func)
    mod = tvm.tir.transform.StorageFlatten(64)(mod)
Italian Trulli
In pattern: SUPERPATTERN

Frequency: 3

Non-data size: 6

Instances


Project Name: apache/incubator-tvm
Commit Name: c3511c5e2c2f903606209c9eb6d56c2221570a24
Time: 2020-04-19
Author: tqchen@users.noreply.github.com
File Name: tests/python/unittest/test_tir_transform_storage_rewrite.py
Class Name:
Method Name: test_inplace_rule2


Project Name: apache/incubator-tvm
Commit Name: c3511c5e2c2f903606209c9eb6d56c2221570a24
Time: 2020-04-19
Author: tqchen@users.noreply.github.com
File Name: tests/python/unittest/test_tir_transform_storage_rewrite.py
Class Name:
Method Name: test_storage_share_gpu


Project Name: apache/incubator-tvm
Commit Name: 82942fb33fd6e3572897c815af16905c4f75c2a4
Time: 2020-12-21
Author: masahi129@gmail.com
File Name: python/tvm/topi/cuda/nms.py
Class Name:
Method Name: non_max_suppression