Skip to content

Commit 34e1c00

Browse files
use array4_to_xp in MultiFab.py to avoid needing mf_to_cupy and mf_to_dpnp
Signed-off-by: Roelof Groenewald <rgroenewald@tae.com>
1 parent 506101a commit 34e1c00

File tree

1 file changed

+4
-11
lines changed

1 file changed

+4
-11
lines changed

src/amrex/extensions/MultiFab.py

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -175,18 +175,11 @@ def mf_to_xp(self, copy=False, order="F"):
175175
A list of NumPy, CuPy or dpnp n-dimensional arrays, for each local block
176176
in the MultiFab.
177177
"""
178-
import inspect
179-
180-
amr = inspect.getmodule(self)
181-
182-
if amr.Config.have_gpu:
183-
if amr.Config.gpu_backend == "SYCL":
184-
return self.to_dpnp(copy, order)
185-
else: # if not SYCL use cupy
186-
return self.to_cupy(copy, order)
178+
views = []
179+
for mfi in self:
180+
views.append(self.array(mfi).to_xp(copy, order))
187181

188-
# if no GPU, use NumPy
189-
return self.to_numpy(copy, order)
182+
return views
190183

191184

192185
def copy_multifab(amr, self):

0 commit comments

Comments
 (0)