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

Fix parameter names and code style

parent df59d8aa
No related branches found
No related tags found
1 merge request!429Iteration Slices: Extended GPU support + bugfixes
Pipeline #70421 passed
......@@ -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
......
......@@ -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
......@@ -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
......
......@@ -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),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment