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

Added mlx_device_get_index, and mlx_stream_get_index #49

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

stemann
Copy link
Contributor

@stemann stemann commented Dec 15, 2024

Added mlx_device_get_index, and mlx_stream_get_index

Also exposed mlx_device_equal.

Also exposed mlx_device_equal.
@stemann
Copy link
Contributor Author

stemann commented Dec 15, 2024

This will aid custom printing, e.g. for https://github.com/stemann/MLX.jl/blob/master/src/device.jl#L39

(and make it possible to compare devices)

@stemann stemann changed the title Added mlx_device_get_index Added mlx_device_get_index, and mlx_stream_get_index Dec 16, 2024
@stemann
Copy link
Contributor Author

stemann commented Jan 7, 2025

@awni Comments?

@awni
Copy link
Member

awni commented Jan 7, 2025

Looks good to me. I don't see a reason not to expose these.

@awni awni requested a review from andresy January 7, 2025 14:15
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