Automatically fetch and register PjRT Metal plugin

[mofeing]

• Move artifact to a new PjRT_Metal_jll package
•  Try local build

@wsmoses is there a Metal infra available for testing?

[github-actions]

Benchmark Results

	main	0980193...	main/098019337b01a7...
comptime/basics/2D sum	27.8 ± 0.78 ms	27.3 ± 0.91 ms	1.02
comptime/basics/Basic cos	26 ± 0.83 ms	26 ± 0.59 ms	0.999
comptime/basics/Basic grad cos	0.045 ± 0.0012 s	0.045 ± 0.00082 s	1
time_to_load	1.29 ± 0.039 s	1.3 ± 0.0042 s	0.987

Benchmark Plots

A plot of the benchmark results have been uploaded as an artifact to the workflow run for this PR.
Go to "Actions"->"Benchmark a pull request"->[the most recent run]->"Artifacts" (at the bottom).

[wsmoses]

I think just macos CI should do it, no?

[wsmoses]

Can we avoid making a separate api change atm since the oher loadpjrt plugins have an api presently?

Just for the sake of setting it up without a jll bump

[avik-pal]

I think just macos CI should do it, no?

We might want to use the buildkite ones here. The macos-large GH runners have GPUs but those are paid. The free ones don't have GPU.

Buildkite script for that https://github.com/LuxDL/MLDataDevices.jl/blob/b7e05516c0c0ec554259d90b5d620b6d0bf12154/.buildkite/testing.yml#L110-L134.

[wsmoses]

Fair nuff/sgtm

…

On Wed, Sep 4, 2024 at 7:41 PM Avik Pal ***@***.***> wrote: I think just macos CI should do it, no? We might want to use the buildkite ones here. The macos-large GH runners have GPUs but those are paid. The free ones don't have GPU. Buildkite script for that https://github.com/LuxDL/MLDataDevices.jl/blob/b7e05516c0c0ec554259d90b5d620b6d0bf12154/.buildkite/testing.yml#L110-L134 . — Reply to this email directly, view it on GitHub <#99 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AAJTUXFRVUORGMVZHQ2KIQTZU6SEJAVCNFSM6AAAAABNVBCWW2VHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMZDGMZQGM4DGMBXGY> . You are receiving this because you were mentioned.Message ID: ***@***.***>

[wsmoses]

We might as well just label this as gpu