module Examples.Test (afun) where

import Data.Array.Accelerate


afun :: Acc (Matrix Int, Vector (Int, Int))
     -> Acc (Matrix Int)
afun (T2 a b) = generate (I2 2 3) (\(I2 i j) ->
                  let T2 x y = b ! I1 i
                  in i * j + a ! I2 i j + x * y)