diff options
Diffstat (limited to 'ldc_2d')
-rw-r--r-- | ldc_2d/cpp/CMakeLists.txt | 37 | ||||
-rwxr-xr-x | ldc_2d/cpp/generate.py | 24 | ||||
-rw-r--r-- | ldc_2d/cpp/ldc_2d.cc | 12 | ||||
-rw-r--r-- | ldc_2d/cpp/shell.nix | 42 |
4 files changed, 115 insertions, 0 deletions
diff --git a/ldc_2d/cpp/CMakeLists.txt b/ldc_2d/cpp/CMakeLists.txt new file mode 100644 index 0000000..fe04297 --- /dev/null +++ b/ldc_2d/cpp/CMakeLists.txt @@ -0,0 +1,37 @@ +cmake_minimum_required(VERSION 3.10) +project(ldc_2d LANGUAGES CXX) + +add_custom_target( + boltzgen_kernel + COMMAND + ${CMAKE_CURRENT_SOURCE_DIR}/generate.py +) + +include_directories( + ${CMAKE_BINARY_DIR} +) + +add_executable( + ldc_2d + ldc_2d.cc +) + +add_dependencies( + ldc_2d + boltzgen_kernel +) + +target_compile_features( + ldc_2d + PUBLIC + cxx_std_17 +) + +find_package(OpenMP) +if(OpenMP_CXX_FOUND) + target_link_libraries( + ldc_2d + PUBLIC + OpenMP::OpenMP_CXX + ) +endif() diff --git a/ldc_2d/cpp/generate.py b/ldc_2d/cpp/generate.py new file mode 100755 index 0000000..a1aa561 --- /dev/null +++ b/ldc_2d/cpp/generate.py @@ -0,0 +1,24 @@ +#!/usr/bin/env python + +from boltzgen import LBM, Generator, Geometry +from boltzgen.lbm.model import D2Q9 + +geometry = Geometry(256, 256) + +functions = ['collide_and_stream', 'equilibrilize', 'collect_moments', 'momenta_boundary', 'example'] +extras = ['omp_parallel_for', 'moments_vtk'] + +precision = 'double' + +lbm = LBM(D2Q9) +generator = Generator( + descriptor = D2Q9, + moments = lbm.moments(), + collision = lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.52), + target = 'cpp', + precision = precision, + index = 'XYZ', + layout = 'AOS') + +with open("kernel.h", "w") as kernel: + kernel.write(generator.kernel(geometry, functions, extras)) diff --git a/ldc_2d/cpp/ldc_2d.cc b/ldc_2d/cpp/ldc_2d.cc new file mode 100644 index 0000000..d519490 --- /dev/null +++ b/ldc_2d/cpp/ldc_2d.cc @@ -0,0 +1,12 @@ +#include <cstdint> +#include <memory> +#include <vector> +#include <chrono> +#include <iostream> +#include <fstream> + +#include "kernel.h" + +int main() { + test_ldc(10000); +} diff --git a/ldc_2d/cpp/shell.nix b/ldc_2d/cpp/shell.nix new file mode 100644 index 0000000..0b9db9f --- /dev/null +++ b/ldc_2d/cpp/shell.nix @@ -0,0 +1,42 @@ +{ pkgs ? import <nixpkgs> { }, ... }: + +pkgs.stdenvNoCC.mkDerivation rec { + name = "boltzgen-env"; + env = pkgs.buildEnv { name = name; paths = buildInputs; }; + + buildInputs = let + boltzgen = pkgs.python3.pkgs.buildPythonPackage rec { + pname = "boltzgen"; + version = "0.1"; + + src = pkgs.fetchFromGitHub { + owner = "KnairdA"; + repo = "boltzgen"; + rev = "v0.1.1"; + sha256 = "03fv7krhgc43gfjill8wb4aafr8xi69i2yh9zr68knnmrkrb8vpi"; + }; + + propagatedBuildInputs = with pkgs.python37Packages; [ + sympy + numpy + Mako + ]; + + doCheck = false; + }; + + local-python = pkgs.python3.withPackages (python-packages: with python-packages; [ + boltzgen + ]); + + in [ + local-python + pkgs.gcc9 + pkgs.cmake + ]; + + shellHook = '' + export NIX_SHELL_NAME="${name}" + export PYTHONPATH="$PWD:$PYTHONPATH" + ''; +} |