jasigal

pub module approx-recip

import smooth

fun approx-recip(iters : int, x : a) : <smooth<a>,div> a {
  var acc := c(1.0)
  var prev := c(1.0)
  repeat(iters) {
    prev := (prev *. (~.)(x -. c(1.0)))
    acc := (acc +. prev)

jasigal

module Main where

import           Data.Array.Accelerate             as A
import           Data.Array.Accelerate.LLVM.Native

g :: Scalar Double -> Scalar Double
g = (runN (A.map f :: Acc (Scalar Double) -> Acc (Scalar Double)))
  where
    f x = let y = recip x
              b = (-y) * y