From 94cc58a21c0410541d882d4ba128339b5c4ae28a Mon Sep 17 00:00:00 2001 From: Max Date: Tue, 27 Oct 2026 23:21:22 +0100 Subject: [PATCH] Convert the grid points to NumPy in evaluate_field, so it works on the cupy backend. --- src/mpifield/evaluate.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/mpifield/evaluate.py b/src/mpifield/evaluate.py index 6cc2cb3..c5d9111 100644 --- a/src/mpifield/evaluate.py +++ b/src/mpifield/evaluate.py @@ -4,6 +4,7 @@ from typing import TYPE_CHECKING, Any +import cunumpy as xp import numpy as np if TYPE_CHECKING: @@ -74,7 +75,8 @@ def evaluate_field( # Collective under MPI: every rank gathers the global field. data = np.asarray(field.to_numpy()) grid_points = [ - np.asarray(points, dtype=float) for points in field.get_grid_points() + np.asarray(xp.to_numpy(points), dtype=float) + for points in field.get_grid_points() ] lower = [float(bound) for bound in field.lower_bound] upper = [float(bound) for bound in field.upper_bound]