Skip to content

Commit b7c832e

Browse files
committed
Chunk size computation bug fix
1 parent d2cc8ec commit b7c832e

1 file changed

Lines changed: 9 additions & 3 deletions

File tree

bin/wfbench

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -396,12 +396,18 @@ def compute_num_chunks(time_limit, cpu_work, gpu_work, num_chunks):
396396
if cpu_work:
397397
num_chunks_cpu = min(num_chunks, cpu_work // min_chunk_size_cpu_work)
398398
else:
399-
num_chunks_cpu = 1
399+
num_chunks_cpu = None
400400
if gpu_work:
401401
num_chunks_gpu = min(num_chunks, gpu_work // min_chunk_size_gpu_work)
402402
else:
403-
num_chunks_gpu = 1
404-
num_chunks = min(num_chunks_cpu, num_chunks_gpu)
403+
num_chunks_gpu = None
404+
405+
if num_chunks_cpu is None:
406+
num_chunks = num_chunks_gpu
407+
elif num_chunks_gpu is None:
408+
num_chunks = num_chunks_cpu
409+
else:
410+
num_chunks = min(num_chunks_cpu, num_chunks_gpu)
405411

406412
num_chunks = max(num_chunks, 1) # The above computations may say "zero"
407413
return num_chunks

0 commit comments

Comments
 (0)