allowscalar do block functionality missing?

[leios]

Note on scalar indexing with AMDGPU:

julia> CUDA.allowscalar() do
           a[1]
       end
0.0f0

Works in CUDA

julia> AMDGPU.allowscalar() do
           b[1]
       end
ERROR: MethodError: no method matching allowscalar(::var"#5#6")

Closest candidates are:
  allowscalar(::Bool)
   @ AMDGPU ~/.julia/packages/AMDGPU/Debbg/src/AMDGPU.jl:128

Stacktrace:
 [1] top-level scope
   @ REPL[13]:1

Does not work on AMD.

The @allowscalar macro and allowscalar(true) work as intended

The warnings suggest that AMDGPU.allowscalar() do ... end should work:

┌ Warning: It's not recommended to use allowscalar([true]) to allow scalar indexing.
│ Instead, use `allowscalar() do end` or `@allowscalar` to denote exactly which operations can use scalar operations.
└ @ GPUArraysCore ~/.julia/packages/GPUArraysCore/GMsgk/src/GPUArraysCore.jl:188

[vchuravy]

AMDGPU should just import GPUArraysCore: allowscalar instead of

AMDGPU.jl/src/AMDGPU.jl

Line 119 in 03ca435

allowscalar(x::Bool) = GPUArrays.allowscalar(x)