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

Update XLA. (#1997)

Browse Source 
Drop six dependency from jaxlib, since xla_client.py no longer uses six.
This commit is contained in:
Peter Hawkins 2020-01-14 11:05:54 -05:00 committed by GitHub
parent 681ba37f7e
commit 64bf55dc6f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
WORKSPACE
View File

@ -23,10 +23,10 @@ http_archive(
# and update the sha256 with the result.
http_archive(
name = "org_tensorflow",
sha256 = "d01c8cafd20c842b7e1af0eb15c45699c12aceb159e002a5cab56af187906d58",
strip_prefix = "tensorflow-84add98ff286a5b805d6b45d016f484b7c4f42ba",
sha256 = "516d4568ee2506f0df2b66d672124232a08fede139643d974d269676c16f5f3d",
strip_prefix = "tensorflow-d2318f541e51ac3afe573c5e7d91b93f8570d63d",
urls = [
"https://github.com/tensorflow/tensorflow/archive/84add98ff286a5b805d6b45d016f484b7c4f42ba.tar.gz",
"https://github.com/tensorflow/tensorflow/archive/d2318f541e51ac3afe573c5e7d91b93f8570d63d.tar.gz",
],
)

2
build/setup.py
View File

@ -32,7 +32,7 @@ setup(
author_email='jax-dev@google.com',
packages=['jaxlib'],
python_requires='>=3.5',
install_requires=['scipy', 'numpy>=1.12', 'six', 'absl-py'],
install_requires=['scipy', 'numpy>=1.12', 'absl-py'],
url='https://github.com/google/jax',
license='Apache-2.0',
package_data={'jaxlib': binary_libs},