diff --git a/src/pystencils/__init__.py b/src/pystencils/__init__.py index 01c480d7fd1e19a887ac695d4ca2b6c7aaf8c606..3d3b7846a84bb9c477b38e90839d7f67fe12933c 100644 --- a/src/pystencils/__init__.py +++ b/src/pystencils/__init__.py @@ -8,7 +8,12 @@ from .display_utils import get_code_obj, get_code_str, show_code, to_dot from .field import Field, FieldType, fields from .types import create_type from .cache import clear_cache -from .config import CreateKernelConfig, CpuOptimConfig, VectorizationConfig +from .config import ( + CreateKernelConfig, + CpuOptimConfig, + VectorizationConfig, + OpenMpConfig, +) from .kernel_decorator import kernel, kernel_config from .kernelcreation import create_kernel from .backend.kernelfunction import KernelFunction @@ -40,6 +45,7 @@ __all__ = [ "CreateKernelConfig", "CpuOptimConfig", "VectorizationConfig", + "OpenMpConfig", "create_kernel", "KernelFunction", "Target",