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

cleanup in cmake integration test

parent 61cdb0ff
No related branches found
No related tags found
No related merge requests found
Pipeline #57207 passed
# type: ignore
import sympy as sp import sympy as sp
import numpy as np
from pystencils.session import * from pystencils import fields, kernel
from pystencilssfg import SourceFileGenerator from pystencilssfg import SourceFileGenerator
from pystencilssfg.source_concepts.cpp import std_mdspan
with SourceFileGenerator() as sfg: 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') h = sp.Symbol('h')
@ps.kernel @kernel
def poisson_jacobi(): 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) poisson_kernel = sfg.kernels.create(poisson_jacobi)
......
# type: ignore
import sympy as sp import sympy as sp
import numpy as np
from pystencils.session import * from pystencils import fields, kernel, Field
from pystencilssfg import SourceFileGenerator from pystencilssfg import SourceFileGenerator
from pystencilssfg.source_concepts.cpp import std_mdspan
with SourceFileGenerator() as sfg: with SourceFileGenerator() as sfg:
src = ps.fields("src: double[2D]") src: Field = fields("src: double[2D]")
h = sp.Symbol('h') h = sp.Symbol('h')
@ps.kernel @kernel
def poisson_gs(): 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) 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