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

Support destructuring tuples in cube macro #66

Merged
merged 1 commit into from
Aug 12, 2024

Conversation

cBournhonesque
Copy link
Contributor

I'm trying to convert the reduce kernels to use the #[cube] macro: tracel-ai/burn#2117

One problem I'm running into is that the accumulator for some of the kernels is a tuple, and tuple destructuring doesn't seem to be supported currently.

This PR adds tuple destructuring, with a unit test

Copy link
Member

@nathanielsimard nathanielsimard left a comment

Choose a reason for hiding this comment

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

LGTM 🔥

@nathanielsimard nathanielsimard merged commit 4e17724 into tracel-ai:main Aug 12, 2024
1 of 2 checks passed
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.

2 participants