Skip to content

Commit c0ec6dc

Browse files
Update ext/ReactantCUDAExt.jl
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent d24d09b commit c0ec6dc

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ext/ReactantCUDAExt.jl

+3-1
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,9 @@ function ka_with_reactant(ndrange, workgroupsize, obj, args...)
352352

353353
# figure out the optimal workgroupsize automatically
354354
if KA.workgroupsize(obj) <: KA.DynamicSize && workgroupsize === nothing
355-
if !Reactant.Compiler.PartitionKA[] || raising() || Reactant.Compiler.backend() in ("cpu", "tpu")
355+
if !Reactant.Compiler.PartitionKA[] ||
356+
raising() ||
357+
Reactant.Compiler.backend() in ("cpu", "tpu")
356358
threads = prod(ndrange)
357359
else
358360
config = CUDA.launch_configuration(kernel.fun; max_threads=prod(ndrange))

0 commit comments

Comments
 (0)