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.

Sign up for GitHub

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

Jump to bottom

Add workaround to assure that the weights of the convolution op are on the host #1528

Open
ajakovljevicTT opened this issue Dec 6, 2024 · 2 comments
Open

Add workaround to assure that the weights of the convolution op are on the host #1528

ajakovljevicTT opened this issue Dec 6, 2024 · 2 comments
Assignees
@sdjordjevicTT

Comments

@ajakovljevicTT
Copy link
Contributor

Currently, the ttnn conv2d operation requires that the weights of the convolution must be on host. This logic is currently added to the lib/Dialect/TTNN/Transforms/TTNNLayout.cpp, but probably should be moved to the newly made workaround pass.
@ajakovljevicTT
Copy link
Contributor Author

FYI: @sdjordjevicTT

@ajakovljevicTT ajakovljevicTT mentioned this issue Dec 6, 2024
Merged
@sdjordjevicTT sdjordjevicTT self-assigned this Dec 9, 2024
@sdjordjevicTT
Copy link
Contributor

Thanks for pointing this out, I will assign this one to myself and pick it up!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@sdjordjevicTT sdjordjevicTT

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@sdjordjevicTT @ajakovljevicTT