Add Apple silicon GPU(mps) support to ray

[NripeshN]

Why are these changes needed?

Related issue number

Closes #39136

Checks

• I've signed off every commit(by using the -s flag, i.e., git commit -s) in this PR.
• I've run scripts/format.sh to lint the changes in this PR.
• I've included any doc changes needed for https://docs.ray.io/en/master/.
  • I've added any new APIs to the API Reference. For example, if I added a
    method in Tune, I've added it in doc/source/tune/api/ under the
    corresponding .rst file.
• I've made sure the tests are passing. Note that there might be a few flaky tests, see the recent failures at https://flakey-tests.ray.io/
• Testing Strategy
  • Unit tests
  • Release tests
  • This PR is not tested :(

[NripeshN]

hi @krfricke, created a new PR because I messed up my old one.

[stale]

This pull request has been automatically marked as stale because it has not had recent activity. It will be closed in 14 days if no further activity occurs. Thank you for your contributions.

• If you'd like to keep this open, just leave any comment, and the stale label will be removed.

[NripeshN]

hi @krfricke, please review this PR

[jjyao]

Hi @NripeshN,

Thanks for the contribution. As the first step, we need to support Apple silicon GPU in Ray core. Could you follow #38553 as an example to update the PR? Thanks!

[jjyao]

Can you rebase with master

[anyscalesam]

@jjyao this feels like a good thing to push through to enable better support for local dev/test on apple silicon. reminder to take a look. im adding this to core-sprint so we don't forget.

[NripeshN]

Can you run

hey @NripeshN are you still interested in updating this and putting it through?

Hiya @dayshah Just attempted the PR, feel free to review it😁

can you fix the pre-commit, can follow the directions here https://docs.ray.io/en/latest/ray-contribute/getting-involved.html#lint-and-formatting

Sorry no idea how I missed this comment, just fixed the lint errors.

[jjyao]

I'll try to see if we can find someone from Apple to review this.

[github-actions]

This pull request has been automatically marked as stale because it has not had
any activity for 14 days. It will be closed in another 14 days if no further activity occurs.
Thank you for your contributions.

You can always ask for help on our discussion forum or Ray's public slack channel.

If you'd like to keep this open, just leave any comment, and the stale label will be removed.

[jjyao]

Quick update: we currently expect that an Apple employee will review this in the coming weeks.