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 #10265 from hawkinsp:conda

Browse Source 
PiperOrigin-RevId: 441554415
This commit is contained in:
jax authors 2022-04-13 12:50:21 -07:00
commit 553924f774
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
jax/_src/lib/__init__.py
View File

@ -34,6 +34,20 @@ if platform.system() == "Darwin" and platform.machine() == "arm64":
"Please see https://github.com/google/jax/issues/5501 in the "
"event of problems.")
# This apparently-unused import is to work around
# https://github.com/google/jax/issues/9218#issuecomment-1016949739
# If the user is using Conda, we want to ensure that Conda's libstdc++ is chosen
# by the dynamic linker in preference to the system libstdc++. Before importing
# jaxlib, we first must import a library that:
# a) is likely to be provided by Conda and not hand-installed by the user, and
# b) uses C++ and links to libstdc++.
# Some submodules of scipy (e.g., scipy.signal) satisfy this criterion.
# If the user isn't using Conda, this code merely wastes a bit of time, but
# we're likely to import scipy anyway later.
import scipy.signal as _signal
del _signal
try:
import jaxlib as jaxlib
except ModuleNotFoundError as err: