pjrt_c_api_cpu_plugin - MacOS amd and arm64 compilation

[dmarro89]

Hi,
I'm trying to compile on macOS amd and arm64 the plugin pjrt_c_api_cpu_plugin.so.
The build succeeds but when I try to use it in gopjrt I'm getting a segmentation violation error.

What I've done:
On the xla project :

• ./configure.py --backend CPU --os DARWIN
• bazel build //xla/pjrt/c:pjrt_c_api_cpu_plugin.so

Then I've just tried to run the gopjrt tests and they are showing me the error.

If I use the JAX metal plugin -> pjrt_plugin_metal_14.dylib the project is working fine.

Someone has already compiled and used the plugin on macOS?

Thanks

[dmarr0]

Just to add more context.
I have run run :
bazel test //xla/pjrt/c:pjrt_c_api_cpu_test

And I got one error. Here's the log.
exec ${PAGER:-/usr/bin/less} "$0" || exit 1
Executing tests from //xla/pjrt/c:pjrt_c_api_cpu_test
dyld[88604]: symbol not found in flat namespace '___xla_cpu_runtime_EigenSingleThreadedMatMulC128'
external/bazel_tools/tools/test/test-setup.sh: line 353: 88609 Killed: 9 sleep 10