scripts: Clean up IREE_COMPILER_DIR (trial 2)

Move `IREE_COMPILER_DIR` out of the environment variable.

Also fix iree-import-tflite symlink (CI doesn't carry-over the symlink
files across stages).

Change-Id: Ic2b3fac2f9b752a9d5812d0f9b2fb112584e6be3

check-iree-commit.sh

diff --git a/check-iree-commit.sh b/check-iree-commit.sh
index 5119956..f466e11 100755
--- a/check-iree-commit.sh
+++ b/check-iree-commit.sh
@@ -17,13 +17,10 @@
 # Check IREE runtime commit version to be consistent with the compiler.
 
 IREE_SRC="$1"
+IREE_COMPILER_DIR="$2"
 
-if [[ -z "${IREE_COMPILER_DIR}" ]]; then
-  echo "Source build/setup.sh first"
-  exit 1
-fi
-if [[ -z "${IREE_SRC}" ]]; then
-  echo "Usage: check-iree-commit.sh <iree source dir>"
+if [[ $# -ne 2 ]]; then
+  echo "Usage: check-iree-commit.sh <iree source dir> <iree compiler dir>"
   exit 1
 fi
 

download_iree_compiler.py

diff --git a/download_iree_compiler.py b/download_iree_compiler.py
index 3c45055..39e624e 100755
--- a/download_iree_compiler.py
+++ b/download_iree_compiler.py
@@ -4,6 +4,7 @@
 import errno
 import os
 import sys
+import shutil
 import tarfile
 import time
 import argparse
@@ -47,12 +48,6 @@
 
 def main():
     """ Download IREE host compiler from the snapshot release."""
-    iree_compiler_dir = os.getenv("IREE_COMPILER_DIR")
-    if not iree_compiler_dir:
-        print("Please run 'source build/setup.sh' first")
-        sys.exit(1)
-    iree_compiler_dir = Path(iree_compiler_dir)
-
     parser = argparse.ArgumentParser(
         description="Download IREE host compiler from snapshot releases")
     parser.add_argument(
@@ -64,8 +59,15 @@
         help=("URL to check the IREE release."
               "(default: https://api.github.com/repos/google/iree/releases)")
     )
+    parser.add_argument(
+        "--iree_compiler_dir", action="store", required=True,
+        default="",
+        help=("IREE compiler installed directory")
+    )
     args = parser.parse_args()
 
+    iree_compiler_dir = Path(args.iree_compiler_dir)
+
     snapshot = None
     if args.tag_name:
         r = requests.get((f"{args.release_url}/tags/{args.tag_name}"),
@@ -125,13 +127,15 @@
         tar.extractall(path=install_dir)
 
     try:
-        os.symlink(f"{iree_compiler_dir}/bin/iree-import-tflite",
-                f"{install_dir}/bin/iree-import-tflite")
+        shutil.copy2(f"{iree_compiler_dir}/bin/iree-import-tflite",
+                     f"{install_dir}/bin/iree-import-tflite",
+                     follow_symlinks=True)
     except OSError as e:
         if e.errno == errno.EEXIST:
             os.remove(f"{install_dir}/bin/iree-import-tflite")
-            os.symlink(f"{iree_compiler_dir}/bin/iree-import-tflite",
-                f"{install_dir}/bin/iree-import-tflite")
+            shutil.copy2(f"{iree_compiler_dir}/bin/iree-import-tflite",
+                         f"{install_dir}/bin/iree-import-tflite",
+                         follow_symlinks=True)
 
     os.remove(tar_file)
     os.remove(whl_file)