Skip to content

Commit

Permalink
remove warnings
Browse files Browse the repository at this point in the history
  • Loading branch information
wavewave committed Jun 1, 2022
1 parent 241f6d4 commit 51103de
Showing 1 changed file with 6 additions and 10 deletions.
16 changes: 6 additions & 10 deletions examples/grad-descent/Main.hs
Original file line number Diff line number Diff line change
Expand Up @@ -5,14 +5,10 @@

module Main where

import Categorifier.C.Codegen.FFI.ArraysCC (fromArraysCC)
import Categorifier.C.Codegen.FFI.Spec (SBVFunCall)
import Categorifier.C.Codegen.FFI.TH (embedFunction)
import Categorifier.C.Generate (writeCFiles)
import Categorifier.C.KTypes.C (C (unsafeC))
import Categorifier.C.KTypes.KLiteral (kliteral)
import Data.List (iterate)
import Data.Proxy (Proxy (..))
import Data.Foldable (traverse_)
import F
( Input (..),
Output (..),
Expand All @@ -36,18 +32,18 @@ step ::
((Double, Double) -> IO (Double, Double)) ->
(Double, Double) ->
IO (Double, Double)
step f df (x0, y0) = do
step _f df (x0, y0) = do
(dfdx, dfdy) <- df (x0, y0)
let (x1, y1) = (x0 - gamma * dfdx, y0 - gamma * dfdy)
pure (x1, y1)

iterateNM :: (Monad m) => Int -> (a -> m a) -> a -> m [a]
iterateNM n f x0 = go n f x0 id
iterateNM n f x0 = go n x0 id
where
go k f x acc
go k x acc
| k > 0 = do
y <- f x
go (k - 1) f y (acc . (y :))
y <- f x
go (k - 1) y (acc . (y :))
| otherwise = pure (acc [])

main :: IO ()
Expand Down

0 comments on commit 51103de

Please sign in to comment.