From f4abda02df0bb9eb1bd20dca61969e5eff681892 Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Thu, 16 Jan 2020 13:23:17 +0100
Subject: [PATCH] Fix in boundary handling - check for GPU_LIKE_TARGETS
---
pystencils/boundaries/boundaryhandling.py | 15 +++++++++------
1 file changed, 9 insertions(+), 6 deletions(-)
diff --git a/pystencils/boundaries/boundaryhandling.py b/pystencils/boundaries/boundaryhandling.py
index 94d126196..afad0221a 100644
--- a/pystencils/boundaries/boundaryhandling.py
+++ b/pystencils/boundaries/boundaryhandling.py
@@ -105,7 +105,8 @@ class BoundaryHandling:
class_ = self.IndexFieldBlockData
class_.to_cpu = to_cpu
class_.to_gpu = to_gpu
- data_handling.add_custom_class(self._index_array_name, class_)
+ gpu = self._target in data_handling._GPU_LIKE_TARGETS
+ data_handling.add_custom_class(self._index_array_name, class_, cpu=True, gpu=gpu)
@property
def data_handling(self):
@@ -253,11 +254,13 @@ class BoundaryHandling:
"""
Writes a VTK field where each cell with the given boundary is marked with 1, other cells are 0
This can be used to display the simulation geometry in Paraview
- :param file_name: vtk filename
- :param boundaries: boundary object, or special string 'domain' for domain cells or special string 'all' for all
- boundary conditions.
- can also be a sequence, to write multiple boundaries to VTK file
- :param ghost_layers: number of ghost layers to write, or True for all, False for none
+
+ Params:
+ file_name: vtk filename
+ boundaries: boundary object, or special string 'domain' for domain cells or special string 'all' for all
+ boundary conditions.
+ can also be a sequence, to write multiple boundaries to VTK file
+ ghost_layers: number of ghost layers to write, or True for all, False for none
"""
if boundaries == 'all':
boundaries = list(self._boundary_object_to_boundary_info.keys()) + ['domain']
--
GitLab