feat: add progress bar for Autotuner

GPUSimulators/Autotuner.py

@@ -25,6 +25,7 @@ import gc
 import numpy as np
 import logging
 from socket import gethostname
+from tqdm.auto import tqdm
 
 import pycuda.driver as cuda
 
@@ -155,9 +156,9 @@ class Autotuner:
         sim_arguments = arguments.copy()
                     
         with Common.Timer(simulator.__name__) as t:
-            for j, block_height in enumerate(block_heights):
+            for j, block_height in enumerate(tqdm(block_heights, desc='Autotuner Progress')):
                 sim_arguments.update({'block_height': block_height})
-                for i, block_width in enumerate(block_widths):
+                for i, block_width in enumerate(tqdm(block_widths, desc=f'Iteration {j} Progress', leave=False)):
                     sim_arguments.update({'block_width': block_width})
                     megacells[j, i] = Autotuner.run_benchmark(simulator, sim_arguments)
                         

conda_environment.yml

@@ -18,6 +18,7 @@ dependencies:
 - pycuda
 - ipyparallel
 - line_profiler
+- tqdm
 
 # Install conda environment (one-time operation):
 # $ conda env create -f conda_environment.yml

conda_environment_hpc.yml

@@ -12,6 +12,7 @@ dependencies:
 - pytools
 - netcdf4
 - scipy
+- tqdm
 
 # Install conda environment (one-time operation):
 # $ conda env create -f conda_environment_hpc.yml