Fix mapreduce launch configuration. (#472)

maleadt · web-flow · commit c46e3c194ab7 · 2024-10-07T15:46:30.000+02:00
diff --git a/src/mapreduce.jl b/src/mapreduce.jl
@@ -146,7 +146,7 @@ function GPUArrays.mapreducedim!(f::F, op::OP, R::oneWrappedArray{T},
     kernel_args = kernel_convert.(args)
     kernel_tt = Tuple{Core.Typeof.(kernel_args)...}
     kernel = zefunction(partial_mapreduce_device, kernel_tt)
-    reduce_items = launch_configuration(kernel)
+    reduce_items = compute_items(launch_configuration(kernel))
 
     # how many groups should we launch?
     #

Original file line number	Diff line number	Diff line change
`@@ -146,7 +146,7 @@ function GPUArrays.mapreducedim!(f::F, op::OP, R::oneWrappedArray{T},`
`146`	`146`	`kernel_args = kernel_convert.(args)`
`147`	`147`	`kernel_tt = Tuple{Core.Typeof.(kernel_args)...}`
`148`	`148`	`kernel = zefunction(partial_mapreduce_device, kernel_tt)`
`149`		`- reduce_items = launch_configuration(kernel)`
	`149`	`+ reduce_items = compute_items(launch_configuration(kernel))`
`150`	`150`
`151`	`151`	`# how many groups should we launch?`
`152`	`152`	`#`