Skip to content
This repository has been archived by the owner on May 22, 2023. It is now read-only.

[PASS] Shape lowering #16

Merged
merged 10 commits into from
Sep 30, 2021
Merged

[PASS] Shape lowering #16

merged 10 commits into from
Sep 30, 2021

Conversation

ZihengJiang
Copy link
Contributor

Thanks for contributing to TVM! Please refer to guideline https://tvm.apache.org/docs/contribute/ for useful information and tips. After the pull request is submitted, please request code reviews from Reviewers by @ them in the pull request thread.

@ZihengJiang ZihengJiang marked this pull request as draft September 29, 2021 16:57
@ZihengJiang ZihengJiang marked this pull request as ready for review September 29, 2021 21:47
Copy link
Collaborator

@YuchenJin YuchenJin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM! Thanks @ZihengJiang!

include/tvm/relax/expr_functor.h Show resolved Hide resolved
tests/python/relax/test_transform.py Outdated Show resolved Hide resolved
@ZihengJiang ZihengJiang merged commit b60c0d7 into relax Sep 30, 2021
@ZihengJiang ZihengJiang deleted the ziheng branch September 30, 2021 00:17
tqchen pushed a commit that referenced this pull request Oct 30, 2021
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
YuchenJin pushed a commit that referenced this pull request Nov 10, 2021
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
YuchenJin pushed a commit that referenced this pull request Jan 27, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
YuchenJin pushed a commit that referenced this pull request Mar 2, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
yongwww pushed a commit to yongwww/relax that referenced this pull request Mar 14, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
yongwww pushed a commit to yongwww/relax that referenced this pull request Mar 22, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
yongwww pushed a commit to yongwww/relax that referenced this pull request Mar 23, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
MasterJH5574 added a commit to MasterJH5574/tlc-relax that referenced this pull request Apr 2, 2022
…im buffer (tlc-pack#16)

* Format

* Test for broadcast

* Test for element-wise

* Test for zero-rank element-wise operator

* Broadcast with dim-len 1 and index 0

* Element-wise with dim-len 1 and index 0

* Regard zero-dim broadcast as element-wise

* Rename buffer and use IntEnum

* Revert "Element-wise with dim-len 1 and index 0"

This reverts commit 6eb0309.

* Revert "Test for element-wise"

This reverts commit 189e8ae.

* Remove redundant `is_const_int`...
yongwww pushed a commit to yongwww/relax that referenced this pull request Apr 6, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
yongwww pushed a commit to yongwww/relax that referenced this pull request Apr 12, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
MasterJH5574 added a commit to MasterJH5574/tlc-relax that referenced this pull request Apr 13, 2022
…im buffer (tlc-pack#16)

* Format

* Test for broadcast

* Test for element-wise

* Test for zero-rank element-wise operator

* Broadcast with dim-len 1 and index 0

* Element-wise with dim-len 1 and index 0

* Regard zero-dim broadcast as element-wise

* Rename buffer and use IntEnum

* Revert "Element-wise with dim-len 1 and index 0"

This reverts commit 6eb0309.

* Revert "Test for element-wise"

This reverts commit 189e8ae.

* Remove redundant `is_const_int`...
jinhongyii pushed a commit to jinhongyii/relax that referenced this pull request May 7, 2022
…im buffer (tlc-pack#16)

* Format

* Test for broadcast

* Test for element-wise

* Test for zero-rank element-wise operator

* Broadcast with dim-len 1 and index 0

* Element-wise with dim-len 1 and index 0

* Regard zero-dim broadcast as element-wise

* Rename buffer and use IntEnum

* Revert "Element-wise with dim-len 1 and index 0"

This reverts commit 6eb0309.

* Revert "Test for element-wise"

This reverts commit 189e8ae.

* Remove redundant `is_const_int`...
yongwww pushed a commit that referenced this pull request May 10, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
yongwww pushed a commit that referenced this pull request May 11, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
YuchenJin pushed a commit that referenced this pull request Jun 1, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
jinhongyii pushed a commit to jinhongyii/relax that referenced this pull request Jun 5, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
yongwww pushed a commit to yongwww/relax that referenced this pull request Jun 12, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
yongwww pushed a commit to yongwww/relax that referenced this pull request Jul 15, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
yongwww pushed a commit that referenced this pull request Aug 14, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit that referenced this pull request Oct 14, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
YuchenJin pushed a commit that referenced this pull request Nov 18, 2022
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
MasterJH5574 pushed a commit to MasterJH5574/tlc-relax that referenced this pull request Nov 23, 2022
* Initial attempt

* rebase

* rebase

* profile

* profile

* ...

* rebase

* fix

* split cutlass

* upd

* refactor

* refactor

* refactor

* refactor

* refactor

* rebase

* ?

* rebase

* rebase

* rebase

* amp

* amp

* rebase

* rebase

* fp16 param wrapper

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: jinhongyi <[email protected]>
jinhongyii added a commit to jinhongyii/relax that referenced this pull request Dec 10, 2022
* Initial attempt

* rebase

* rebase

* profile

* profile

* ...

* rebase

* fix

* split cutlass

* upd

* refactor

* refactor

* refactor

* refactor

* refactor

* rebase

* ?

* rebase

* rebase

* rebase

* amp

* amp

* rebase

* rebase

* fp16 param wrapper

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: jinhongyi <[email protected]>
vinx13 pushed a commit to vinx13/relax that referenced this pull request Dec 14, 2022
* Initial attempt

* rebase

* rebase

* profile

* profile

* ...

* rebase

* fix

* split cutlass

* upd

* refactor

* refactor

* refactor

* refactor

* refactor

* rebase

* ?

* rebase

* rebase

* rebase

* amp

* amp

* rebase

* rebase

* fp16 param wrapper

Co-authored-by: Junru Shao <[email protected]>
Co-authored-by: jinhongyi <[email protected]>
YuchenJin pushed a commit that referenced this pull request Jan 13, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
YuchenJin pushed a commit that referenced this pull request Jan 14, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit to junrushao/relax that referenced this pull request Jan 25, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit to junrushao/relax that referenced this pull request Jan 26, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit to junrushao/relax that referenced this pull request Jan 29, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit to junrushao/relax that referenced this pull request Feb 5, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit to junrushao/relax that referenced this pull request Feb 6, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit that referenced this pull request Feb 8, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit that referenced this pull request Feb 8, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit that referenced this pull request Feb 8, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit that referenced this pull request Feb 8, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit that referenced this pull request Feb 9, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
junrushao pushed a commit that referenced this pull request Feb 9, 2023
* [PASS] Shape lowering.

* Update to IRModule based.

* TIR function generation.

* Improve.

* Improve.

* Improve test.

* Improve.

* Address comment.
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants