Skip to content

Commit 82b3106

Browse files
Fix post_op_callback in reductions (#2236)
1 parent dd0f95e commit 82b3106

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ext/cuda/fields.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ function Base.maximum(field::Field, dev::ClimaComms.CUDADevice)
4242
context = ClimaComms.context(axes(field))
4343
localmax = mapreduce_cuda(identity, max, field)
4444
ClimaComms.allreduce!(context, parent(localmax), max)
45-
call_post_op_callback() && post_op_callback(localmax[], fn, field, dev)
45+
call_post_op_callback() && post_op_callback(localmax[], field, dev)
4646
return localmax[]
4747
end
4848

@@ -58,7 +58,7 @@ function Base.minimum(field::Field, ::ClimaComms.CUDADevice)
5858
context = ClimaComms.context(axes(field))
5959
localmin = mapreduce_cuda(identity, min, field)
6060
ClimaComms.allreduce!(context, parent(localmin), min)
61-
call_post_op_callback() && post_op_callback(localmin[], fn, field, dev)
61+
call_post_op_callback() && post_op_callback(localmin[], field, dev)
6262
return localmin[]
6363
end
6464

0 commit comments

Comments
 (0)