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

make setup.py copy over .so files

Browse Source
This commit is contained in:
Matthew Johnson 2018-11-18 15:28:22 -08:00
parent 1b836515c2
commit 0dfa736ece
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@ -1,4 +1,5 @@
from setuptools import setup
from glob import glob
setup(
name='jax',
@ -11,4 +12,5 @@ setup(
install_requires=['numpy>=1.12', 'six', 'protobuf'],
url='https://github.com/google/jax',
license='Apache-2.0',
package_data={'jax.lib': glob('jax/lib/*.so')},
)