Skip to content
Snippets Groups Projects
Commit e5eafb79 authored by Frederik Hennig's avatar Frederik Hennig
Browse files

fix __global__ again

parent f76aa827
1 merge request!445Object-Oriented CPU JIT API and Prototype Implementation
Pipeline #72868 passed with stages
in 6 minutes and 18 seconds
......@@ -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)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment