From 7512a49e637533add13af136aada69e7cd1eab20 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Fri, 20 Dec 2024 13:34:26 +0100
Subject: [PATCH] add config module
---
CMakeLists.txt | 11 +++
cmake/CodegenConfig.template.py | 31 ++++++++
examples/CMakeLists.txt | 4 +-
examples/ForceDrivenChannel/CMakeLists.txt | 8 +-
.../GeneratorScriptBasics/BasicCodegen.py | 5 +-
examples/GeneratorScriptBasics/CMakeLists.txt | 4 +-
src/sfg_walberla/__init__.py | 2 +
src/sfg_walberla/build_config.py | 74 +++++++++++++++++++
8 files changed, 130 insertions(+), 9 deletions(-)
create mode 100644 cmake/CodegenConfig.template.py
create mode 100644 src/sfg_walberla/build_config.py
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 5b1c017..24fafb8 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -2,8 +2,19 @@ cmake_minimum_required( VERSION 3.24 )
project ( sfg-walberla )
list (APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake )
+
find_package( PystencilsSfg REQUIRED )
+set( SfgConfigModule ${CMAKE_BINARY_DIR}/CodegenConfig.py )
+configure_file(
+ ${CMAKE_CURRENT_SOURCE_DIR}/cmake/CodegenConfig.template.py
+ ${SfgConfigModule}
+)
+
+message( STATUS "Wrote project-wide code generator configuration to ${SfgConfigModule}" )
+
+set( PystencilsSfg_CONFIG_MODULE ${SfgConfigModule} PARENT_SCOPE )
+
add_library( sfg_walberla INTERFACE )
target_sources( sfg_walberla
diff --git a/cmake/CodegenConfig.template.py b/cmake/CodegenConfig.template.py
new file mode 100644
index 0000000..07f2999
--- /dev/null
+++ b/cmake/CodegenConfig.template.py
@@ -0,0 +1,31 @@
+from pystencilssfg import SfgConfig
+from sfg_walberla import WalberlaBuildConfig
+
+
+def configure_sfg(cfg: SfgConfig):
+ cfg.extensions.header = "hpp"
+ cfg.extensions.impl = "cpp"
+
+
+def project_info() -> WalberlaBuildConfig:
+ from sfg_walberla.build_config import cmake_parse_bool
+
+ return WalberlaBuildConfig(
+ c_compiler_id="${CMAKE_C_COMPILER_ID}",
+ cxx_compiler_id="${CMAKE_CXX_COMPILER_ID}",
+ use_double_precision=cmake_parse_bool("${WALBERLA_DOUBLE_ACCURACY}"),
+ optimize_for_localhost=cmake_parse_bool("${WALBERLA_OPTIMIZE_FOR_LOCALHOST}"),
+ mpi_enabled=cmake_parse_bool("${WALBERLA_BUILD_WITH_MPI}"),
+ openmp_enabled=cmake_parse_bool("${WALBERLA_BUILD_WITH_OPENMP}"),
+ cuda_enabled=cmake_parse_bool("${WALBERLA_BUILD_WITH_CUDA}"),
+ hip_enabled=cmake_parse_bool("${WALBERLA_BUILD_WITH_HIP}"),
+ likwid_enabled=cmake_parse_bool("${WALBERLA_BUILD_WITH_LIKWID_MARKERS}"),
+ )
+
+
+def validate():
+ _ = project_info()
+
+
+if __name__ == "__main__":
+ validate()
diff --git a/examples/CMakeLists.txt b/examples/CMakeLists.txt
index 56d5261..59c36c2 100644
--- a/examples/CMakeLists.txt
+++ b/examples/CMakeLists.txt
@@ -8,10 +8,10 @@ FetchContent_Declare(
GIT_REPOSITORY https://i10git.cs.fau.de/walberla/walberla.git
)
+message( STATUS "Fetching waLBerla sources (this might take a while)..." )
FetchContent_MakeAvailable(walberla)
add_subdirectory(${CMAKE_SOURCE_DIR}/.. ${CMAKE_BINARY_DIR}/sfg-walberla)
add_subdirectory( GeneratorScriptBasics )
-add_subdirectory( PoiseuilleChannel )
-
+add_subdirectory( ForceDrivenChannel )
diff --git a/examples/ForceDrivenChannel/CMakeLists.txt b/examples/ForceDrivenChannel/CMakeLists.txt
index bebe118..4fbb890 100644
--- a/examples/ForceDrivenChannel/CMakeLists.txt
+++ b/examples/ForceDrivenChannel/CMakeLists.txt
@@ -1,14 +1,14 @@
waLBerla_link_files_to_builddir( Channel.prm )
-add_executable( Ex_PoiseuilleChannel )
-target_sources( Ex_PoiseuilleChannel PRIVATE PoiseuilleChannel.cpp )
+add_executable( Ex_ForceDrivenChannel )
+target_sources( Ex_ForceDrivenChannel PRIVATE ForceDrivenChannel.cpp )
-pystencilssfg_generate_target_sources( Ex_PoiseuilleChannel
+pystencilssfg_generate_target_sources( Ex_ForceDrivenChannel
SCRIPTS LbmAlgorithms.py
)
target_link_libraries(
- Ex_PoiseuilleChannel
+ Ex_ForceDrivenChannel
PRIVATE
core stencil blockforest geometry vtk sfg_walberla
)
diff --git a/examples/GeneratorScriptBasics/BasicCodegen.py b/examples/GeneratorScriptBasics/BasicCodegen.py
index a23e622..61b2e31 100644
--- a/examples/GeneratorScriptBasics/BasicCodegen.py
+++ b/examples/GeneratorScriptBasics/BasicCodegen.py
@@ -1,4 +1,7 @@
from pystencilssfg import SourceFileGenerator
+from sfg_walberla import WalberlaBuildConfig
with SourceFileGenerator() as sfg:
- pass
+ build_config = WalberlaBuildConfig.from_sfg(sfg)
+
+ print(build_config)
diff --git a/examples/GeneratorScriptBasics/CMakeLists.txt b/examples/GeneratorScriptBasics/CMakeLists.txt
index ab855a4..118434c 100644
--- a/examples/GeneratorScriptBasics/CMakeLists.txt
+++ b/examples/GeneratorScriptBasics/CMakeLists.txt
@@ -1,7 +1,7 @@
add_executable( Ex_GeneratorScriptBasics )
-target_sources( Ex_GeneratorScriptBasics PRIVATE Ex_GeneratorScriptBasics.cpp )
+# target_sources( Ex_GeneratorScriptBasics PRIVATE Ex_GeneratorScriptBasics.cpp )
-pystencilssfg_generate_target_sources( Ex_PoiseuilleChannel
+pystencilssfg_generate_target_sources( Ex_GeneratorScriptBasics
SCRIPTS BasicCodegen.py
)
diff --git a/src/sfg_walberla/__init__.py b/src/sfg_walberla/__init__.py
index 063b350..50097ef 100644
--- a/src/sfg_walberla/__init__.py
+++ b/src/sfg_walberla/__init__.py
@@ -1,6 +1,7 @@
from .api import real_t, Vector3, GhostLayerFieldPtr, glfield, IBlockPtr
from .postprocessing import CaptureToClass
from .sweep import Sweep
+from .build_config import WalberlaBuildConfig
__all__ = [
"real_t",
@@ -10,6 +11,7 @@ __all__ = [
"IBlockPtr",
"CaptureToClass",
"Sweep",
+ "WalberlaBuildConfig",
]
from . import _version
diff --git a/src/sfg_walberla/build_config.py b/src/sfg_walberla/build_config.py
new file mode 100644
index 0000000..55fe411
--- /dev/null
+++ b/src/sfg_walberla/build_config.py
@@ -0,0 +1,74 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+from pystencils import CreateKernelConfig
+from pystencils.types.quick import Fp
+from pystencils.backend.jit import no_jit
+
+from pystencilssfg import SfgContext
+from pystencilssfg.composer import SfgIComposer
+
+
+def cmake_parse_bool(var: str):
+ var = var.upper()
+ if var in ("ON", "1", "TRUE"):
+ return True
+ elif var in ("OFF", "0", "FALSE"):
+ return False
+ else:
+ raise ValueError(f"Could not parse cmake value `{var}` as boolean.")
+
+
+@dataclass
+class WalberlaBuildConfig:
+ """Represents a waLBerla build system configuration"""
+
+ c_compiler_id: str
+ """Value of `CMAKE_C_COMPILER_ID`."""
+
+ cxx_compiler_id: str
+ """Value of `CMAKE_CXX_COMPILER_ID`."""
+
+ use_double_precision: bool
+ """Value of `WALBERLA_DOUBLE_ACCURACY`"""
+
+ optimize_for_localhost: bool
+ """Value of `WALBERLA_OPTIMIZE_FOR_LOCALHOST`."""
+
+ mpi_enabled: bool
+ """Value of `WALBERLA_BUILD_WITH_MPI`."""
+
+ openmp_enabled: bool
+ """Value of `WALBERLA_BUILD_WITH_OPENMP`."""
+
+ cuda_enabled: bool
+ """Value of `WALBERLA_BUILD_WITH_CUDA`."""
+
+ hip_enabled: bool
+ """Value of `WALBERLA_BUILD_WITH_HIP`."""
+
+ likwid_enabled: bool
+ """Value of `WALBERLA_BUILD_WITH_LIKWID_MARKERS`"""
+
+ @staticmethod
+ def from_sfg(sfg: SfgContext | SfgIComposer) -> WalberlaBuildConfig:
+ if isinstance(sfg, SfgIComposer):
+ ctx = sfg.context
+ else:
+ ctx = sfg
+
+ if isinstance(ctx.project_info, WalberlaBuildConfig):
+ return ctx.project_info
+ else:
+ raise ValueError(
+ "The given SfgContext does not encapsulate a waLBerla build config object."
+ )
+
+ def get_pystencils_config(self) -> CreateKernelConfig:
+ dtype = Fp(64) if self.use_double_precision else Fp(32)
+
+ return CreateKernelConfig(
+ default_dtype=dtype,
+ jit=no_jit,
+ )
--
GitLab