diff --git a/GPUSimulators/Autotuner.py b/GPUSimulators/Autotuner.py index 97e8959..33ba8aa 100644 --- a/GPUSimulators/Autotuner.py +++ b/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) diff --git a/conda_environment.yml b/conda_environment.yml index 759f011..c92b52b 100644 --- a/conda_environment.yml +++ b/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 diff --git a/conda_environment_hpc.yml b/conda_environment_hpc.yml index b181023..7ca9fd2 100644 --- a/conda_environment_hpc.yml +++ b/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