module Crypto.Lol.Benchmarks.CycBenches (cycBenches1, cycBenches2) where
import Control.Applicative
import Control.Monad.Random hiding (lift)
import Crypto.Lol
import Crypto.Lol.Benchmarks
import Crypto.Lol.Cyclotomic.Tensor (TElt)
import Crypto.Lol.Types
import Crypto.Random
cycBenches1 :: (Monad rnd, _) => Proxy '(t,m,r) -> Proxy gen -> rnd Benchmark
cycBenches1 ptmr pgen = benchGroup "Cyc" $ ($ ptmr) <$> [
genBenchArgs "unzipPow" bench_unzipCycPow,
genBenchArgs "unzipDec" bench_unzipCycDec,
genBenchArgs "unzipCRT" bench_unzipCycCRT,
genBenchArgs "zipWith (*)" bench_mul,
genBenchArgs "crt" bench_crt,
genBenchArgs "crtInv" bench_crtInv,
genBenchArgs "l" bench_l,
genBenchArgs "lInv" bench_lInv,
genBenchArgs "*g Pow" bench_mulgPow,
genBenchArgs "*g Dec" bench_mulgDec,
genBenchArgs "*g CRT" bench_mulgCRT,
genBenchArgs "divg Pow" bench_divgPow,
genBenchArgs "divg Dec" bench_divgDec,
genBenchArgs "divg CRT" bench_divgCRT,
genBenchArgs "lift" bench_liftPow,
genBenchArgs "error" (bench_errRounded 0.1) . addGen pgen
]
cycBenches2 :: (Monad rnd, _) => Proxy '(t,m,m',r) -> rnd Benchmark
cycBenches2 p = benchGroup "Cyc" $ ($ p) <$> [
genBenchArgs "twacePow" bench_twacePow,
genBenchArgs "twaceDec" bench_twaceDec,
genBenchArgs "twaceCRT" bench_twaceCRT,
genBenchArgs "embedPow" bench_embedPow,
genBenchArgs "embedDec" bench_embedDec,
genBenchArgs "embedCRT" bench_embedCRT
]
bench_unzipCycPow :: _ => Cyc t m (r,r) -> Bench '(t,m,r)
bench_unzipCycPow = bench unzipCyc . advisePow
bench_unzipCycDec :: _ => Cyc t m (r,r) -> Bench '(t,m,r)
bench_unzipCycDec = bench unzipCyc . adviseDec
bench_unzipCycCRT :: _ => Cyc t m (r,r) -> Bench '(t,m,r)
bench_unzipCycCRT = bench unzipCyc . adviseCRT
bench_mul :: _ => Cyc t m r -> Cyc t m r -> Bench '(t,m,r)
bench_mul a b =
let a' = adviseCRT a
b' = adviseCRT b
in bench (a' *) b'
bench_crt :: _ => Cyc t m r -> Bench '(t,m,r)
bench_crt = bench adviseCRT . advisePow
bench_crtInv :: _ => Cyc t m r -> Bench '(t,m,r)
bench_crtInv = bench advisePow . adviseCRT
bench_l :: _ => Cyc t m r -> Bench '(t,m,r)
bench_l = bench advisePow . adviseDec
bench_lInv :: _ => Cyc t m r -> Bench '(t,m,r)
bench_lInv = bench adviseDec . advisePow
bench_liftPow :: _ => Cyc t m r -> Bench '(t,m,r)
bench_liftPow = bench (liftCyc Pow) . advisePow
bench_mulgPow :: _ => Cyc t m r -> Bench '(t,m,r)
bench_mulgPow = bench mulG . advisePow
bench_mulgDec :: _ => Cyc t m r -> Bench '(t,m,r)
bench_mulgDec = bench mulG . adviseDec
bench_mulgCRT :: _ => Cyc t m r -> Bench '(t,m,r)
bench_mulgCRT = bench mulG . adviseCRT
bench_divgPow :: _ => Cyc t m r -> Bench '(t,m,r)
bench_divgPow = bench divG . advisePow . mulG
bench_divgDec :: _ => Cyc t m r -> Bench '(t,m,r)
bench_divgDec = bench divG . adviseDec . mulG
bench_divgCRT :: _ => Cyc t m r -> Bench '(t,m,r)
bench_divgCRT = bench divG . adviseCRT
bench_errRounded :: forall t m r gen . (TElt t r, Fact m, CryptoRandomGen gen, _)
=> Double -> Bench '(t,m,r,gen)
bench_errRounded v = benchIO $ do
gen <- newGenIO
return $ evalRand (errorRounded v :: Rand (CryptoRand gen) (Cyc t m (LiftOf r))) gen
bench_twacePow :: forall t m m' r . (Fact m, _)
=> Cyc t m' r -> Bench '(t,m,m',r)
bench_twacePow = bench (twace :: Cyc t m' r -> Cyc t m r) . advisePow
bench_twaceDec :: forall t m m' r . (Fact m, _)
=> Cyc t m' r -> Bench '(t,m,m',r)
bench_twaceDec = bench (twace :: Cyc t m' r -> Cyc t m r) . adviseDec
bench_twaceCRT :: forall t m m' r . (Fact m, _)
=> Cyc t m' r -> Bench '(t,m,m',r)
bench_twaceCRT = bench (twace :: Cyc t m' r -> Cyc t m r) . adviseCRT
bench_embedPow :: forall t m m' r . (Fact m', _)
=> Cyc t m r -> Bench '(t,m,m',r)
bench_embedPow = bench (advisePow . embed :: Cyc t m r -> Cyc t m' r) . advisePow
bench_embedDec :: forall t m m' r . (Fact m', _)
=> Cyc t m r -> Bench '(t,m,m',r)
bench_embedDec = bench (adviseDec . embed :: Cyc t m r -> Cyc t m' r) . adviseDec
bench_embedCRT :: forall t m m' r . (Fact m', _)
=> Cyc t m r -> Bench '(t,m,m',r)
bench_embedCRT = bench (adviseCRT . embed :: Cyc t m r -> Cyc t m' r) . adviseCRT