dph-examples-0.7.0.2: examples/spectral/SMVM/dph/Vector.hs

{-# LANGUAGE BangPatterns #-}

module Vector (smvm) where
import qualified Data.Vector            as V
import qualified Data.Vector.Unboxed    as U

{-# NOINLINE smvm #-}
smvm    :: Int
        -> V.Vector (U.Vector (Int, Double))
        -> U.Vector Double
        -> U.Vector Double

smvm _ matrix vector
 = let term (ix, coeff) = coeff * (vector U.! ix)
       sumRow row       = U.sum (U.map term row)
   in  U.convert $ V.map sumRow matrix