diff --git a/src/pystencils/backend/platforms/cuda.py b/src/pystencils/backend/platforms/cuda.py index fa246c1286a048fef5386e653aa76556ed74c185..a89225a08cf9cd4ea92dce16523fb19615a49a39 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