Skip to content

Add Support for vmap Function Transformation#245

Merged
davidkoski merged 3 commits intoml-explore:mainfrom
tanmay-bakshi:tanmay/add-vmap
Jun 7, 2025
Merged

Add Support for vmap Function Transformation#245
davidkoski merged 3 commits intoml-explore:mainfrom
tanmay-bakshi:tanmay/add-vmap

Conversation

@tanmay-bakshi
Copy link
Contributor

This PR adds vmap support for automatic vectorization, including documentation and test cases.

#define TARGETCONDITIONALS_H
#define TARGET_OS_IOS 0
#define TARGET_OS_VISION 0
#endif
Copy link
Collaborator

Choose a reason for hiding this comment

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

Do we need this or is it for testing? We should have access to the actual TargetConditionals.h at build time, I think.

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Ah good catch, forgot to remove that file, thank you!

Copy link
Collaborator

@davidkoski davidkoski left a comment

Choose a reason for hiding this comment

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

Looks great, thank you!

@davidkoski davidkoski merged commit 098f3b6 into ml-explore:main Jun 7, 2025
3 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