Skip to content

Commit

Permalink
add forward mode tests
Browse files Browse the repository at this point in the history
  • Loading branch information
julmb committed Mar 11, 2024
1 parent 5bf4abc commit 9be541b
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/Regression.hs
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@

module Main (main) where

import qualified Numeric.AD.Mode.Forward as F
import qualified Numeric.AD.Mode.Forward.Double as FD
import qualified Numeric.AD.Mode.Reverse as R
import qualified Numeric.AD.Mode.Reverse.Double as RD

Expand All @@ -20,6 +22,8 @@ main = defaultMain tests

tests :: TestTree
tests = testGroup "tests" [
mode "forward" (\ f -> F.diff' f) (\ f -> F.grad f) (\ f -> F.jacobian f) (\ f -> F.jacobian $ F.grad f),
--mode "forward-double" (\ f -> FD.diff' f) (\ f -> FD.grad f) (\ f -> FD.jacobian f) (\ f -> FD.jacobian $ F.grad f),
mode "reverse" (\ f -> R.diff' f) (\ f -> R.grad f) (\ f -> R.jacobian f) (\ f -> R.hessian f),
mode "reverse-double" (\ f -> RD.diff' f) (\ f -> RD.grad f) (\ f -> RD.jacobian f) (\ f -> RD.hessian f)]

Expand Down

0 comments on commit 9be541b

Please sign in to comment.