__init__.py 1.09 KB
Newer Older
1
"""Module to generate stencil kernels in C or CUDA using sympy expressions and call them as Python functions"""
Martin Bauer's avatar
Martin Bauer committed
2
from . import sympy_gmpy_bug_workaround  # NOQA
3
from .field import Field, FieldType, fields
4
5
from .data_types import TypedSymbol
from .slicing import make_slice
Martin Bauer's avatar
Martin Bauer committed
6
from .kernelcreation import create_kernel, create_indexed_kernel, create_staggered_kernel
7
from .display_utils import show_code, to_dot
8
from .simp import AssignmentCollection
9
from .assignment import Assignment, assignment_from_stencil
10
from .sympyextensions import SymbolCreator
11
from .datahandling import create_data_handling
12
from .kernel_decorator import kernel
Martin Bauer's avatar
Martin Bauer committed
13
from . import fd
14
from . import stencil as stencil
Martin Bauer's avatar
Martin Bauer committed
15

16

17
__all__ = ['Field', 'FieldType', 'fields',
Martin Bauer's avatar
Martin Bauer committed
18
           'TypedSymbol',
Martin Bauer's avatar
Martin Bauer committed
19
           'make_slice',
Martin Bauer's avatar
Martin Bauer committed
20
           'create_kernel', 'create_indexed_kernel', 'create_staggered_kernel',
Martin Bauer's avatar
Martin Bauer committed
21
           'show_code', 'to_dot',
22
           'AssignmentCollection',
Martin Bauer's avatar
Martin Bauer committed
23
           'Assignment',
24
           'assignment_from_stencil',
25
           'SymbolCreator',
26
           'create_data_handling',
Martin Bauer's avatar
Martin Bauer committed
27
           'kernel',
28
           'fd',
29
           'stencil']