Added a version 0 for the binaries repo workflow

.gitignore

@@ -18,7 +18,6 @@ with_time_logs/
 *.csv
 *.csv#
 *.ods
-*.json
 *.jpg
 *.jpeg
 *.png

binaries.json

@@ -0,0 +1,31 @@
+{
+    "ivy": {
+        "compiler": [
+            "_compiler.so",
+            {
+                "utils": [
+                    "III.so",
+                    "IIV.so",
+                    "IIX.so",
+                    "IVI.so",
+                    "IVV.so",
+                    "IXI.so",
+                    "IXV.so",
+                    "IXX.so",
+                    "VII.so",
+                    "VIV.so",
+                    "VIX.so",
+                    "VVI.so",
+                    "VVV.so",
+                    "VVX.so",
+                    "VXI.so",
+                    "XII.so",
+                    "XIX.so",
+                    "XVV.so",
+                    "XVX.so",
+                    "XXI.so"
+                ]
+            }
+        ]
+    }
+}

requirements/requirements.txt

@@ -7,6 +7,7 @@ packaging
 nvidia-ml-py # mod_name=pynvml
 diskcache
 google-auth # mod_name=google.auth
+urllib3<2.0
 requests
 pyvis
 dill

setup.py

@@ -18,13 +18,68 @@
 import setuptools
 from setuptools import setup
 from pathlib import Path
+from packaging import tags
+import os
+import json
+import requests
+import itertools
 import re
+from tqdm import tqdm
+
+
+def _get_paths(binaries, root_dir=""):
+    paths = []
+    if isinstance(binaries, str):
+        return [os.path.join(root_dir, binaries)]
+    elif isinstance(binaries, dict):
+        for k, v in binaries.items():
+            paths += _get_paths(v, os.path.join(root_dir, k))
+    else:
+        for i in binaries:
+            paths += _get_paths(i, root_dir)
+    return paths
 
 
 def _strip(line):
     return line.split(" ")[0].split("#")[0].split(",")[0]
 
 
+all_tags = list(tags.sys_tags())
+binaries = json.load(open("binaries.json"))
+paths = _get_paths(binaries)
+end = False
+pbar = None
+spinner = itertools.cycle(["-", "\\", "|", "/"])
+print(f"Locating the binaries {next(spinner)} ", end="")
+
+
+for tag in all_tags:
+    print(f"\rLocating the binaries {next(spinner)} ", end="")
+    if end:
+        pbar.close()
+        break
+    for i, path in enumerate(paths):
+        folders = path.split(os.sep)
+        folder_path, file_path = os.sep.join(folders[:-1]), folders[-1]
+        file_name = f"{file_path[:-3]}_{tag}.so"
+        search_path = f"compiler/{file_name}"
+        r = requests.get(
+            f"https://github.com/vedpatwardhan/ivy/raw/fixes/{search_path}",
+            timeout=20,
+        )
+        if r.status_code == 200:
+            if pbar is None:
+                print()
+                print("Downloading binaries ...")
+                pbar = tqdm(total=len(paths))
+            with open(path, "wb") as f:
+                f.write(r.content)
+            end = path == paths[-1]
+            pbar.update(1)
+        else:
+            break
+
+
 this_directory = Path(__file__).parent
 long_description = (this_directory / "README.md").read_text(encoding="utf-8")