some stuff to get it to run on rocm

JuliaMolSim · Sep 9, 2023 · aa2b13f · aa2b13f
1 parent 7c5597e
commit aa2b13f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/spatial.jl b/src/spatial.jl
@@ -867,7 +867,8 @@ function scale_coords!(sys, scale_factor; ignore_molecules=false)
             coords_nounits[i] = wrap_coords(
                     coords_nounits[i] .+ shift_vecs[mi] .- center_shifts[mi], boundary_nounits)
         end
-        sys.coords = coords_nounits * coord_units
+        ArrayType = get_array_type(sys.coords)
+        sys.coords = ArrayType(coords_nounits * coord_units)
     end
     return sys
 end