Skip to content
Snippets Groups Projects
  • Martin Bauer's avatar
    6a01f3e2
    Random number generation support for pystencils · 6a01f3e2
    Martin Bauer authored
    - counter-based philox RNG: counter/key is filled with cell coordinate
      and optional external parameters like block position and time step
    - works on CPU and GPU - on CPU only for non-vectorized versions
    
    - introduced more flexible "CustomCodeNode" that can inject
      backend-specific hand-written code
    6a01f3e2
    History
    Random number generation support for pystencils
    Martin Bauer authored
    - counter-based philox RNG: counter/key is filled with cell coordinate
      and optional external parameters like block position and time step
    - works on CPU and GPU - on CPU only for non-vectorized versions
    
    - introduced more flexible "CustomCodeNode" that can inject
      backend-specific hand-written code
__init__.py 1.13 KiB
"""Module to generate stencil kernels in C or CUDA using sympy expressions and call them as Python functions"""
from . import sympy_gmpy_bug_workaround  # NOQA
from .field import Field, FieldType, fields
from .data_types import TypedSymbol
from .slicing import make_slice
from .kernelcreation import create_kernel, create_indexed_kernel, create_staggered_kernel
from .display_utils import show_code, to_dot
from .simp import AssignmentCollection
from .assignment import Assignment, assignment_from_stencil
from .sympyextensions import SymbolCreator
from .datahandling import create_data_handling
from .kernel_decorator import kernel
from .stencils import visualize_stencil_expression
from . import fd


__all__ = ['Field', 'FieldType', 'fields',
           'TypedSymbol',
           'make_slice',
           'create_kernel', 'create_indexed_kernel', 'create_staggered_kernel',
           'show_code', 'to_dot',
           'AssignmentCollection',
           'Assignment',
           'assignment_from_stencil',
           'SymbolCreator',
           'create_data_handling',
           'kernel',
           'fd',
           'visualize_stencil_expression']