Specialize GPUArrays' global_size query. (#139)

JuliaGPU · Mar 17, 2023 · 8f51c87 · 8f51c87
1 parent 77b8551
commit 8f51c87
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/gpuarrays.jl b/src/gpuarrays.jl
@@ -34,6 +34,7 @@ GPUArrays.blockdim(ctx::mtlKernelContext)     = Metal.threads_per_threadgroup_1d
 GPUArrays.threadidx(ctx::mtlKernelContext)    = Metal.thread_position_in_threadgroup_1d()
 GPUArrays.griddim(ctx::mtlKernelContext)      = Metal.threadgroups_per_grid_1d()
 GPUArrays.global_index(ctx::mtlKernelContext) = Metal.thread_position_in_grid_1d()
+GPUArrays.global_size(ctx::mtlKernelContext)  = Metal.threads_per_grid_1d()
 
 # math