We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 8af931c + 1a4130b commit 240ea04Copy full SHA for 240ea04
httomo/runner/task_runner.py
@@ -256,9 +256,9 @@ def _execute_section_block(
256
) -> DataSetBlock:
257
if_previous_block_is_on_gpu = False
258
convert_gpu_block_to_cpu = False
259
+ if_current_block_is_on_gpu = False
260
261
for ind, method in enumerate(section):
- if_current_block_is_on_gpu = False
262
if method.implementation == "gpu_cupy":
263
if_current_block_is_on_gpu = True
264
if method.method_name == "calculate_stats" and if_previous_block_is_on_gpu:
0 commit comments