diff options
author | Simon Peyton Jones <simonpj@microsoft.com> | 2017-03-29 09:00:02 +0100 |
---|---|---|
committer | Simon Peyton Jones <simonpj@microsoft.com> | 2017-03-29 09:02:53 +0100 |
commit | 8674883c137401873fd53a6963acd33af651c2af (patch) | |
tree | e95de3232b884fe9a7cdc973c0d07cce13e2e1a5 /testsuite/tests/simplCore/should_compile/T13410.hs | |
parent | e07211f752b9b98e2bd6957f126bd537d178041a (diff) | |
download | haskell-8674883c137401873fd53a6963acd33af651c2af.tar.gz |
Allow unbound Refl binders in a RULE
Trac #13410 was failing because we had a RULE with a binder
(c :: t~t)
and the /occurrences/ of c on the LHS were being optimised to Refl,
leaving a binder that would not be filled in by matching the LHS
of the rule.
I flirted with trying to ensure that occurrences (c :: t~t) are
not optimised to Relf, but that turned out to be fragile; it was
being done, for good reasons, in multiple places, including
- TyCoRep.substCoVarBndr
- Simplify.simplCast
- Corecion.mkCoVarCo
So I fixed it in one place by making Rules.matchN deal happily
with an unbound binder (c :: t~t). Quite easy. See "Coercion
variables" in Note [Unbound RULE binders] in Rules.
In addition, I needed to make CoreLint be happy with an bound
RULE binder that is a Relf coercion variable
In debugging this, I was perplexed that occurrences of a variable
(c :: t~t) mysteriously turned into Refl. I found out how it
was happening, and decided to move it:
* In TyCoRep.substCoVarBndr, do not substitute Refl for a
binder (c :: t~t).
* In mkCoVarCo do not optimise (c :: t~t) to Refl.
Instead, we do this optimisation in optCoercion (specifically
opt_co4) where, surprisingly, the optimisation was /not/
being done. This has no effect on what programs compile;
it just moves a relatively-expensive optimisation to optCoercion,
where it seems more properly to belong. It's actually not clear
to me which is really "better", but this way round is less
surprising.
One small simplifying refactoring
* Eliminate TyCoRep.substCoVarBndrCallback, which was only
called locally.
Diffstat (limited to 'testsuite/tests/simplCore/should_compile/T13410.hs')
-rw-r--r-- | testsuite/tests/simplCore/should_compile/T13410.hs | 152 |
1 files changed, 152 insertions, 0 deletions
diff --git a/testsuite/tests/simplCore/should_compile/T13410.hs b/testsuite/tests/simplCore/should_compile/T13410.hs new file mode 100644 index 0000000000..9db017d777 --- /dev/null +++ b/testsuite/tests/simplCore/should_compile/T13410.hs @@ -0,0 +1,152 @@ +{-# LANGUAGE BangPatterns #-} +{-# LANGUAGE FlexibleContexts #-} +{-# LANGUAGE GADTs #-} +{-# LANGUAGE KindSignatures #-} +{-# LANGUAGE MultiParamTypeClasses #-} +{-# LANGUAGE RankNTypes #-} +{-# LANGUAGE TypeFamilies #-} +{-# LANGUAGE UndecidableInstances #-} + +module Data.Vector.Hybrid.Internal (Vector) where + +import Control.Monad (liftM2) +import Data.Functor.Identity (Identity(..)) +import GHC.ST (ST, runST) +import Text.Read (ReadPrec, readPrec) + +----- + +class Monad m => PrimMonad m where + type PrimState m + +instance PrimMonad (ST s) where + type PrimState (ST s) = s + +class GMVector v a where + gmbasicLength :: v s a -> Int + gmbasicUnsafeSlice :: Int -> Int -> v s a -> v s a + gmbasicUnsafeNew :: PrimMonad m => Int -> m (v (PrimState m) a) + gmbasicUnsafeWrite :: PrimMonad m => v (PrimState m) a -> Int -> a -> m () + +type family GMutable (v :: * -> *) :: * -> * -> * + +class GMVector (GMutable v) a => GVector v a where + gbasicUnsafeFreeze :: PrimMonad m => GMutable v (PrimState m) a -> m (v a) + +data Step s a where + Yield :: a -> s -> Step s a + +instance Functor (Step s) where + {-# INLINE fmap #-} + fmap f (Yield x s) = Yield (f x) s + +data Stream m a = forall s. Stream (s -> m (Step s a)) s +data Chunk v a = Chunk Int (forall m. (PrimMonad m, GVector v a) => GMutable v (PrimState m) a -> m ()) +data New v a = New { newrun :: forall s. ST s (GMutable v s a) } +type MBundle m v a = Stream m (Chunk v a) +type Bundle v a = MBundle Identity v a + +mbfromStream :: Monad m => Stream m a -> MBundle m v a +{-# INLINE mbfromStream #-} +mbfromStream (Stream step t) = Stream step' t + where + step' s = do r <- step s + return $ fmap (\x -> Chunk 1 (\v -> gmbasicUnsafeWrite v 0 x)) r + +mbunsafeFromList :: Monad m => [a] -> MBundle m v a +{-# INLINE [1] mbunsafeFromList #-} +mbunsafeFromList xs = mbfromStream (sfromList xs) + +blift :: Monad m => Bundle v a -> MBundle m v a +{-# INLINE [1] blift #-} +blift (Stream vstep t) = Stream (return . runIdentity . vstep) t + +sfromList :: Monad m => [a] -> Stream m a +{-# INLINE sfromList #-} +sfromList zs = Stream step zs + where + step (x:xs) = return (Yield x xs) + step _ = undefined + +sfoldlM :: Monad m => (a -> b -> m a) -> a -> Stream m b -> m a +{-# INLINE [1] sfoldlM #-} +sfoldlM m w (Stream step t) = foldlM_loop w t + where + foldlM_loop z s + = do + r <- step s + case r of + Yield x s' -> do { z' <- m z x; foldlM_loop z' s' } + +gmvunstream :: (PrimMonad m, GVector v a) + => Bundle v a -> m (GMutable v (PrimState m) a) +{-# INLINE [1] gmvunstream #-} +gmvunstream s = gmvmunstreamUnknown (blift s) + +gmvmunstreamUnknown :: (PrimMonad m, GVector v a) + => MBundle m v a -> m (GMutable v (PrimState m) a) +{-# INLINE gmvmunstreamUnknown #-} +gmvmunstreamUnknown s + = do + v <- gmbasicUnsafeNew 0 + (_, _) <- sfoldlM copyChunk (v,0) s + return undefined + where + {-# INLINE [0] copyChunk #-} + copyChunk (v,i) (Chunk n f) + = do + let j = i+n + v' <- if gmbasicLength v < j + then gmbasicUnsafeNew undefined + else return v + f (gmbasicUnsafeSlice i n v') + return (v',j) + +newunstream :: GVector v a => Bundle v a -> New v a +{-# INLINE [1] newunstream #-} +newunstream s = s `seq` New (gmvunstream s) + +gnew :: GVector v a => New v a -> v a +{-# INLINE [1] gnew #-} +gnew m = m `seq` runST (gbasicUnsafeFreeze =<< newrun m) + +gunstream :: GVector v a => Bundle v a -> v a +{-# INLINE gunstream #-} +gunstream s = gnew (newunstream s) + +gfromList :: GVector v a => [a] -> v a +{-# INLINE gfromList #-} +gfromList = gunstream . mbunsafeFromList + +greadPrec :: (GVector v a, Read a) => ReadPrec (v a) +{-# INLINE greadPrec #-} +greadPrec = do + xs <- readPrec + return (gfromList xs) + +----- + +data MVector :: (* -> * -> *) -> (* -> * -> *) -> * -> * -> * where + MV :: !(u s a) -> !(v s b) -> MVector u v s (a, b) + +instance (GMVector u a, GMVector v b) => GMVector (MVector u v) (a, b) where + gmbasicLength (MV ks _) = gmbasicLength ks + gmbasicUnsafeSlice s e (MV ks vs) = MV (gmbasicUnsafeSlice s e ks) (gmbasicUnsafeSlice s e vs) + + gmbasicUnsafeNew n = liftM2 MV (gmbasicUnsafeNew n) (gmbasicUnsafeNew n) + -- Removing this INLINE pragma makes it compile + {-# INLINE gmbasicUnsafeNew #-} + + gmbasicUnsafeWrite (MV ks vs) n (k,v) = do + gmbasicUnsafeWrite ks n k + gmbasicUnsafeWrite vs n v + +data Vector :: (* -> *) -> (* -> *) -> * -> * + +type instance GMutable (Vector u v) = MVector (GMutable u) (GMutable v) + +instance (GVector u a, GVector v b) => GVector (Vector u v) (a, b) where + gbasicUnsafeFreeze = undefined + +instance (GVector u a, GVector v b, Read a, Read b, c ~ (a, b)) => Read (Vector u v c) where + readPrec = greadPrec |