Pass extensions argument through @opencl (#355)

Author: simeonschaub

src/compiler/execution.jl

@@ -4,7 +4,7 @@ export @opencl, clfunction
 ## high-level @opencl interface
 
 const MACRO_KWARGS = [:launch]
-const COMPILER_KWARGS = [:kernel, :name, :always_inline]
+const COMPILER_KWARGS = [:kernel, :name, :always_inline, :extensions]
 const LAUNCH_KWARGS = [:global_size, :local_size, :queue]
 
 macro opencl(ex...)

test/Project.toml

@@ -12,6 +12,7 @@ Preferences = "21216c6a-2e73-6563-6e65-726566657250"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 REPL = "3fa0cd96-eef1-5676-8a61-b3b8758bbffb"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+SPIRVIntrinsics = "71d1d633-e7e8-4a92-83a1-de8814b09ba8"
 SPIRV_LLVM_Backend_jll = "4376b9bf-cff8-51b6-bb48-39421dff0d0c"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"

test/atomics.jl

@@ -1,3 +1,5 @@
+using SPIRVIntrinsics: @builtin_ccall, @typed_ccall, LLVMPtr
+
 @testset "atomics" begin
 
 function atomic_count(counter)
@@ -13,4 +15,27 @@ end
     end
 end
 
+    if "cl_ext_float_atomics" in cl.device().extensions
+        function atomic_float_add(counter, val)
+            @builtin_ccall(
+                "atomic_add", Float32,
+                (LLVMPtr{Float32, AS.CrossWorkgroup}, Float32),
+                pointer(counter), val,
+            )
+            return
+        end
+
+        @testset "SPV_EXT_shader_atomic_float_add extension" begin
+            a = OpenCL.zeros(Float32)
+            @opencl global_size = 1000 extensions = ["SPV_EXT_shader_atomic_float_add"] atomic_float_add(a, 1.0f0)
+            @test OpenCL.@allowscalar a[] == 1000.0f0
+
+            spv = sprint() do io
+                OpenCL.code_native(io, atomic_float_add, Tuple{CLDeviceArray{Float32, 0, 1}, Float32}; extensions = ["SPV_EXT_shader_atomic_float_add"])
+            end
+            @test occursin("OpExtension \"SPV_EXT_shader_atomic_float_add\"", spv)
+            @test occursin("OpAtomicFAddEXT", spv)
+        end
+    end
+
 end