Skip to content

Commit 4a8b417

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

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
@@ -295,7 +295,9 @@ function ka_with_reactant(ndrange, workgroupsize, obj, args...)
295295

296296
# figure out the optimal workgroupsize automatically
297297
if KA.workgroupsize(obj) <: KA.DynamicSize && workgroupsize === nothing
298-
if !Reactant.Compiler.PartitionKA[] || raising() || Reactant.Compiler.backend() in ("cpu", "tpu")
298+
if !Reactant.Compiler.PartitionKA[] ||
299+
raising() ||
300+
Reactant.Compiler.backend() in ("cpu", "tpu")
299301
threads = prod(ndrange)
300302
else
301303
config = CUDA.launch_configuration(kernel.fun; max_threads=prod(ndrange))

0 commit comments

Comments
 (0)