Skip to content

Commit bbfe28b

Browse files
committed
Lint
1 parent eae1659 commit bbfe28b

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

python/cuda_cccl/benchmarks/compute/bench_reduce.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -217,12 +217,13 @@ def reduce_pointer_lambda(input_array, build_only):
217217
h_init = np.zeros(1, dtype=input_array.dtype)
218218

219219
# Use a lambda function directly as the reducer
220-
op = lambda a, b: a + b
221-
alg = cuda.compute.make_reduce_into(input_array, res, op, h_init)
220+
alg = cuda.compute.make_reduce_into(input_array, res, lambda x, y: x + y, h_init)
222221
if not build_only:
223-
temp_storage_bytes = alg(None, input_array, res, op, size, h_init)
222+
temp_storage_bytes = alg(
223+
None, input_array, res, lambda x, y: x + y, size, h_init
224+
)
224225
temp_storage = cp.empty(temp_storage_bytes, dtype=np.uint8)
225-
alg(temp_storage, input_array, res, op, size, h_init)
226+
alg(temp_storage, input_array, res, lambda x, y: x + y, size, h_init)
226227

227228
cp.cuda.runtime.deviceSynchronize()
228229

0 commit comments

Comments
 (0)