Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[examples] Add Gemmini depthwise_conv examples #457

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

cmx-Y
Copy link
Contributor

@cmx-Y cmx-Y commented Feb 10, 2025

Two Gemmini depthwise convolution examples have been added.

In dw_conv_2d_nchw_chw_i8.mlir, the -convert-linalg-to-loops option is used to convert linalg.depthwise_conv_2d_nchw_chw into loops for computation. In the future, a new pass may be added to enable its computation to leverage gemmini.tile_conv.

In dw-tile-conv.mlir, the multiple channels of the input tensor are split, and gemmini.tile_conv is invoked multiple times to compute the depthwise convolution. The correctness of this approach has been verified, although its performance might be relatively low and could be improved later.

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.

1 participant