Add importer utils to linux build

requirements-importer.txt

@@ -0,0 +1,22 @@
+-f https://github.com/nod-ai/SHARK-Runtime/releases
+#iree-compiler should already be installed
+#iree-runtime
+iree-tools-tflite
+iree-tools-xla
+iree-tools-tf
+
+# TensorFlow and JAX.
+gin-config
+tensorflow
+tf-models-nightly
+tensorflow-text-nightly
+transformers
+jax[cpu]
+
+
+# tflitehub dependencies.
+Pillow
+
+# Testing and support.
+lit
+pyyaml

requirements.txt

@@ -1,4 +1,5 @@
 -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
+-f https://github.com/nod-ai/SHARK-Runtime/releases
 --pre
 
 numpy
@@ -12,3 +13,23 @@ ninja
 pytest
 pytest-parallel
 
+iree-compiler
+iree-runtime
+iree-tools-tflite
+iree-tools-xla
+iree-tools-tf
+
+# TensorFlow and JAX.
+gin-config
+tensorflow
+tf-models-nightly
+tensorflow-text-nightly
+transformers
+jax[cpu]
+
+# tflitehub dependencies.
+Pillow
+
+# Testing and support.
+lit
+pyyaml

setup_venv.sh

@@ -105,6 +105,11 @@ else
   fi
 fi
 
+if [[ $(uname -s) = 'Linux' ]]; then
+  echo "${Yellow}Linux detected.. installing importer tools"
+  $PYTHON -m pip install --upgrade -r "$TD/requirements-importer.txt" -f https://github.com/nod-ai/SHARK-Runtime/releases
+fi
+
 $PYTHON -m pip install transformers
 $PYTHON -m pip wheel -v -w $TD/wheelhouse $TD -f https://github.com/nod-ai/SHARK-Runtime/releases -f https://github.com/llvm/torch-mlir/releases --extra-index-url https://download.pytorch.org/whl/nightly/cpu
 $PYTHON -m pip install . --extra-index-url https://download.pytorch.org/whl/nightly/cpu -f https://github.com/llvm/torch-mlir/releases -f https://github.com/nod-ai/SHARK-Runtime/releases