From 0b8cb3bc5b77e5c4703f4a5c31b066e946b8c720 Mon Sep 17 00:00:00 2001 From: Michael Schlottke-Lakemper Date: Fri, 3 Nov 2023 15:21:18 +0100 Subject: [PATCH] Undo change to create_cache --- src/indicators_1d.jl | 16 ++++++++-------- src/indicators_2d.jl | 20 ++++++++++---------- 2 files changed, 18 insertions(+), 18 deletions(-) diff --git a/src/indicators_1d.jl b/src/indicators_1d.jl index e333b0c..ee9ff54 100644 --- a/src/indicators_1d.jl +++ b/src/indicators_1d.jl @@ -7,14 +7,14 @@ # this method is used when the indicator is constructed as for shock-capturing volume integrals # empty cache is default -function Trixi.create_cache(::Type{<:IndicatorNeuralNetwork}, - equations::AbstractEquations{1}, basis::LobattoLegendreBasis) +function create_cache(::Type{<:IndicatorNeuralNetwork}, + equations::AbstractEquations{1}, basis::LobattoLegendreBasis) return NamedTuple() end # cache for NeuralNetworkPerssonPeraire-type indicator -function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}}, - equations::AbstractEquations{1}, basis::LobattoLegendreBasis) +function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}}, + equations::AbstractEquations{1}, basis::LobattoLegendreBasis) alpha = Vector{real(basis)}() alpha_tmp = similar(alpha) A = Array{real(basis), ndims(equations)} @@ -27,8 +27,8 @@ function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPe end # cache for NeuralNetworkRayHesthaven-type indicator -function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}}, - equations::AbstractEquations{1}, basis::LobattoLegendreBasis) +function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}}, + equations::AbstractEquations{1}, basis::LobattoLegendreBasis) alpha = Vector{real(basis)}() alpha_tmp = similar(alpha) A = Array{real(basis), ndims(equations)} @@ -41,8 +41,8 @@ function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHestha end # this method is used when the indicator is constructed as for AMR -function Trixi.create_cache(typ::Type{<:IndicatorNeuralNetwork}, - mesh, equations::AbstractEquations{1}, dg::DGSEM, cache) +function create_cache(typ::Type{<:IndicatorNeuralNetwork}, + mesh, equations::AbstractEquations{1}, dg::DGSEM, cache) create_cache(typ, equations, dg.basis) end diff --git a/src/indicators_2d.jl b/src/indicators_2d.jl index 9ed38b3..42e2ad7 100644 --- a/src/indicators_2d.jl +++ b/src/indicators_2d.jl @@ -7,14 +7,14 @@ # this method is used when the indicator is constructed as for shock-capturing volume integrals # empty cache is default -function Trixi.create_cache(::Type{IndicatorNeuralNetwork}, - equations::AbstractEquations{2}, basis::LobattoLegendreBasis) +function create_cache(::Type{IndicatorNeuralNetwork}, + equations::AbstractEquations{2}, basis::LobattoLegendreBasis) return NamedTuple() end # cache for NeuralNetworkPerssonPeraire-type indicator -function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}}, - equations::AbstractEquations{2}, basis::LobattoLegendreBasis) +function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPeraire}}, + equations::AbstractEquations{2}, basis::LobattoLegendreBasis) alpha = Vector{real(basis)}() alpha_tmp = similar(alpha) A = Array{real(basis), ndims(equations)} @@ -30,8 +30,8 @@ function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkPerssonPe end # cache for NeuralNetworkRayHesthaven-type indicator -function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}}, - equations::AbstractEquations{2}, basis::LobattoLegendreBasis) +function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHesthaven}}, + equations::AbstractEquations{2}, basis::LobattoLegendreBasis) alpha = Vector{real(basis)}() alpha_tmp = similar(alpha) A = Array{real(basis), ndims(equations)} @@ -50,8 +50,8 @@ function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkRayHestha end # cache for NeuralNetworkCNN-type indicator -function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkCNN}}, - equations::AbstractEquations{2}, basis::LobattoLegendreBasis) +function create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkCNN}}, + equations::AbstractEquations{2}, basis::LobattoLegendreBasis) alpha = Vector{real(basis)}() alpha_tmp = similar(alpha) A = Array{real(basis), ndims(equations)} @@ -69,8 +69,8 @@ function Trixi.create_cache(::Type{IndicatorNeuralNetwork{NeuralNetworkCNN}}, end # this method is used when the indicator is constructed as for AMR -function Trixi.create_cache(typ::Type{<:IndicatorNeuralNetwork}, - mesh, equations::AbstractEquations{2}, dg::DGSEM, cache) +function create_cache(typ::Type{<:IndicatorNeuralNetwork}, + mesh, equations::AbstractEquations{2}, dg::DGSEM, cache) create_cache(typ, equations, dg.basis) end