From e5eafb796f9bd81593bd772dad6b5eb25d492ad2 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Mon, 27 Jan 2025 17:04:49 +0100
Subject: [PATCH] fix __global__ again
---
src/pystencils/backend/emission/base_printer.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/src/pystencils/backend/emission/base_printer.py b/src/pystencils/backend/emission/base_printer.py
index 6808d4731..cc4b50e21 100644
--- a/src/pystencils/backend/emission/base_printer.py
+++ b/src/pystencils/backend/emission/base_printer.py
@@ -57,6 +57,7 @@ from ..extensions.foreign_ast import PsForeignExpression
from ..memory import PsSymbol
from ..constants import PsConstant
from ...types import PsType
+from ...codegen import Target
if TYPE_CHECKING:
from ...codegen import Kernel
@@ -382,7 +383,7 @@ class BasePrinter(ABC):
from ...codegen import GpuKernel
sig_parts = [self._func_prefix] if self._func_prefix is not None else []
- if isinstance(func, GpuKernel):
+ if isinstance(func, GpuKernel) and func.target == Target.CUDA:
sig_parts.append("__global__")
sig_parts += ["void", func.name, f"({params_str})"]
signature = " ".join(sig_parts)
--
GitLab