Fix issue with FASTAPI pip install. (#382)

setup_venv.sh

@@ -102,7 +102,7 @@ if [[ ! -z "${IMPORTER}" ]]; then
   echo "${Yellow}Installing importer tools.."
   if [[ $(uname -s) = 'Linux' ]]; then
     echo "${Yellow}Linux detected.. installing Linux importer tools"
-    $PYTHON -m pip install --upgrade -r "$TD/requirements-importer.txt" -f https://github.com/${RUNTIME}/releases --extra-index-url https://test.pypi.org/simple/ --extra-index-url https://download.pytorch.org/whl/nightly/cu116
+    $PYTHON -m pip install --upgrade -r "$TD/requirements-importer.txt" -f https://github.com/${RUNTIME}/releases --extra-index-url https://download.pytorch.org/whl/nightly/cpu
   elif [[ $(uname -s) = 'Darwin' ]]; then
     echo "${Yellow}macOS detected.. installing macOS importer tools"
     #Conda seems to have some problems installing these packages and hope they get resolved upstream.

tank/test_models.py

@@ -163,7 +163,7 @@ class SharkModuleTester:
             if any([self.ci, self.save_repro, self.save_fails]) == True:
                 self.save_reproducers()
             if self.ci == True:
-                self.upload_repro(self.ci_sha)
+                self.upload_repro()
             raise
 
         result = shark_module.forward(inputs)
@@ -220,7 +220,7 @@ class SharkModuleTester:
     def upload_repro(self):
         import subprocess
 
-        bashCommand = f"gsutil cp -r ./shark_tmp/saved/{self.tmp_prefix}/* gs://shark-public/builder/repro_artifacts/"
+        bashCommand = f"gsutil cp -r ./shark_tmp/saved/{self.tmp_prefix}/* gs://shark-public/builder/repro_artifacts/{self.ci_sha}/{self.tmp_prefix}/"
         process = subprocess.run(bashCommand.split())
 
     def postprocess_outputs(self, golden_out, result):
@@ -335,6 +335,21 @@ class SharkModuleTest(unittest.TestCase):
             pytest.xfail(
                 reason="Numerics issues -- https://github.com/nod-ai/SHARK/issues/344"
             )
+        if (
+            config["model_name"] == "facebook/deit-small-distilled-patch16-224"
+            and device == "cuda"
+        ):
+            pytest.xfail(
+                reason="Fails during iree-compile without reporting diagnostics."
+            )
+        if (
+            config["model_name"]
+            == "microsoft/beit-base-patch16-224-pt22k-ft22k"
+            and device == "cuda"
+        ):
+            pytest.xfail(
+                reason="Fails during iree-compile without reporting diagnostics."
+            )
         if config["framework"] == "tf" and dynamic == True:
             pytest.skip(
                 reason="Dynamic shapes not supported for this framework."