Installing iree-jax on M1 Mac errors

[Yi Wang]

I am reading and following https://github.com/iree-org/iree-jax#development -- not sure if this is the right way to build and install iree-jax -- it failed with the following errors.

16:48 $ python -m pip install -e .[test,xla,cpu] -f https://iree-org.github.io/iree/pip-release-links.html
Looking in indexes: https://pypi.apple.com/simple
Looking in links: https://iree-org.github.io/iree/pip-release-links.html
Obtaining file:///Users/y/w/iree-jax
  Installing build dependencies ... done
  Checking if build backend supports build_editable ... done
  Getting requirements to build editable ... done
  Preparing editable metadata (pyproject.toml) ... done
WARNING: iree-jax 0.1.dev1 does not provide the extra 'cpu'
Requirement already satisfied: numpy in /Users/y/miniforge3/lib/python3.9/site-packages (from iree-jax==0.1.dev1) (1.24.0)
Collecting iree-compiler>=20221121.334
  Downloading https://github.com/iree-org/iree/releases/download/candidate-20230121.404/iree_compiler-20230121.404-cp39-cp39-macosx_11_0_universal2.whl (75.0 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 75.0/75.0 MB 7.6 MB/s eta 0:00:00
Collecting jax>=0.4.1
  Downloading https://pypi.apple.com/packages/packages/e5/aa/3adbd63fe645ee4fbc0daeba9423b21970929145bc63cbf2d813b9d150f5/jax-0.4.1.tar.gz (1.2 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.2/1.2 MB 1.6 MB/s eta 0:00:00
  Preparing metadata (setup.py) ... done
Collecting jaxlib>=0.4.1
  Downloading https://pypi.apple.com/packages/packages/81/e5/e5a912dfd84695bbae9d61dae66339fc9956e4a8926032e5c0b88d2ff7c7/jaxlib-0.4.1-cp39-cp39-macosx_11_0_arm64.whl (49.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 49.7/49.7 MB 3.0 MB/s eta 0:00:00
Collecting iree-runtime>=20221121.334
  Downloading https://github.com/iree-org/iree/releases/download/candidate-20230121.404/iree_runtime-20230121.404-cp39-cp39-macosx_11_0_universal2.whl (3.3 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 3.3/3.3 MB 3.8 MB/s eta 0:00:00
Collecting lit
  Downloading https://pypi.apple.com/packages/packages/2f/5e/8b27ba3715becd8ee26f1f0c2ef9d18fa425a9f6bbf535573b90c08171b4/lit-15.0.7.tar.gz (132 kB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 132.3/132.3 kB 1.7 MB/s eta 0:00:00
  Preparing metadata (setup.py) ... done
ERROR: Could not find a version that satisfies the requirement iree-tools-xla>=20221121.334; extra == "xla" (from iree-jax[cpu,test,xla]) (from versions: none)
ERROR: No matching distribution found for iree-tools-xla>=20221121.334; extra == "xla"

[Yi Wang]

The command worked after I removed xla from pip -e.

$ python -m pip install -e .[cpu] -f https://iree-org.github.io/iree/pip-release-links.html
....
Successfully installed iree-compiler-20230121.404 iree-jax-0.1.dev1 iree-runtime-20230121.404 jax-0.4.1 jaxlib-0.4.1 opt_einsum-3.3.0 scipy-1.10.0

[Stella Laurenzo]

Thank you - the iree-tools-xla package used to be required and must never have been removed from the requirements and old versions of it rolled off of the repository.

[Yi Wang]

Got it. Thank you, @stellaraccident !