From 2b7af25e896b0ede046af0a7b720755c148774f4 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Fri, 19 Jul 2024 16:59:16 +0200
Subject: [PATCH] fix equivalence check
---
tests/test_cpu_gpu_equivalence.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/tests/test_cpu_gpu_equivalence.py b/tests/test_cpu_gpu_equivalence.py
index 49f32314..ae71ca17 100644
--- a/tests/test_cpu_gpu_equivalence.py
+++ b/tests/test_cpu_gpu_equivalence.py
@@ -16,6 +16,9 @@ def run_equivalence_test(domain_size, lbm_config, lbm_opt, base_config, time_ste
cpu_scenario = create_channel(domain_size=domain_size, pressure_difference=0.001,
lbm_config=lbm_config, lbm_optimisation=lbm_opt, config=config)
config = replace(base_config, target=Target.GPU)
+ if not IS_PYSTENCILS_2:
+ from pystencils.enums import Backend
+ config = replace(config, backend=Backend.CUDA)
gpu_scenario = create_channel(domain_size=domain_size, pressure_difference=0.001,
lbm_config=lbm_config, lbm_optimisation=lbm_opt, config=config)
--
GitLab