Add support for make_q_bounds with new datalayouts

CliMA · Oct 26, 2024 · 9ef6066 · 9ef6066
1 parent eb5517f
commit 9ef6066
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/Limiters/quasimonotone.jl b/src/Limiters/quasimonotone.jl
@@ -74,13 +74,27 @@ function make_q_bounds(
     _, _, _, _, Nh = size(ρq)
     return DataLayouts.IFH{S, 2}(similar(parent(ρq), (2, Nf, Nh)))
 end
+function make_q_bounds(
+    ρq::Union{DataLayouts.IHF{S}, DataLayouts.IJHF{S}},
+) where {S}
+    Nf = DataLayouts.ncomponents(ρq)
+    _, _, _, _, Nh = size(ρq)
+    return DataLayouts.IHF{S, 2}(similar(parent(ρq), (2, Nh, Nf)))
+end
 function make_q_bounds(
     ρq::Union{DataLayouts.VIFH{S}, DataLayouts.VIJFH{S}},
 ) where {S}
     Nf = DataLayouts.ncomponents(ρq)
     _, _, _, Nv, Nh = size(ρq)
     return DataLayouts.VIFH{S, Nv, 2}(similar(parent(ρq), (Nv, 2, Nf, Nh)))
 end
+function make_q_bounds(
+    ρq::Union{DataLayouts.VIHF{S}, DataLayouts.VIJHF{S}},
+) where {S}
+    Nf = DataLayouts.ncomponents(ρq)
+    _, _, _, Nv, Nh = size(ρq)
+    return DataLayouts.VIHF{S, Nv, 2}(similar(parent(ρq), (Nv, 2, Nh, Nf)))
+end
 
 
 """