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

[EmitC] Add Conv2D to EmitC op conversion #2365

Draft
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

jserbedzijaTT
Copy link
Contributor

Ticket

Link to Github Issue

What's changed

Added Conv2D to EmitC op conversion using the new TTNN→EmitC op conversion infrastructure created by @azecevicTT.

@jserbedzijaTT jserbedzijaTT force-pushed the jserbedzija/conv2d_to_emitc_op_conversion branch from ca2c603 to 4b341e7 Compare March 5, 2025 11:49
@jserbedzijaTT jserbedzijaTT changed the title Add Conv2D to EmitC op conversion [EmitC] Add Conv2D to EmitC op conversion Mar 5, 2025
@jserbedzijaTT jserbedzijaTT force-pushed the jserbedzija/conv2d_to_emitc_op_conversion branch from 4b341e7 to 94fa96a Compare March 6, 2025 10:36
@azecevicTT azecevicTT force-pushed the jserbedzija/conv2d_to_emitc_op_conversion branch from 94fa96a to 4d9846f Compare March 6, 2025 11:27
@@ -681,6 +723,24 @@ class EmitCTTNNEmitter {
template <typename OpConversionPatternTy>
emitc::CallOpaqueOp replaceOp(OpConversionPatternTy &&opConversionPattern,
llvm::ArrayRef<mlir::Attribute> args) {
if constexpr (std::is_same_v<TTNNOp, tt::ttnn::Conv2dOp> ||
Copy link
Contributor

Choose a reason for hiding this comment

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

Add comment explaining why this special treatment.

@@ -0,0 +1,16 @@
// RUN: ttmlir-opt --ttir-to-ttnn-backend-pipeline="system-desc-path=%system_desc_path%" %s > %t.mlir
Copy link
Contributor

Choose a reason for hiding this comment

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

Let's put this into pooling dir instead of conv2d.

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.

3 participants