diff options
author | David Terei <davidterei@gmail.com> | 2011-07-20 11:09:03 -0700 |
---|---|---|
committer | David Terei <davidterei@gmail.com> | 2011-07-20 11:26:35 -0700 |
commit | 16514f272fb42af6e9c7674a9bd6c9dce369231f (patch) | |
tree | e4f332b45fe65e2a7a2451be5674f887b42bf199 /testsuite/tests/dph/primespj/PrimesVect.hs | |
parent | ebd422aed41048476aa61dd4c520d43becd78682 (diff) | |
download | haskell-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.hs | 25 |
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:] + |