From 5f36c0135208c4cd802365f9118288b5bf1042ef Mon Sep 17 00:00:00 2001 From: Sikorski Date: Sat, 11 May 2024 16:35:01 +0200 Subject: [PATCH] store only features, not the coordinates on gpu --- src/simulation.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/simulation.jl b/src/simulation.jl index 599d308..36f4802 100644 --- a/src/simulation.jl +++ b/src/simulation.jl @@ -90,8 +90,8 @@ end #features(sim::SimulationData, x) = sim.featurizer(x) -gpu(d::SimulationData) = SimulationData(d.sim, gpu(d.features), gpu(d.coords), d.featurizer) -cpu(d::SimulationData) = SimulationData(d.sim, cpu(d.features), cpu(d.coords), d.featurizer) +gpu(d::SimulationData) = SimulationData(d.sim, gpu(d.features), d.coords, d.featurizer) +cpu(d::SimulationData) = SimulationData(d.sim, cpu(d.features), d.coords, d.featurizer) features(d::SimulationData, x) = d.featurizer(x)