From fc871ef253dffda12f60fa3ee6fe1c8100df1a36 Mon Sep 17 00:00:00 2001 From: "Navid C. Constantinou" Date: Mon, 1 Apr 2024 14:10:21 +0300 Subject: [PATCH] friendlier for gpu (#3528) --- examples/kelvin_helmholtz_instability.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/kelvin_helmholtz_instability.jl b/examples/kelvin_helmholtz_instability.jl index 9608f83d79..843c23d37f 100644 --- a/examples/kelvin_helmholtz_instability.jl +++ b/examples/kelvin_helmholtz_instability.jl @@ -253,7 +253,7 @@ function estimate_growth_rate(simulation, energy, ω, b; convergence_criterion=1 σ = [] power_method_data = [] compute!(ω) - push!(power_method_data, (ω=collect(interior(ω)[:, 1, :]), b=collect(interior(b)[:, 1, :]), σ=deepcopy(σ))) + push!(power_method_data, (ω=collect(interior(ω, :, 1, :)), b=collect(interior(b, :, 1, :)), σ=deepcopy(σ))) while convergence(σ) > convergence_criterion compute!(energy) @@ -267,7 +267,7 @@ function estimate_growth_rate(simulation, energy, ω, b; convergence_criterion=1 compute!(ω) rescale!(simulation.model, energy) - push!(power_method_data, (ω=collect(interior(ω)[:, 1, :]), b=collect(interior(b)[:, 1, :]), σ=deepcopy(σ))) + push!(power_method_data, (ω=collect(interior(ω, :, 1, :)), b=collect(interior(b, :, 1, :)), σ=deepcopy(σ))) end return σ, power_method_data