How to use Apple Metal plugin (without JAX) #25318
Replies: 1 comment
-
|
I think I have solved this by using a non-debug print of the MLIR module. The debug print failed intermittently. Bytecode serialization failed consistently (also on Linux). |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi Jax team. I'm trying to use Apple Metal plugin without JAX, and was informed by the plugin that it only works with StableHLO, so I've started using that. However, it appears the plugin refuses to compile my StableHLO (I'm using
PJRT_Client_Compile), with errorAre you able to give some info about how you made it work? Any particulars you needed to change?
Beta Was this translation helpful? Give feedback.
All reactions