Skip to content

Commit bd1d3cb

Browse files
Fix scoping issue-- call adapt directly
1 parent 3d0c9ca commit bd1d3cb

File tree

2 files changed

+10
-4
lines changed

2 files changed

+10
-4
lines changed

src/Remapping/Remapping.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ import ..DataLayouts,
1616
..Hypsography
1717
import ClimaCore.Utilities: half
1818
import ClimaCore.Spaces: cuda_synchronize
19-
import ..to_cpu
19+
import Adapt
2020

2121
using ..RecursiveApply
2222

src/Remapping/distributed_remapping.jl

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -399,10 +399,11 @@ function _Remapper(
399399
)
400400
num_dims = num_hdims
401401
else
402-
cpu_space = if ClimaComms.device(space) isa ClimaComms.AbstractCPUDevice
402+
device = ClimaComms.device(space)
403+
cpu_space = if device isa ClimaComms.AbstractCPUDevice
403404
space
404405
else
405-
to_cpu(space)
406+
Adapt.adapt(Array, space)
406407
end
407408
vert_interpolation_weights =
408409
ArrayType(vertical_interpolation_weights(cpu_space, target_zcoords))
@@ -471,7 +472,12 @@ function _Remapper(
471472
cpu_space = if ClimaComms.device(space) isa ClimaComms.AbstractCPUDevice
472473
space
473474
else
474-
to_cpu(space)
475+
device = ClimaComms.device(space)
476+
cpu_space = if device isa ClimaComms.AbstractCPUDevice
477+
space
478+
else
479+
Adapt.adapt(Array, space)
480+
end
475481
end
476482

477483
vert_interpolation_weights =

0 commit comments

Comments
 (0)