summaryrefslogtreecommitdiff
path: root/testsuite/tests/dph/primespj/PrimesVect.hs
diff options
context:
space:
mode:
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:]
+