summaryrefslogtreecommitdiff
path: root/testsuite/tests/dph/smvm/Main.hs
diff options
context:
space:
mode:
Diffstat (limited to 'testsuite/tests/dph/smvm/Main.hs')
-rw-r--r--testsuite/tests/dph/smvm/Main.hs65
1 files changed, 65 insertions, 0 deletions
diff --git a/testsuite/tests/dph/smvm/Main.hs b/testsuite/tests/dph/smvm/Main.hs
new file mode 100644
index 0000000000..e30938bc21
--- /dev/null
+++ b/testsuite/tests/dph/smvm/Main.hs
@@ -0,0 +1,65 @@
+{-# LANGUAGE TypeOperators #-}
+
+import SMVMVect (smvm)
+
+import Control.Exception (evaluate)
+import System.IO
+import System.Environment
+
+import qualified Data.Array.Parallel.Unlifted as U
+import Data.Array.Parallel.Prelude
+import Data.Array.Parallel.PArray as P
+
+
+-- Load sparse matrix from a file
+loadSM :: String
+ -> IO (PArray (PArray (Int, Double)), PArray Double)
+
+loadSM s
+ = do
+ (segd, m, v) <- loadSM' s
+ return $ (nestUSegdPA' segd (fromUArrPA_2' m), fromUArrPA' v)
+
+
+loadSM' :: String
+ -> IO ( U.Segd
+ , U.Array (Int, Double)
+ , U.Array Double)
+loadSM' fname =
+ do
+ h <- openBinaryFile fname ReadMode
+ lengths <- U.hGet h
+ indices <- U.hGet h
+ values <- U.hGet h
+ dv <- U.hGet h
+ let segd = U.lengthsToSegd lengths
+ m = U.zip indices values
+ evaluate lengths
+ evaluate indices
+ evaluate values
+ evaluate dv
+ return (segd, m, dv)
+
+main
+ = do [inFile, outFile] <- getArgs
+ (m, v) <- loadSM inFile
+ let result = smvm m v
+
+ -- ignore wibbles in low-order bits
+ let output
+ = (unlines
+ $ map (take 12)
+ $ map show
+ $ P.toList result)
+ ++ ("SUM = "
+ ++ (take 12 $ show $ sum $ P.toList result)
+ ++ "\n")
+
+ -- check our result against the provided outFile
+ outputCheck <- readFile outFile
+ print $ output == outputCheck
+
+
+
+
+