Merge pull request haskell-repa#15 from lehins/append

dintxaustip · Aug 21, 2017 · be40870 · be40870
2 parents 8639803 + ac13a94
commit be40870
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/repa/Data/Array/Repa/Operators/IndexSpace.hs b/repa/Data/Array/Repa/Operators/IndexSpace.hs
@@ -54,8 +54,8 @@ append arr1 arr2
  where
         (_ :. n)        = extent arr1
 
-        fnExtent (sh :. i) (_  :. j)
-                = sh :. (i + j)
+        fnExtent (sh1 :. i) (sh2  :. j)
+                = intersectDim sh1 sh2 :. (i + j)
 
         fnElem f1 f2 (sh :. i)
                 | i < n         = f1 (sh :. i)