Skip to content

Commit daa4cb2

Browse files
committed
Added the zeromatrix function to NamedArrayPartition
A function similar to what is implemented commit 2094a78 but for NamedArrayPartitions rather than ArrayPartitions. Tested privately to work with Implicit solvers in OrdinaryDiffEq.jl
1 parent cd9a975 commit daa4cb2

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/named_array_partition.jl

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -145,6 +145,13 @@ end
145145
return dest
146146
end
147147

148+
#Overwrite ArrayInterface zeromatrix to work with NamedArrayPartitions & implicit solvers within OrdinaryDiffEq
149+
function ArrayInterface.zeromatrix(A::NamedArrayPartition)
150+
B = ArrayPartition(A)
151+
x = reduce(vcat,vec.(B.x))
152+
x .* x' .* false
153+
end
154+
148155
# `x = find_NamedArrayPartition(x)` returns the first `NamedArrayPartition` among broadcast arguments.
149156
find_NamedArrayPartition(bc::Base.Broadcast.Broadcasted) = find_NamedArrayPartition(bc.args)
150157
function find_NamedArrayPartition(args::Tuple)

0 commit comments

Comments
 (0)