Skip to content
Snippets Groups Projects
Commit a0b5f900 authored by Frederik Hennig's avatar Frederik Hennig
Browse files

Merge branch 'master' into build_tools

parents 4f032da9 a9598b19
No related branches found
No related tags found
No related merge requests found
# type: ignore
import sympy as sp
import numpy as np
from pystencils.session import *
from pystencils import fields, kernel
from pystencilssfg import SourceFileGenerator
from pystencilssfg.source_concepts.cpp import std_mdspan
with SourceFileGenerator() as sfg:
src, dst = ps.fields("src, dst(1) : double[2D]")
src, dst = fields("src, dst(1) : double[2D]")
h = sp.Symbol('h')
@ps.kernel
@kernel
def poisson_jacobi():
dst[0,0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
dst[0, 0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
poisson_kernel = sfg.kernels.create(poisson_jacobi)
......
# type: ignore
import sympy as sp
import numpy as np
from pystencils.session import *
from pystencils import fields, kernel, Field
from pystencilssfg import SourceFileGenerator
from pystencilssfg.source_concepts.cpp import std_mdspan
with SourceFileGenerator() as sfg:
src = ps.fields("src: double[2D]")
src: Field = fields("src: double[2D]")
h = sp.Symbol('h')
@ps.kernel
@kernel
def poisson_gs():
src[0,0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
src[0, 0] @= (src[1, 0] + src[-1, 0] + src[0, 1] + src[0, -1]) / 4
poisson_kernel = sfg.kernels.create(poisson_gs)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment