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

Merge branch 'fix_headerinclude' into 'master'

Headerfile.parse removes strips " from strings

See merge request pycodegen/pystencils-sfg!7
parents 94fbd0a9 8a15ea20
Branches
No related merge requests found
from pystencils import Target, CreateKernelConfig, create_kernel, no_jit
from pystencils import Target, CreateKernelConfig, no_jit
from lbmpy import create_lb_update_rule, LBMOptimisation
from pystencilssfg import SourceFileGenerator, SfgConfiguration
from pystencilssfg.lang.cpp import mdspan_ref
from pystencilssfg import SourceFileGenerator, SfgConfig
sfg_config = SfgConfiguration(
output_directory="out/test_cuda",
outer_namespace="gen_code",
impl_extension="cu"
)
sfg_config = SfgConfig()
sfg_config.extensions.impl = "cu"
sfg_config.output_directory = "out/test_cuda"
sfg_config.outer_namespace = "gen_code"
with SourceFileGenerator(sfg_config) as sfg:
gen_config = CreateKernelConfig(target=Target.CUDA, jit=no_jit)
......@@ -15,6 +13,4 @@ with SourceFileGenerator(sfg_config) as sfg:
update = create_lb_update_rule()
kernel = sfg.kernels.create(update, "lbm_update", gen_config)
sfg.function("lb_update")(
sfg.call(kernel)
)
sfg.function("lb_update")(sfg.call(kernel))
......@@ -24,6 +24,9 @@ class HeaderFile:
return header
system_header = False
if header.startswith('"') and header.endswith('"'):
header = header[1:-1]
if header.startswith("<") and header.endswith(">"):
header = header[1:-1]
system_header = True
......
from pystencilssfg.lang import HeaderFile
import pytest
def test_parse_system():
headerfile = HeaderFile.parse("<test>")
assert str(headerfile) == "<test>" and headerfile.system_header
@pytest.mark.parametrize("header_string", ["test.hpp", '"test.hpp"'])
def test_parse_private(header_string):
headerfile = HeaderFile.parse(header_string)
assert str(headerfile) == "test.hpp" and not headerfile.system_header
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment