diff --git a/src/nodes/gpgpu/AtomicFunctionNode.js b/src/nodes/gpgpu/AtomicFunctionNode.js
index 15e4a329145abf..d039abb02ca03b 100644
--- a/src/nodes/gpgpu/AtomicFunctionNode.js
+++ b/src/nodes/gpgpu/AtomicFunctionNode.js
@@ -1,4 +1,5 @@
 import Node from '../core/Node.js';
+import { expression } from '../code/ExpressionNode.js';
 import { nodeProxy } from '../tsl/TSLCore.js';
 
 /**
@@ -89,7 +90,8 @@ class AtomicFunctionNode extends Node {
 
 	generate( builder ) {
 
-		const parents = builder.getNodeProperties( this ).parents;
+		const properties = builder.getNodeProperties( this );
+		const parents = properties.parents;
 
 		const method = this.method;
 
@@ -117,9 +119,17 @@ class AtomicFunctionNode extends Node {
 
 			builder.addLineFlowCode( methodSnippet, this );
 
-		}
+		} else {
+
+			if ( properties.constNode === undefined ) {
+
+				properties.constNode = expression( methodSnippet, type ).toConst();
 
-		return methodSnippet;
+			}
+
+			return properties.constNode.build( builder );
+
+		}
 
 	}