From d35acf15374b7424978bc63a5d3f1bead1dcae2a Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Wed, 12 Mar 2025 10:26:48 +0100
Subject: [PATCH] use Target.HIP

---
 src/pystencilssfg/ir/call_tree.py            | 5 ++---
 tests/generator_scripts/source/HipKernels.py | 2 +-
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/src/pystencilssfg/ir/call_tree.py b/src/pystencilssfg/ir/call_tree.py
index ab84db4..61c5bbb 100644
--- a/src/pystencilssfg/ir/call_tree.py
+++ b/src/pystencilssfg/ir/call_tree.py
@@ -226,13 +226,12 @@ class SfgGpuKernelInvocation(SfgCallTreeNode):
         shared_memory_bytes: SfgStatements | None,
         stream: SfgStatements | None,
     ):
-        from pystencils import Target
         from pystencils.codegen import GpuKernel
 
         kernel = kernel_handle.kernel
-        if not (isinstance(kernel, GpuKernel) and kernel.target == Target.CUDA):
+        if not isinstance(kernel, GpuKernel):
             raise ValueError(
-                "An `SfgCudaKernelInvocation` node can only call a CUDA kernel."
+                "An `SfgGpuKernelInvocation` node can only call GPU kernels."
             )
 
         super().__init__()
diff --git a/tests/generator_scripts/source/HipKernels.py b/tests/generator_scripts/source/HipKernels.py
index 78464b5..32d9b1d 100644
--- a/tests/generator_scripts/source/HipKernels.py
+++ b/tests/generator_scripts/source/HipKernels.py
@@ -14,7 +14,7 @@ with SourceFileGenerator() as sfg:
     sfg.use_hip()
     sfg.namespace("gen")
 
-    base_config = ps.CreateKernelConfig(target=ps.Target.CUDA)
+    base_config = ps.CreateKernelConfig(target=ps.Target.HIP)
 
     block_size = sfg.gpu_api.dim3().var("blockSize")
     grid_size = sfg.gpu_api.dim3().var("gridSize")
-- 
GitLab