diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/Main.hs | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/test/Main.hs b/test/Main.hs index f779352..a3fa484 100644 --- a/test/Main.hs +++ b/test/Main.hs @@ -241,6 +241,9 @@ tests = checkParallel $ Group "AD" let_ #q (pair (snd_ #p * fst_ #p + #y) #x) $ fst_ #q * #x + snd_ #q * fst_ #p) + ,("build0 const", adTest $ fromNamed $ lambda @(TScal TF64) #x $ body $ + idx0 $ build SZ nil $ #idx :-> const_ 0.0) + ,("build0", adTest $ fromNamed $ lambda @(TArr N0 _) #x $ body $ idx0 $ build SZ (shape #x) $ #idx :-> #x ! #idx) |