remove view (#380)

impICNF · Feb 27, 2024 · 15681e2 · 15681e2
1 parent 40a1a8a
commit 15681e2
Show file tree

Hide file tree

Showing 8 changed files with 69 additions and 69 deletions.
diff --git a/src/base.jl b/src/base.jl
@@ -153,7 +153,7 @@ end
     Array
 end
 
-@views function inference_sol(
+function inference_sol(
     icnf::AbstractFlows{T, <:VectorMode, INPLACE},
     mode::Mode,
     prob::SciMLBase.AbstractODEProblem{<:AbstractVector{<:Real}, NTuple{2, T}, INPLACE},
@@ -168,7 +168,7 @@ end
     (logp̂x, augs)
 end
 
-@views function inference_sol(
+function inference_sol(
     icnf::AbstractFlows{T, <:MatrixMode, INPLACE},
     mode::Mode,
     prob::SciMLBase.AbstractODEProblem{<:AbstractMatrix{<:Real}, NTuple{2, T}, INPLACE},
@@ -183,7 +183,7 @@ end
     (logp̂x, eachrow(augs))
 end
 
-@views function generate_sol(
+function generate_sol(
     icnf::AbstractFlows{T, <:VectorMode, INPLACE},
     mode::Mode,
     prob::SciMLBase.AbstractODEProblem{<:AbstractVector{<:Real}, NTuple{2, T}, INPLACE},
@@ -196,7 +196,7 @@ end
     z
 end
 
-@views function generate_sol(
+function generate_sol(
     icnf::AbstractFlows{T, <:MatrixMode, INPLACE},
     mode::Mode,
     prob::SciMLBase.AbstractODEProblem{<:AbstractMatrix{<:Real}, NTuple{2, T}, INPLACE},
@@ -209,10 +209,10 @@ end
     z
 end
 
-@inline @views function get_fsol(sol::SciMLBase.AbstractODESolution)
+@inline function get_fsol(sol::SciMLBase.AbstractODESolution)
     last(sol.u)
 end
 
-@inline @views function get_fsol(sol::AbstractArray{T, N}) where {T, N}
+@inline function get_fsol(sol::AbstractArray{T, N}) where {T, N}
     selectdim(sol, N, lastindex(sol, N))
 end
diff --git a/src/base_cond_icnf.jl b/src/base_cond_icnf.jl
@@ -1,6 +1,6 @@
 export inference, generate, loss
 
-@views function inference_prob(
+function inference_prob(
     icnf::AbstractCondICNF{T, <:VectorMode, INPLACE},
     mode::Mode,
     xs::AbstractVector{<:Real},
@@ -30,7 +30,7 @@ export inference, generate, loss
     )
 end
 
-@views function inference_prob(
+function inference_prob(
     icnf::AbstractCondICNF{T, <:MatrixMode, INPLACE},
     mode::Mode,
     xs::AbstractMatrix{<:Real},
@@ -60,7 +60,7 @@ end
     )
 end
 
-@views function generate_prob(
+function generate_prob(
     icnf::AbstractCondICNF{T, <:VectorMode, INPLACE},
     mode::Mode,
     ys::AbstractVector{<:Real},
@@ -91,7 +91,7 @@ end
     )
 end
 
-@views function generate_prob(
+function generate_prob(
     icnf::AbstractCondICNF{T, <:MatrixMode, INPLACE},
     mode::Mode,
     ys::AbstractMatrix{<:Real},

diff --git a/src/base_icnf.jl b/src/base_icnf.jl
@@ -1,6 +1,6 @@
 export inference, generate, loss
 
-@views function inference_prob(
+function inference_prob(
     icnf::AbstractICNF{T, <:VectorMode, INPLACE},
     mode::Mode,
     xs::AbstractVector{<:Real},
@@ -29,7 +29,7 @@ export inference, generate, loss
     )
 end
 
-@views function inference_prob(
+function inference_prob(
     icnf::AbstractICNF{T, <:MatrixMode, INPLACE},
     mode::Mode,
     xs::AbstractMatrix{<:Real},
@@ -58,7 +58,7 @@ end
     )
 end
 
-@views function generate_prob(
+function generate_prob(
     icnf::AbstractICNF{T, <:VectorMode, INPLACE},
     mode::Mode,
     ps::Any,
@@ -88,7 +88,7 @@ end
     )
 end
 
-@views function generate_prob(
+function generate_prob(
     icnf::AbstractICNF{T, <:MatrixMode, INPLACE},
     mode::Mode,
     ps::Any,

diff --git a/src/cond_ffjord.jl b/src/cond_ffjord.jl
@@ -76,7 +76,7 @@ struct CondRNODE{
     λ₂::T
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -99,7 +99,7 @@ end
     vcat(ż, l̇)
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -124,7 +124,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -143,7 +143,7 @@ end
     vcat(ż, l̇)
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -164,7 +164,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -183,7 +183,7 @@ end
     vcat(ż, l̇)
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -204,7 +204,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -234,7 +234,7 @@ end
     end
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -264,7 +264,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -295,7 +295,7 @@ end
     end
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -326,7 +326,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -352,7 +352,7 @@ end
     end
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -378,7 +378,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -409,7 +409,7 @@ end
     end
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -440,7 +440,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -471,7 +471,7 @@ end
     end
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -502,7 +502,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -528,7 +528,7 @@ end
     end
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,

diff --git a/src/cond_planar.jl b/src/cond_planar.jl
@@ -36,7 +36,7 @@ struct CondPlanar{
     rng::RNG
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -65,7 +65,7 @@ end
     vcat(ż, l̇)
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -96,7 +96,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -125,7 +125,7 @@ end
     vcat(ż, l̇)
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -156,7 +156,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -179,7 +179,7 @@ end
     vcat(ż, l̇)
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,
@@ -204,7 +204,7 @@ end
     nothing
 end
 
-@views function augmented_f(
+function augmented_f(
     u::Any,
     p::Any,
     t::Any,
@@ -227,7 +227,7 @@ end
     vcat(ż, l̇)
 end
 
-@views function augmented_f(
+function augmented_f(
     du::Any,
     u::Any,
     p::Any,