mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #21494 from dfm:mac-arm-x86

Browse Source 
PiperOrigin-RevId: 640613602
This commit is contained in:
jax authors 2024-06-05 12:36:00 -07:00
commit 0cbd0a023d
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
jaxlib/cpu_feature_guard.c
View File

@ -77,9 +77,19 @@ static int GetXCR0EAX() {
static void ReportMissingCpuFeature(const char* name) {
PyErr_Format(
PyExc_RuntimeError,
#if defined(__APPLE__)
"This version of jaxlib was built using %s instructions, which your "
"CPU and/or operating system do not support. This error is frequently "
"encountered on macOS when running an x86 Python installation on ARM "
"hardware. In this case, try installing an ARM build of Python. "
"Otherwise, you may be able work around this issue by building jaxlib "
"from source.",
#else
"This version of jaxlib was built using %s instructions, which your "
"CPU and/or operating system do not support. You may be able work around "
"this issue by building jaxlib from source.", name);
"this issue by building jaxlib from source.",
#endif
name);
}
static PyObject *CheckCpuFeatures(PyObject *self, PyObject *args) {