Skip to content

Commit 240ea04

Browse files
authored
Merge pull request #686 from DiamondLightSource/snapshot_bugfix
fixing a bug around block to cpu logic
2 parents 8af931c + 1a4130b commit 240ea04

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

httomo/runner/task_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -256,9 +256,9 @@ def _execute_section_block(
256256
) -> DataSetBlock:
257257
if_previous_block_is_on_gpu = False
258258
convert_gpu_block_to_cpu = False
259+
if_current_block_is_on_gpu = False
259260

260261
for ind, method in enumerate(section):
261-
if_current_block_is_on_gpu = False
262262
if method.implementation == "gpu_cupy":
263263
if_current_block_is_on_gpu = True
264264
if method.method_name == "calculate_stats" and if_previous_block_is_on_gpu:

0 commit comments

Comments
 (0)