Make floating-point atomics work on the LLVM back-end path

Project.toml

@@ -56,3 +56,8 @@ oneAPI_Support_jll = "0.9.2"
 
 [extras]
 libigc_jll = "94295238-5935-5bd7-bb0f-b00942e9bdd5"
+
+[sources]
+# Pin to a commit SHA, not the mutable `atomic-float-ops` branch head, so builds are
+# reproducible and a force-push upstream cannot silently change what users resolve.
+SPIRVIntrinsics = {url = "https://github.com/michel2323/OpenCL.jl", rev = "4257d4075162276d9985168923788490e4071e9e", subdir = "lib/intrinsics"}

src/compiler/compilation.jl

@@ -328,7 +328,14 @@ end
     # advertise the extension). We lower bfloat→i16 in finish_ir! when needed.
     supports_bfloat16 = _device_supports_bfloat16(dev)
 
-    extensions = String[]
+    # SPIR-V extensions the LLVM back-end may emit. Declaring them permits the
+    # corresponding instructions during translation: without
+    # SPV_EXT_shader_atomic_float_add, floating-point atomic operations fail to
+    # translate ("The atomic float instruction requires ... SPV_EXT_shader_atomic_float_add").
+    extensions = String[
+        "SPV_EXT_relaxed_printf_string_address_space",
+        "SPV_EXT_shader_atomic_float_add",
+    ]
     # Only add the SPIR-V extension if the runtime actually supports it
     if _driver_supports_bfloat16_spirv(dev)
         push!(extensions, "SPV_KHR_bfloat16")

test/onemkl.jl

@@ -14,7 +14,7 @@ k = 13
 
 @testset "Version" begin
     version_onemkl = oneMKL.version()
-    @test version_onemkl ≥ v"2026.0.0"
+    @test version_onemkl ≥ v"2025.2.0"
 end
 
 ############################################################################################