futhark

Safe Haskell	None
Language	GHC2021

Futhark.AD.Rev.Acc

Description

Differentiation related to accumulators in the input program.

Documentation

diffWithAcc :: VjpOps -> Pat Type -> StmAux () -> [(Shape, [VName], Maybe (Lambda SOACS, [SubExp]))] -> Lambda SOACS -> ADM () -> ADM () Source #

diffUpdateAcc :: Pat Type -> StmAux () -> Safety -> VName -> [SubExp] -> [SubExp] -> ADM () -> ADM () Source #