From 01886e87c39febcb9da1f059715bbf9f9fa37f91 Mon Sep 17 00:00:00 2001 From: Frederik Hennig <frederik.hennig@fau.de> Date: Tue, 18 Feb 2025 11:19:52 +0100 Subject: [PATCH] add Lambda to API reference --- docs/source/api/codegen.md | 1 + src/pystencils/codegen/__init__.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/docs/source/api/codegen.md b/docs/source/api/codegen.md index 8e374d4e5..c52f28f51 100644 --- a/docs/source/api/codegen.md +++ b/docs/source/api/codegen.md @@ -177,4 +177,5 @@ The following categories with target-specific options are exposed: Kernel GpuKernel Parameter + Lambda ``` diff --git a/src/pystencils/codegen/__init__.py b/src/pystencils/codegen/__init__.py index fc1b70ca0..d06c18382 100644 --- a/src/pystencils/codegen/__init__.py +++ b/src/pystencils/codegen/__init__.py @@ -6,6 +6,7 @@ from .config import ( from .parameters import Parameter from .kernel import Kernel, GpuKernel from .driver import create_kernel, get_driver +from .functions import Lambda __all__ = [ "Target", @@ -14,6 +15,7 @@ __all__ = [ "Parameter", "Kernel", "GpuKernel", + "Lambda", "create_kernel", "get_driver", ] -- GitLab