From 84416f2b62ffbd2a1a7552a4f315eaf8edb80046 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Mon, 22 Jul 2024 18:22:32 +0200
Subject: [PATCH] disable GPU codegen
---
python/pystencils_walberla/kernel_info.py | 6 +++++-
python/pystencils_walberla/kernel_selection.py | 6 +++++-
2 files changed, 10 insertions(+), 2 deletions(-)
diff --git a/python/pystencils_walberla/kernel_info.py b/python/pystencils_walberla/kernel_info.py
index 67f1a34ad..09a8af5e2 100644
--- a/python/pystencils_walberla/kernel_info.py
+++ b/python/pystencils_walberla/kernel_info.py
@@ -3,7 +3,7 @@ from functools import reduce
from pystencils import Target, TypedSymbol
from pystencils_walberla.utility import merge_sorted_lists
-from pystencils_walberla.compat import backend_printer, SHAPE_DTYPE, KernelFunction
+from pystencils_walberla.compat import backend_printer, SHAPE_DTYPE, KernelFunction, IS_PYSTENCILS_2
# TODO KernelInfo and KernelFamily should have same interface
@@ -15,6 +15,10 @@ class KernelInfo:
self.varying_parameters = tuple(varying_parameters)
self.parameters = ast.get_parameters() # cache parameters here
+ if ast.target == Target.GPU and IS_PYSTENCILS_2:
+ # TODO
+ raise NotImplementedError("Generating GPU kernels is not yet supported with pystencils 2.0")
+
@property
def fields_accessed(self):
return self.ast.fields_accessed
diff --git a/python/pystencils_walberla/kernel_selection.py b/python/pystencils_walberla/kernel_selection.py
index f6d76d07b..6fa95460d 100644
--- a/python/pystencils_walberla/kernel_selection.py
+++ b/python/pystencils_walberla/kernel_selection.py
@@ -6,7 +6,7 @@ from jinja2.filters import do_indent
from pystencils import Target, TypedSymbol
from pystencils_walberla.utility import merge_lists_of_symbols, merge_sorted_lists
-from pystencils_walberla.compat import backend_printer, get_headers, SHAPE_DTYPE
+from pystencils_walberla.compat import backend_printer, get_headers, SHAPE_DTYPE, IS_PYSTENCILS_2
"""
@@ -159,6 +159,10 @@ class KernelCallNode(AbstractKernelSelectionNode):
self.ast = ast
self.parameters = ast.get_parameters() # cache parameters here
+ if ast.target == Target.GPU and IS_PYSTENCILS_2:
+ # TODO
+ raise NotImplementedError("Generating GPU kernels is not yet supported with pystencils 2.0")
+
@property
def selection_parameters(self) -> Set[TypedSymbol]:
return set()
--
GitLab