diff options
Diffstat (limited to 'ops/Data/Array/Strided/Array.hs')
-rw-r--r-- | ops/Data/Array/Strided/Array.hs | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/ops/Data/Array/Strided/Array.hs b/ops/Data/Array/Strided/Array.hs index df455c7..9280fe0 100644 --- a/ops/Data/Array/Strided/Array.hs +++ b/ops/Data/Array/Strided/Array.hs @@ -1,12 +1,13 @@ {-# LANGUAGE DataKinds #-} +{-# LANGUAGE ImportQualifiedPost #-} {-# LANGUAGE KindSignatures #-} {-# LANGUAGE ScopedTypeVariables #-} {-# LANGUAGE TypeApplications #-} module Data.Array.Strided.Array where -import qualified Data.List.NonEmpty as NE +import Data.List.NonEmpty qualified as NE import Data.Proxy -import qualified Data.Vector.Storable as VS +import Data.Vector.Storable qualified as VS import Foreign.Storable import GHC.TypeLits @@ -30,7 +31,7 @@ arrayFromVector sh vec shsize = product sh strides = NE.tail (NE.scanr (*) 1 sh) -arrayFromConstant :: (Storable a, KnownNat n) => [Int] -> a -> Array n a +arrayFromConstant :: Storable a => [Int] -> a -> Array n a arrayFromConstant sh x = Array sh (0 <$ sh) 0 (VS.singleton x) arrayRevDims :: [Bool] -> Array n a -> Array n a |