futhark-0.21.12: An optimising compiler for a functional, array-oriented language.

Safe Haskell	None
Language	Haskell2010

Futhark.AD.Rev.Loop

Synopsis

diffLoop :: (Stms SOACS -> ADM ()) -> Pat Type -> StmAux () -> Exp SOACS -> ADM () -> ADM ()
stripmineStms :: Stms SOACS -> ADM (Stms SOACS)

Documentation

diffLoop :: (Stms SOACS -> ADM ()) -> Pat Type -> StmAux () -> Exp SOACS -> ADM () -> ADM () Source #

Transforms a loop into its reverse-mode derivative.

stripmineStms :: Stms SOACS -> ADM (Stms SOACS) Source #