localmemory

Author: christiangnrd

src/KernelAbstractions.jl

@@ -814,7 +814,9 @@ include("macros.jl")
 ###
 
 function Scratchpad end
-function SharedMemory end
+function SharedMemory(t::Type{T}, dims::Val{Dims}, id::Val{Id}) where {T, Dims, Id}
+    KernelIntrinsics.localmemory(t, dims, id)
+end
 
 function __synchronize()
     error("@synchronize used outside kernel or not captured")

src/pocl/backend.jl

@@ -166,7 +166,7 @@ end
 
 ## Shared and Scratch Memory
 
-@device_override @inline function KA.SharedMemory(::Type{T}, ::Val{Dims}, ::Val{Id}) where {T, Dims, Id}
+@device_override @inline function KI.localmemory(::Type{T}, ::Val{Dims}, ::Val{Id}) where {T, Dims, Id}
     ptr = SPIRVIntrinsics.emit_localmemory(T, Val(prod(Dims)))
     CLDeviceArray(Dims, ptr)
 end