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