modular C matrix product

1 files changed, 8 insertions, 8 deletions

diff --git a/packages/base/src/Internal/LAPACK.hs b/packages/base/src/Internal/LAPACK.hs
index 469b0d5..8df568d 100644
--- a/packages/base/src/Internal/LAPACK.hs
+++ b/packages/base/src/Internal/LAPACK.hs
@@ -36,8 +36,8 @@ foreign import ccall unsafe "multiplyR" dgemmc :: CInt -> CInt -> TMMM R
 foreign import ccall unsafe "multiplyC" zgemmc :: CInt -> CInt -> TMMM C
 foreign import ccall unsafe "multiplyF" sgemmc :: CInt -> CInt -> TMMM F
 foreign import ccall unsafe "multiplyQ" cgemmc :: CInt -> CInt -> TMMM Q
-foreign import ccall unsafe "multiplyI" c_multiplyI :: CInt ::> CInt ::> CInt ::> Ok
-foreign import ccall unsafe "multiplyL" c_multiplyL :: Z ::> Z ::> Z ::> Ok
+foreign import ccall unsafe "multiplyI" c_multiplyI :: I -> I ::> I ::> I ::> Ok
+foreign import ccall unsafe "multiplyL" c_multiplyL :: Z -> Z ::> Z ::> Z ::> Ok
 
 isT Matrix{order = ColumnMajor} = 0
 isT Matrix{order = RowMajor} = 1
@@ -68,20 +68,20 @@ multiplyF a b = multiplyAux sgemmc "sgemmc" a b
 multiplyQ :: Matrix (Complex Float) -> Matrix (Complex Float) -> Matrix (Complex Float)
 multiplyQ a b = multiplyAux cgemmc "cgemmc" a b
 
-multiplyI :: Matrix CInt -> Matrix CInt -> Matrix CInt
-multiplyI a b = unsafePerformIO $ do
+multiplyI :: I -> Matrix CInt -> Matrix CInt -> Matrix CInt
+multiplyI m a b = unsafePerformIO $ do
     when (cols a /= rows b) $ error $
         "inconsistent dimensions in matrix product "++ shSize a ++ " x " ++ shSize b
     s <- createMatrix ColumnMajor (rows a) (cols b)
-    app3 c_multiplyI omat a omat b omat s "c_multiplyI"
+    app3 (c_multiplyI m) omat a omat b omat s "c_multiplyI"
     return s
 
-multiplyL :: Matrix Z -> Matrix Z -> Matrix Z
-multiplyL a b = unsafePerformIO $ do
+multiplyL :: Z -> Matrix Z -> Matrix Z -> Matrix Z
+multiplyL m a b = unsafePerformIO $ do
     when (cols a /= rows b) $ error $
         "inconsistent dimensions in matrix product "++ shSize a ++ " x " ++ shSize b
     s <- createMatrix ColumnMajor (rows a) (cols b)
-    app3 c_multiplyL omat a omat b omat s "c_multiplyL"
+    app3 (c_multiplyL m) omat a omat b omat s "c_multiplyL"
     return s
 
 -----------------------------------------------------------------------------