Skip to content

[Unity][Transform] SplitCallTIRByPattern and CUTLASS backend#14274

Merged
junrushao merged 2 commits intoapache:unityfrom
spectrometerHBH:split-call-tir
Mar 24, 2023
Merged

[Unity][Transform] SplitCallTIRByPattern and CUTLASS backend#14274
junrushao merged 2 commits intoapache:unityfrom
spectrometerHBH:split-call-tir

Conversation

@spectrometerHBH
Copy link
Copy Markdown
Contributor

@spectrometerHBH spectrometerHBH commented Mar 11, 2023

Currently, the BYOC system is based on op-level pattern matching, this PR intends to provide primary support for TIR-level pattern matching based on backend registration and dispatching.

For now, it simply matches the first set of for loops in PrimFunc.

Co-authored-by: Hongyi Jin (@jinhongyii)

@tvm-bot
Copy link
Copy Markdown
Collaborator

tvm-bot commented Mar 11, 2023

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

Generated by tvm-bot

@spectrometerHBH spectrometerHBH marked this pull request as ready for review March 20, 2023 21:43
Comment thread tests/python/relax/test_codegen_tir_cutlass.py Outdated
Comment thread python/tvm/relax/transform/transform.py
Copy link
Copy Markdown
Contributor

@jinhongyii jinhongyii left a comment

Choose a reason for hiding this comment

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

LGTM!

Comment thread python/tvm/contrib/cutlass/gemm_operation.py Outdated
@junrushao
Copy link
Copy Markdown
Member

CC @yongwww please take another look :-)

Comment thread python/tvm/contrib/cc.py Outdated
Comment thread python/tvm/contrib/cc.py Outdated
Comment thread tests/python/relax/test_codegen_tir_cutlass.py Outdated
Comment thread python/tvm/_ffi/libinfo.py Outdated
Comment thread python/tvm/contrib/cc.py Outdated
Comment thread python/tvm/contrib/cutlass/gemm_operation.py Outdated
@vinx13
Copy link
Copy Markdown
Member

vinx13 commented Mar 22, 2023

cc @masahi

Comment thread tests/python/relax/test_codegen_tir_cutlass.py Outdated
@spectrometerHBH
Copy link
Copy Markdown
Contributor Author

cc @vinx13 @masahi

@junrushao junrushao merged commit 517d045 into apache:unity Mar 24, 2023
farshidsp pushed a commit to farshidsp/tvm that referenced this pull request Mar 27, 2023
…14274)

Currently, the BYOC system is based on op-level pattern matching, this PR intends to provide primary support for TIR-level pattern matching based on backend registration and dispatching.

For now, it simply matches the first set of for loops in PrimFunc.

Co-authored-by: Hongyi Jin (@jinhongyii)
tqchen pushed a commit that referenced this pull request Apr 1, 2023
Currently, the BYOC system is based on op-level pattern matching, this PR intends to provide primary support for TIR-level pattern matching based on backend registration and dispatching.

For now, it simply matches the first set of for loops in PrimFunc.

Co-authored-by: Hongyi Jin (@jinhongyii)
tqchen pushed a commit that referenced this pull request Apr 1, 2023
Currently, the BYOC system is based on op-level pattern matching, this PR intends to provide primary support for TIR-level pattern matching based on backend registration and dispatching.

For now, it simply matches the first set of for loops in PrimFunc.

Co-authored-by: Hongyi Jin (@jinhongyii)
tqchen pushed a commit that referenced this pull request Apr 1, 2023
Currently, the BYOC system is based on op-level pattern matching, this PR intends to provide primary support for TIR-level pattern matching based on backend registration and dispatching.

For now, it simply matches the first set of for loops in PrimFunc.

Co-authored-by: Hongyi Jin (@jinhongyii)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

9 participants