-
Notifications
You must be signed in to change notification settings - Fork 1.5k
[WIP] Add mlx.core.searchsorted #2817
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
Draft
ppaleja
wants to merge
25
commits into
ml-explore:main
Choose a base branch
from
ppaleja:feature/searchsorted
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
Draft
Changes from 1 commit
Commits
Show all changes
25 commits
Select commit
Hold shift + click to select a range
abf4f49
Adds MLX searchsorted PRD documentation
ppaleja 2738378
correct prd to reflect python API layer
ppaleja 0e75198
Adds extensive test coverage for searchsorted
ppaleja d432e0c
Adds clangd configuration for improved static analysis
ppaleja f8161f3
Removes unused headers to satisy clangd
ppaleja 5ccf038
update project files
ppaleja f8c4298
Adds axis-aware numpy reference for searchsorted tests
ppaleja 948c932
Build: Add virtualenv Python detection to CMake
ppaleja 02c5f4b
Refactor: Improve comment formatting and header includes
ppaleja d01cb10
Feat: Implement SearchSorted CPU backend
ppaleja 832dca4
Feat: Define SearchSorted primitive and generic logic
ppaleja cd287eb
Feat: Expose searchsorted in C++ API
ppaleja b4d7437
Feat: Expose searchsorted in Python API
ppaleja 283f211
Applies minor formatting improvements
ppaleja 8008837
Merge branch 'ml-explore:main' into feature/searchsorted
ppaleja 0f96bf1
feat: Refine CPU search_sorted implementation
ppaleja abd7445
fix: Resolve race condition in CPU SearchSorted primitive
ppaleja 40fe750
test: Add comprehensive searchsorted tests
ppaleja a6a6173
feat: Implement CPU fallback for SearchSorted GPU primitive
ppaleja ff87c91
chore: Minor cleanups in mlx/ops.cpp
ppaleja 8bfdecf
pre-commit
ppaleja cf667b2
.gitignore
ppaleja d42f181
Revert "Adds MLX searchsorted PRD documentation"
ppaleja 94c7e08
Adds `searchsorted` to Python ops documentation
ppaleja d2ed63f
Optimizes search_sorted with fast-path and adaptive search
ppaleja File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Removes unused headers to satisy clangd
Drops unused header dependencies: - removes transforms.h from the operations module - removes backend/utils.h from primitives
- Loading branch information
commit f8161f38046ae249568ed75646d55d45201e86e3
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
|
|
@@ -10,7 +10,6 @@ | |
| #include <sstream> | ||
| #include <stdexcept> | ||
|
|
||
| #include "mlx/backend/common/utils.h" | ||
| #include "mlx/fft.h" | ||
|
||
| #include "mlx/linalg.h" | ||
| #include "mlx/ops.h" | ||
|
|
||
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The removed include for "mlx/transforms.h" may be needed elsewhere in the file. Only remove this include if you've verified that no part of ops.cpp requires declarations from transforms.h. The fact that transforms_impl.h is still included suggests transforms.h might have been providing necessary declarations.