From 5c53ab0f6fead691250337c9abb634369d828dc3 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Tue, 19 Nov 2024 17:08:35 +0100
Subject: [PATCH] Fix parameter names and code style
---
src/pystencils/backend/platforms/cuda.py | 5 +++++
src/pystencils/backend/platforms/generic_gpu.py | 2 +-
src/pystencils/backend/platforms/platform.py | 2 +-
tests/kernelcreation/test_functions.py | 1 -
4 files changed, 7 insertions(+), 3 deletions(-)
diff --git a/src/pystencils/backend/platforms/cuda.py b/src/pystencils/backend/platforms/cuda.py
index dbade47d1..7ebbd4fd4 100644
--- a/src/pystencils/backend/platforms/cuda.py
+++ b/src/pystencils/backend/platforms/cuda.py
@@ -115,6 +115,11 @@ class CudaPlatform(GenericGpu):
case MathFunctions.Abs if dtype.width == 16:
cfunc = CFunction(" __habs", arg_types, dtype)
+ case _:
+ raise MaterializationError(
+ f"Cannot materialize call to function {func}"
+ )
+
call.function = cfunc
return call
diff --git a/src/pystencils/backend/platforms/generic_gpu.py b/src/pystencils/backend/platforms/generic_gpu.py
index f6b888a49..4c990429d 100644
--- a/src/pystencils/backend/platforms/generic_gpu.py
+++ b/src/pystencils/backend/platforms/generic_gpu.py
@@ -77,6 +77,6 @@ class GpuThreadsRange:
class GenericGpu(Platform):
@abstractmethod
def materialize_iteration_space(
- self, block: PsBlock, ispace: IterationSpace
+ self, body: PsBlock, ispace: IterationSpace
) -> tuple[PsBlock, GpuThreadsRange | None]:
pass
diff --git a/src/pystencils/backend/platforms/platform.py b/src/pystencils/backend/platforms/platform.py
index cab4d0a7b..2c7ee1c5f 100644
--- a/src/pystencils/backend/platforms/platform.py
+++ b/src/pystencils/backend/platforms/platform.py
@@ -27,7 +27,7 @@ class Platform(ABC):
@abstractmethod
def materialize_iteration_space(
- self, block: PsBlock, ispace: IterationSpace
+ self, body: PsBlock, ispace: IterationSpace
) -> PsBlock | tuple[PsBlock, Any]:
pass
diff --git a/tests/kernelcreation/test_functions.py b/tests/kernelcreation/test_functions.py
index e16201f81..cab1affcb 100644
--- a/tests/kernelcreation/test_functions.py
+++ b/tests/kernelcreation/test_functions.py
@@ -7,7 +7,6 @@ from pystencils.backend.ast import dfs_preorder
from pystencils.backend.ast.expressions import PsCall
-
def unary_function(name, xp):
return {
"exp": (sp.exp, xp.exp),
--
GitLab