From 1989507bc5c2211d7df2e2e2aaf4ed77d8053bad Mon Sep 17 00:00:00 2001
From: Markus Holzer <markus.holzer@fau.de>
Date: Thu, 15 Jun 2023 09:38:18 +0200
Subject: [PATCH] Implement fallback case
---
lbmpy/max_domain_size_info.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/lbmpy/max_domain_size_info.py b/lbmpy/max_domain_size_info.py
index fcd2ed09..3e6b9615 100644
--- a/lbmpy/max_domain_size_info.py
+++ b/lbmpy/max_domain_size_info.py
@@ -41,6 +41,15 @@ except ImportError:
cupy = None
device = None
+if cupy:
+ try:
+ device = cupy.cuda.Device(pystencils.GPU_DEVICE)
+ except AttributeError:
+ warnings.warn("You are using an old pystencils version. Consider updating it")
+ device = cupy.cuda.Device(0)
+else:
+ device = None
+
try:
from psutil import virtual_memory
except ImportError:
--
GitLab