{-# LANGUAGE TypeOperators #-}

module SMVMPrim
where

import Data.Array.Parallel.Unlifted as U

{-
type SparseMatrix = U.SArray (Int :*: Double)
type SparseVector = U.Array (Int :*: Double)
type Vector       = U.Array Double
-}

smvm :: U.Segd -> U.Array (Int :*: Double) -> U.Array Double -> U.Array Double
smvm segd m v = U.sum_s segd (U.zipWith (*) (U.bpermute v (U.fsts m))
                                            (U.snds m))
              

{-
smvm :: SparseMatrix -> Vector -> Vector
smvm sm v = U.sum_s (U.zipWith_s (*) (U.bpermute_s' v (U.fst_s sm)) (U.snd_s sm))
-}