summaryrefslogtreecommitdiff
path: root/testsuite/tests/dph/primespj/PrimesVect.hs
diff options
context:
space:
mode:
authorDavid Terei <davidterei@gmail.com>2011-07-20 11:09:03 -0700
committerDavid Terei <davidterei@gmail.com>2011-07-20 11:26:35 -0700
commit16514f272fb42af6e9c7674a9bd6c9dce369231f (patch)
treee4f332b45fe65e2a7a2451be5674f887b42bf199 /testsuite/tests/dph/primespj/PrimesVect.hs
parentebd422aed41048476aa61dd4c520d43becd78682 (diff)
downloadhaskell-16514f272fb42af6e9c7674a9bd6c9dce369231f.tar.gz
Move tests from tests/ghc-regress/* to just tests/*
Diffstat (limited to 'testsuite/tests/dph/primespj/PrimesVect.hs')
-rw-r--r--testsuite/tests/dph/primespj/PrimesVect.hs25
1 files changed, 25 insertions, 0 deletions
diff --git a/testsuite/tests/dph/primespj/PrimesVect.hs b/testsuite/tests/dph/primespj/PrimesVect.hs
new file mode 100644
index 0000000000..34b3a568b7
--- /dev/null
+++ b/testsuite/tests/dph/primespj/PrimesVect.hs
@@ -0,0 +1,25 @@
+{-# LANGUAGE ParallelArrays #-}
+{-# OPTIONS -fvectorise #-}
+module PrimesVect (primesVect)
+
+where
+import Data.Array.Parallel
+import Data.Array.Parallel.Prelude.Int
+import qualified Prelude
+
+primesVect:: Int -> PArray Int
+primesVect n = toPArrayP (primesVect' n)
+
+primesVect':: Int -> [:Int:]
+primesVect' n
+ | n == 1 = emptyP
+ | n == 2 = singletonP 2
+ | otherwise = sps +:+ [: i | i <- enumFromToP (sq+1) n, notMultiple sps i:]
+ where
+
+ sps = primesVect' sq
+ sq = sqrt n
+
+ notMultiple :: [:Int:] -> Int -> Bool
+ notMultiple ps i = andP [: mod i p /= 0 | p <- ps:]
+