From 04353fb7602b7eb6a8b86193510ebe0c670b7724 Mon Sep 17 00:00:00 2001 From: Anders Leino Date: Wed, 22 Jan 2025 19:10:35 +0200 Subject: Add validation for destination of atomic operations (#6093) * Reimplement the GLSL atomic* functions in terms of __intrinsic_op Many of these functions map directly to atomic IR instructions. The functions taking atomic_uint are left as they are. This helps to address #5989, since the destination pointer type validation can then be written only for the atomic IR instructions. * Add validation for atomic operations Diagnose an error if the destination of the atomic operation is not appropriate, where appropriate means it's either: - 'groupshared' - from a device buffer This closes #5989. * Add tests for GLSL atomics destination validation Attempting to use the GLSL atomic functions on destinations that are neither groupshared nor from a device buffer should fail with the following error: error 41403: cannot perform atomic operation because destination is neither groupshared nor from a device buffer. * Validate atomic operations after address space specialization Address space specialization for SPIR-V is not done as part of `linkAndOptimizeIR`, as it is for e.g. Metal, so opt out and add a separate call for SPIR-V. * Allow unchecked in/inout parameters for non-SPIRV targets * Clean up callees left without uses during address space specialziation * format code --------- Co-authored-by: slangbot <186143334+slangbot@users.noreply.github.com> Co-authored-by: Yong He --- source/slang/slang-emit.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'source/slang/slang-emit.cpp') diff --git a/source/slang/slang-emit.cpp b/source/slang/slang-emit.cpp index ea209b598..cf4070df4 100644 --- a/source/slang/slang-emit.cpp +++ b/source/slang/slang-emit.cpp @@ -1320,6 +1320,14 @@ Result linkAndOptimizeIR( byteAddressBufferOptions); } + // For SPIR-V, this function is called elsewhere, so that it can happen after address space + // specialization + if (target != CodeGenTarget::SPIRV && target != CodeGenTarget::SPIRVAssembly) + { + bool skipFuncParamValidation = true; + validateAtomicOperations(skipFuncParamValidation, sink, irModule->getModuleInst()); + } + // For CUDA targets only, we will need to turn operations // the implicitly reference the "active mask" into ones // that use (and pass around) an explicit mask instead. -- cgit v1.2.3