smyalygames/FiniteVolumeGPU
1
0
Fork 0
mirror of https://github.com/smyalygames/FiniteVolumeGPU.git synced 2025-11-29 17:28:03 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add progress bar for Autotuner

Browse Source
This commit is contained in:
Anthony Berg
2025-03-26 00:28:16 +01:00
parent 079bad08b8
commit e7cd6ae34a
3 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
GPUSimulators/Autotuner.py
View File

@@ -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)