Skip to content

Commit e43098e

Browse files
author
Will Kimmerer
committed
fix matrix memory failures
1 parent af215c3 commit e43098e

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/chainrules/utilityrules.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
function rrule(::typeof(build), I::AbstractVector, J::AbstractVector, J::AbstractVector, m, n, combine::typeof(+))
2+
project_V = ProjectTo(V)
3+
function sparse_pullback(ΔΩ)
4+
ΔΩ = unthunk(ΔΩ)
5+
ΔV = project_V()

0 commit comments

Comments
 (0)