diff --git a/docs/source/api/codegen.md b/docs/source/api/codegen.md index 8e374d4e5a52ba3dd44f338f17c10a2d9fb9c784..c52f28f51944d7d1c7eb35dc1108906c90058c68 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 fc1b70ca0a4dafe4ed099024cd2feb05bf61746b..d06c18382bbc21a7bd9a9a41849d586f88215814 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", ]