From 8c0dc06d56e12048622ed34b1bd321e2d0f9f074 Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Fri, 11 Apr 2025 22:07:03 +0200
Subject: [PATCH] fix gpu target detection again
---
src/lbmpy/lbstep.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/src/lbmpy/lbstep.py b/src/lbmpy/lbstep.py
index e7bb15b1..0f57dc37 100644
--- a/src/lbmpy/lbstep.py
+++ b/src/lbmpy/lbstep.py
@@ -47,10 +47,8 @@ class LatticeBoltzmannStep:
if config is not None:
if IS_PYSTENCILS_2:
target = config.get_target()
- is_gpu = target.is_gpu()
else:
target = config.target
- is_gpu = target == Target.GPU
else:
target = optimization.get('target', Target.CPU)
@@ -93,7 +91,11 @@ class LatticeBoltzmannStep:
self.density_data_name = name + "_density" if density_data_name is None else density_data_name
self.density_data_index = density_data_index
- self._gpu = is_gpu
+ if IS_PYSTENCILS_2:
+ self._gpu = target.is_gpu()
+ else:
+ self._gpu = target == Target.GPU
+
layout = lbm_optimisation.field_layout
alignment = False
--
GitLab