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

Package utils.cc properly in jaxlib so that if jaxlib nightly is installed and then used, jaxlib_utils can be accessed.

Browse Source 
PiperOrigin-RevId: 523374835
This commit is contained in:
Yash Katariya 2023-04-11 05:37:57 -07:00 committed by jax authors
parent d4ccab5a78
commit b38e85b3a4
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
build/build_wheel.py
View File

@ -159,6 +159,7 @@ def prepare_wheel(sources_path):
copy_file("__main__/jaxlib/setup.cfg", dst_dir=sources_path)
copy_to_jaxlib("__main__/jaxlib/init.py", dst_filename="__init__.py")
copy_to_jaxlib(f"__main__/jaxlib/cpu_feature_guard.{pyext}")
copy_to_jaxlib(f"__main__/jaxlib/utils.{pyext}")
copy_to_jaxlib("__main__/jaxlib/lapack.py")
copy_to_jaxlib("__main__/jaxlib/hlo_helpers.py")
copy_to_jaxlib("__main__/jaxlib/ducc_fft.py")