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

feat: add cross_product #1252

Merged
merged 10 commits into from
Sep 17, 2024
Merged

feat: add cross_product #1252

merged 10 commits into from
Sep 17, 2024

Conversation

NripeshN
Copy link
Contributor

@NripeshN NripeshN commented Jul 5, 2024

Closes #1244

mlx/linalg.cpp Outdated Show resolved Hide resolved
Copy link
Member

@angeloskath angeloskath left a comment

Choose a reason for hiding this comment

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

Thanks for the addition!

I would start by making sure that the API follows the corresponding numpy API https://numpy.org/devdocs/reference/generated/numpy.linalg.cross.html including the axis argument.

Regarding the implementation, I would look into split and concatenate.

@awni
Copy link
Member

awni commented Aug 10, 2024

@NripeshN are you planning to come back to this?

@awni
Copy link
Member

awni commented Aug 22, 2024

@NripeshN are you planning to come back to this PR?

@NripeshN
Copy link
Contributor Author

@NripeshN are you planning to come back to this PR?

Hi @awni
Sorry for the delay and late reply, got caught up with few other tasks. I will continue working on this now.

mlx/linalg.cpp Outdated Show resolved Hide resolved
@NripeshN NripeshN requested a review from altaic August 24, 2024 05:34
@awni awni force-pushed the cross_product branch 2 times, most recently from 844413f to 0a28f25 Compare September 16, 2024 17:50
Copy link
Member

@awni awni left a comment

Choose a reason for hiding this comment

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

Thanks for the addition!

@awni awni merged commit 6af5ca3 into ml-explore:main Sep 17, 2024
6 checks passed
@NripeshN NripeshN deleted the cross_product branch September 28, 2024 19:31
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.

[Feature] cross product
4 participants