diff --git a/src/pystencils/backend/platforms/cuda.py b/src/pystencils/backend/platforms/cuda.py index 3fe5be229d46a366174ad5a5bac4922131b40ad8..73c4b3b471390e92bdf6e2ea10ecc163b8fea932 100644 --- a/src/pystencils/backend/platforms/cuda.py +++ b/src/pystencils/backend/platforms/cuda.py @@ -160,7 +160,8 @@ class CudaPlatform(GenericGpu): assert isinstance(ptr_expr, PsSymbolExpr) and isinstance(ptr_expr.dtype, PsPointerType) assert isinstance(symbol_expr, PsSymbolExpr) and isinstance(symbol_expr.dtype, PsScalarType) - call.function = CFunction(f"atomic{op.name}", [ptr_expr.dtype, symbol_expr.dtype], PsCustomType("void")) + call.function = CFunction(f"atomic{op.name}", [ptr_expr.dtype, symbol_expr.dtype], + PsCustomType("void")) call.args = (ptr_expr, symbol_expr) if not isinstance(symbol_expr.dtype, PsIeeeFloatType) or symbol_expr.dtype.width not in (32, 64):