From fa63ecb9978ac872a02a6773629ab746e523d5bb Mon Sep 17 00:00:00 2001 From: Michel Schanen Date: Fri, 31 Oct 2025 12:02:47 -0500 Subject: [PATCH] Add atomic float support --- src/compiler/compilation.jl | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/compiler/compilation.jl b/src/compiler/compilation.jl index aeb83f40..33d7764f 100644 --- a/src/compiler/compilation.jl +++ b/src/compiler/compilation.jl @@ -68,7 +68,10 @@ end supports_fp64 = oneL0.module_properties(device()).fp64flags & oneL0.ZE_DEVICE_MODULE_FLAG_FP64 == oneL0.ZE_DEVICE_MODULE_FLAG_FP64 # TODO: emit printf format strings in constant memory - extensions = String["SPV_EXT_relaxed_printf_string_address_space"] + extensions = String[ + "SPV_EXT_relaxed_printf_string_address_space", + "SPV_EXT_shader_atomic_float_add" + ] # create GPUCompiler objects target = SPIRVCompilerTarget(; extensions, supports_fp16, supports_fp64, kwargs...)