Disallow scalar iteration during JLArray execution.

maleadt · maleadt · commit b16a84b1270e · 2020-01-31T14:39:53.000+01:00
diff --git a/src/host/indexing.jl b/src/host/indexing.jl
@@ -1,6 +1,6 @@
 # host-level indexing
 
-export allowscalar, @allowscalar, assertscalar
+export allowscalar, @allowscalar, @disallowscalar, assertscalar
 
 
 # mechanism to disallow scalar operations
diff --git a/src/reference.jl b/src/reference.jl
@@ -68,7 +68,7 @@ function GPUArrays.gpu_call(::JLBackend, f, args...; blocks::Int, threads::Int)
     ctx = JLKernelContext(threads, blocks)
     device_args = jlconvert.(args)
     tasks = Array{Task}(undef, threads)
-    @allowscalar for blockidx in 1:blocks
+    @disallowscalar for blockidx in 1:blocks
         ctx.blockidx = blockidx
         for threadidx in 1:threads
             thread_ctx = JLKernelContext(ctx, threadidx)
