__init__.py 760 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
"""Module to generate stencil kernels in C or CUDA using sympy expressions and call them as Python functions"""
from . import sympy_gmpy_bug_workaround
from .field import Field, FieldType
from .data_types import TypedSymbol
from .slicing import make_slice
from .kernelcreation import create_kernel, create_indexed_kernel
from .display_utils import show_code, to_dot
from .assignment_collection import AssignmentCollection
from .assignment import Assignment
from .sympyextensions import SymbolCreator
Martin Bauer's avatar
Martin Bauer committed
11

Martin Bauer's avatar
Martin Bauer committed
12
__all__ = ['Field', 'FieldType',
Martin Bauer's avatar
Martin Bauer committed
13
           'TypedSymbol',
Martin Bauer's avatar
Martin Bauer committed
14
           'make_slice',
Martin Bauer's avatar
Martin Bauer committed
15
           'create_kernel', 'create_indexed_kernel',
Martin Bauer's avatar
Martin Bauer committed
16
           'show_code', 'to_dot',
17
           'AssignmentCollection',
Martin Bauer's avatar
Martin Bauer committed
18
19
           'Assignment',
           'SymbolCreator']