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

Add a comment about x86 builds of Python to cpu_feature_guard

Browse Source 
MacOS users frequently encounter a run-time error related to unsupported
AVX instructions (#21491, most recently). This is typically caused by
running an x86 Python build on ARM hardware, and it requires
re-installation of Python. This PR adds this suggestion to the error
message when encountered on macOS.
This commit is contained in:
Dan Foreman-Mackey 2024-05-29 15:25:44 -04:00
parent fafe740400
commit 3cd77ee45e
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) {