As identifiers are z-encoded instead of extended they are no longer case sensitive...
[matthijs/master-project/cλash.git] / mmult.hs
1 {-# LANGUAGE TypeOperators, TemplateHaskell, TypeFamilies, ScopedTypeVariables, FlexibleContexts #-}
2
3 module MMult where
4
5 -- hide default prelude functions
6 import qualified Prelude as P
7
8 -- import CλaSH specific types
9 import CLasH.HardwareTypes
10 import CLasH.Translator.Annotations
11
12 type Word = Signed D16
13
14 foldl1 f xs = foldl f (head xs) (tail xs)
15 as ** bs    = foldl1 (+) (zipWith (*) as bs)
16
17 {-# ANN mmult2x4_4x3 TopEntity #-}
18 mmult2x4_4x3 :: Vector D2 (Vector D4 Word) -> Vector D4 (Vector D3 Word) -> Vector D2 (Vector D3 Word)
19 mmult2x4_4x3 a b = mmult a b
20
21 mmult xss yss = map f xss
22  where
23    f xs  = map (xs **) colsy
24    colsy = transpose yss (iterate (1+) 0)
25
26 transpose :: 
27  ( PositiveT s1
28  , PositiveT s2
29  ) => Vector s1 (Vector s2 a) -> 
30  Vector s2 (Index s2) -> 
31  Vector s2 (Vector s1 a)
32 transpose m ixs = map (\x-> map (!x) m) ixs