Skip to content

[Unity][BYOC] Add conv2d and residual block patterns for Relax cutlass BYOC#14252

Merged
masahi merged 2 commits intoapache:unityfrom
yelite:cutlass-conv2d
Mar 9, 2023
Merged

[Unity][BYOC] Add conv2d and residual block patterns for Relax cutlass BYOC#14252
masahi merged 2 commits intoapache:unityfrom
yelite:cutlass-conv2d

Conversation

@yelite
Copy link
Contributor

@yelite yelite commented Mar 9, 2023

This PR adds cutlass BYOC pattern for conv2d and conv2d residual block.

In additional to the new patterns in the cutlass pattern table, this PR:

  1. Remove the assumption on fused function arg order from the conv2d codegen.
  2. Restructure how the cutlass pattern table is constructed, making it more readable.

The check logic at https://github.com/apache/tvm/blob/main/python/tvm/relay/op/contrib/cutlass.py#L200 isn't implemented in this PR yet because it requires non-trivial change to the interface of FuseOpsByPattern. I will send a separate PR to do this.

cc @vinx13 @masahi

@tvm-bot
Copy link
Collaborator

tvm-bot commented Mar 9, 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

@github-actions github-actions bot requested review from masahi and vinx13 March 9, 2023 18:05
@masahi masahi merged commit e56dc13 into apache:unity Mar 9, 2023
tqchen pushed a commit that referenced this pull request Mar 13, 2023
…s BYOC (#14252)

* Add conv2d and residual block pattern for cutlass

* Fix lint
tqchen pushed a commit that referenced this pull request Mar 13, 2023
…s BYOC (#14252)

* Add conv2d and residual block pattern for cutlass

* Fix lint
tqchen pushed a commit that referenced this pull request Mar 13, 2023
…s BYOC (#14252)

* Add conv2d and residual block pattern for cutlass

* Fix lint
tqchen pushed a commit that referenced this pull request Mar 13, 2023
…s BYOC (#14252)

* Add conv2d and residual block pattern for cutlass

* Fix lint
tqchen pushed a commit that referenced this pull request Mar 20, 2023
…s BYOC (#14252)

* Add conv2d and residual block pattern for cutlass

* Fix lint
tqchen pushed a commit that referenced this pull request Apr 1, 2023
…s BYOC (#14252)

* Add conv2d and residual block pattern for cutlass

* Fix lint
tqchen pushed a commit that referenced this pull request Apr 1, 2023
…s BYOC (#14252)

* Add conv2d and residual block pattern for cutlass

* Fix lint
tqchen pushed a commit that referenced this pull request Apr 1, 2023
…s BYOC (#14252)

* Add conv2d and residual block pattern for cutlass

* Fix lint
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.

4 participants