add scripts

Author: hadim

scripts/.ipynb_checkpoints/build_all-checkpoint.py

@@ -0,0 +1,37 @@
+# This is a work-in-progress.
+# The idea is to generate a combination of builds 
+# with different parameters.
+
+from pathlib import Path
+import yaml
+import itertools
+
+base_dir = Path('../')
+config_path = Path('build_config.yml')
+
+config = yaml.load(open(config_path))
+matrix = config['matrix']
+excluded = config['exclude']
+
+# List all possible combinations
+matrix_keys = sorted(matrix)
+temp_builds = list(itertools.product(*(matrix[key] for key in matrix_keys)))
+
+# Exclude unwanted builds.
+builds = []
+for temp_build in temp_builds:
+    build = dict(zip(matrix_keys, temp_build))
+
+    do_exclude = False
+    for exclude_build in excluded:
+        exclude_count = 0
+        for k, v in exclude_build.items():
+            if build[k] == v:
+                exclude_count += 1
+        if exclude_count == len(exclude_build):
+            do_exclude = True
+            
+    if not do_exclude:
+        builds.append(build)
+
+print(builds)

scripts/.ipynb_checkpoints/build_config-checkpoint.yml

@@ -0,0 +1,21 @@
+matrix:
+    python_version:
+        - 3.6
+        - 3.7
+    tf_version:
+        - v1.13.1
+        - v2.0.0-alpha0
+    use_gpu:
+        - True
+        - False
+    cuda_version:
+        - 9.2
+        - 10.0
+    cudnn_version:
+        - 7.1
+        - 7.5
+exclude:
+    - tf_version: v1.13.1
+      cuda_version: 10.0
+    - cudnn_version: 7.1
+      cuda_version: 10.0

scripts/build_all.py

@@ -0,0 +1,37 @@
+# This is a work-in-progress.
+# The idea is to generate a combination of builds
+# with different parameters.
+
+from pathlib import Path
+import yaml
+import itertools
+
+base_dir = Path('../')
+config_path = Path('build_config.yml')
+
+config = yaml.load(open(config_path))
+matrix = config['matrix']
+excluded = config['exclude']
+
+# List all possible combinations
+matrix_keys = sorted(matrix)
+temp_builds = list(itertools.product(*(matrix[key] for key in matrix_keys)))
+
+# Exclude unwanted builds.
+builds = []
+for temp_build in temp_builds:
+    build = dict(zip(matrix_keys, temp_build))
+
+    do_exclude = False
+    for exclude_build in excluded:
+        exclude_count = 0
+        for k, v in exclude_build.items():
+            if build[k] == v:
+                exclude_count += 1
+        if exclude_count == len(exclude_build):
+            do_exclude = True
+
+    if not do_exclude:
+        builds.append(build)
+
+print(builds)

scripts/build_config.yml

@@ -0,0 +1,21 @@
+matrix:
+    python_version:
+        - 3.6
+        - 3.7
+    tf_version:
+        - v1.13.1
+        - v2.0.0-alpha0
+    use_gpu:
+        - True
+        - False
+    cuda_version:
+        - 9.2
+        - 10.0
+    cudnn_version:
+        - 7.1
+        - 7.5
+exclude:
+    - tf_version: v1.13.1
+      cuda_version: 10.0
+    - cudnn_version: 7.1
+      cuda_version: 10.0