diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2024-04-13 20:54:08 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2024-04-13 20:54:08 +0200 |
| commit | 8a81f7ea9eed9afaec948910caaf0a5c498de6c6 (patch) | |
| tree | 87729f0ed4145645e77fa0b065f6efc83ebb229d /test/Main.hs | |
| parent | d5e02224b16b9d616e5193e0b9c48865b2415699 (diff) | |
Switch to GHC.TypeLits.Nat for shapes
Diffstat (limited to 'test/Main.hs')
| -rw-r--r-- | test/Main.hs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/Main.hs b/test/Main.hs index 8257ff0..7e1a3a1 100644 --- a/test/Main.hs +++ b/test/Main.hs @@ -6,9 +6,9 @@ module Main where import Data.Array.Nested -arr :: Ranked N2 (Shaped [N2, N3] (Double, Int)) +arr :: Ranked N2 (Shaped [2, 3] (Double, Int)) arr = rgenerate (3 ::: 4 ::: IZR) $ \(i ::: j ::: IZR) -> - sgenerate @[N2, N3] (2 ::$ 3 ::$ IZS) $ \(k ::$ l ::$ IZS) -> + sgenerate @[2, 3] (2 ::$ 3 ::$ IZS) $ \(k ::$ l ::$ IZS) -> let s = 24*i + 6*j + 3*k + l in (fromIntegral s, s) |