From a2a59d40b66390cebe849870d1b9cf058da82850 Mon Sep 17 00:00:00 2001
From: zy69guqi <richard.angersbach@fau.de>
Date: Tue, 4 Feb 2025 18:04:15 +0100
Subject: [PATCH] Wrap statement around generated atomic call [skip ci]
---
src/pystencils/backend/platforms/cuda.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/src/pystencils/backend/platforms/cuda.py b/src/pystencils/backend/platforms/cuda.py
index fa246c128..a89225a08 100644
--- a/src/pystencils/backend/platforms/cuda.py
+++ b/src/pystencils/backend/platforms/cuda.py
@@ -16,7 +16,7 @@ from ..kernelcreation import (
)
from ..kernelcreation.context import KernelCreationContext
-from ..ast.structural import PsBlock, PsConditional, PsDeclaration
+from ..ast.structural import PsBlock, PsConditional, PsDeclaration, PsStatement
from ..ast.expressions import (
PsExpression,
PsLiteralExpr,
@@ -170,7 +170,7 @@ class CudaPlatform(GenericGpu):
if not isinstance(symbol_expr.dtype, PsIeeeFloatType) or symbol_expr.dtype.width not in (32, 64):
NotImplementedError("atomicMul is only available for float32/64 datatypes")
- return call
+ return PsStatement(call)
# Internals
--
GitLab