search

microsoft / knossos-ksc

Compiler with automatic differentiation
Other
45 stars 10 forks source link

Bump jaxlib #1055

Closed toelli-msft closed 2 years ago

toelli-msft commented 2 years ago

because 0.1.60 is now the oldest version that is available on PyPI

(same as in https://github.com/microsoft/knossos-ksc/pull/1008)

awf commented 2 years ago

Perhaps need to jump further or it'll just be an issue again in another month?

See #1056 for a more radical approach, that seems to work on my machine (but of course CI machines are not the same)

toelli-msft commented 2 years ago

Hmm, seems like this version of jaxlib might not have things we use

E   ImportError: cannot import name 'pytree' from 'jaxlib' (/opt/hostedtoolcache/Python/3.8.11/x64/lib/python3.8/site-packages/jaxlib/__init__.py)
=========================== short test summary info ============================
ERROR test/python/test_translate.py - ImportError: cannot import name 'pytree...
ERROR test/python/test_translate.py
toelli-msft commented 2 years ago

Although I have no idea why it thinks test_translate.py is trying to import pytree.

awf commented 2 years ago

Closing in favour of #1056