module Numeric.LAPACK.ComfortArray.ComplexFloat where
import qualified Numeric.LAPACK.FFI.ComplexFloat as FFI
import qualified Numeric.Netlib.ComfortArray.Utility as Call
import Numeric.Netlib.ComfortArray.Utility (ZeroInt, (^!))
import qualified Data.Array.Comfort.Storable.Mutable as MutArray
import qualified Data.Array.Comfort.Storable as Array
import Data.Array.Comfort.Storable.Mutable (IOArray)
import Data.Array.Comfort.Storable (Array)
import Data.Complex (Complex)
import Foreign.Storable.Complex ()
import Foreign.Storable (peek)
import Foreign.Ptr (Ptr, FunPtr)
import Foreign.C.String (castCCharToChar)
import Foreign.C.Types (CInt)
import Control.Monad.Trans.Cont (evalContT)
import Control.Monad.IO.Class (liftIO)
import Control.Applicative (pure, (<*>))
bbcsd ::
Char ->
Char ->
Char ->
Char ->
Char ->
Int ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt Float, Int)
bbcsd jobu1 jobu2 jobv1t jobv2t trans m theta phi u1 u2 v1t v2t lrwork = do
let thetaDim0 = Call.sizes1 $ MutArray.shape theta
let phiDim0 = Call.sizes1 $ MutArray.shape phi
let (u1Dim0,u1Dim1) = Call.sizes2 $ MutArray.shape u1
let (u2Dim0,u2Dim1) = Call.sizes2 $ MutArray.shape u2
let (v1tDim0,v1tDim1) = Call.sizes2 $ MutArray.shape v1t
let (v2tDim0,v2tDim1) = Call.sizes2 $ MutArray.shape v2t
let q = thetaDim0
let p = u1Dim0
let ldu1 = u1Dim1
let ldu2 = u2Dim1
let ldv1t = v1tDim1
let ldv2t = v2tDim1
Call.assert "bbcsd: q-1 == phiDim0" (q-1 == phiDim0)
Call.assert "bbcsd: m-p == u2Dim0" (m-p == u2Dim0)
Call.assert "bbcsd: q == v1tDim0" (q == v1tDim0)
Call.assert "bbcsd: m-q == v2tDim0" (m-q == v2tDim0)
b11d <- Call.newArray1 q
b11e <- Call.newArray1 (q-1)
b12d <- Call.newArray1 q
b12e <- Call.newArray1 (q-1)
b21d <- Call.newArray1 q
b21e <- Call.newArray1 (q-1)
b22d <- Call.newArray1 q
b22e <- Call.newArray1 (q-1)
rwork <- Call.newArray1 (maximum[1,lrwork])
evalContT $ do
jobu1Ptr <- Call.char jobu1
jobu2Ptr <- Call.char jobu2
jobv1tPtr <- Call.char jobv1t
jobv2tPtr <- Call.char jobv2t
transPtr <- Call.char trans
mPtr <- Call.cint m
pPtr <- Call.cint p
qPtr <- Call.cint q
thetaPtr <- Call.ioarray theta
phiPtr <- Call.ioarray phi
u1Ptr <- Call.ioarray u1
ldu1Ptr <- Call.cint ldu1
u2Ptr <- Call.ioarray u2
ldu2Ptr <- Call.cint ldu2
v1tPtr <- Call.ioarray v1t
ldv1tPtr <- Call.cint ldv1t
v2tPtr <- Call.ioarray v2t
ldv2tPtr <- Call.cint ldv2t
b11dPtr <- Call.ioarray b11d
b11ePtr <- Call.ioarray b11e
b12dPtr <- Call.ioarray b12d
b12ePtr <- Call.ioarray b12e
b21dPtr <- Call.ioarray b21d
b21ePtr <- Call.ioarray b21e
b22dPtr <- Call.ioarray b22d
b22ePtr <- Call.ioarray b22e
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
infoPtr <- Call.alloca
liftIO $ FFI.bbcsd jobu1Ptr jobu2Ptr jobv1tPtr jobv2tPtr transPtr mPtr pPtr qPtr thetaPtr phiPtr u1Ptr ldu1Ptr u2Ptr ldu2Ptr v1tPtr ldv1tPtr v2tPtr ldv2tPtr b11dPtr b11ePtr b12dPtr b12ePtr b21dPtr b21ePtr b22dPtr b22ePtr rworkPtr lrworkPtr infoPtr
liftIO $ pure (,,,,,,,,)
<*> Call.freezeArray b11d
<*> Call.freezeArray b11e
<*> Call.freezeArray b12d
<*> Call.freezeArray b12e
<*> Call.freezeArray b21d
<*> Call.freezeArray b21e
<*> Call.freezeArray b22d
<*> Call.freezeArray b22e
<*> fmap fromIntegral (peek infoPtr)
bdsqr ::
Char ->
Int ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
bdsqr uplo nru d e vt u c = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let (vtDim0,vtDim1) = Call.sizes2 $ MutArray.shape vt
let (uDim0,uDim1) = Call.sizes2 $ MutArray.shape u
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let n = dDim0
let ncvt = vtDim0
let ldvt = vtDim1
let ldu = uDim1
let ncc = cDim0
let ldc = cDim1
Call.assert "bdsqr: n-1 == eDim0" (n-1 == eDim0)
Call.assert "bdsqr: n == uDim0" (n == uDim0)
rwork <- Call.newArray1 (4*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
ncvtPtr <- Call.cint ncvt
nruPtr <- Call.cint nru
nccPtr <- Call.cint ncc
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
vtPtr <- Call.ioarray vt
ldvtPtr <- Call.cint ldvt
uPtr <- Call.ioarray u
lduPtr <- Call.cint ldu
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.bdsqr uploPtr nPtr ncvtPtr nruPtr nccPtr dPtr ePtr vtPtr ldvtPtr uPtr lduPtr cPtr ldcPtr rworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
gbbrd ::
Char ->
Int ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int)
gbbrd vect m kl ku ab ldq ldpt c = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let n = abDim0
let ldab = abDim1
let ncc = cDim0
let ldc = cDim1
d <- Call.newArray1 (minimum[m,n])
e <- Call.newArray1 (minimum[m,n]-1)
q <- Call.newArray2 m ldq
pt <- Call.newArray2 n ldpt
work <- Call.newArray1 (maximum[m,n])
rwork <- Call.newArray1 (maximum[m,n])
evalContT $ do
vectPtr <- Call.char vect
mPtr <- Call.cint m
nPtr <- Call.cint n
nccPtr <- Call.cint ncc
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
ptPtr <- Call.ioarray pt
ldptPtr <- Call.cint ldpt
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gbbrd vectPtr mPtr nPtr nccPtr klPtr kuPtr abPtr ldabPtr dPtr ePtr qPtr ldqPtr ptPtr ldptPtr cPtr ldcPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray d
<*> Call.freezeArray e
<*> Call.freezeArray q
<*> Call.freezeArray pt
<*> fmap fromIntegral (peek infoPtr)
gbcon ::
Char ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Float ->
IO (Float, Int)
gbcon norm kl ku ab ipiv anorm = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = abDim0
let ldab = abDim1
Call.assert "gbcon: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
normPtr <- Call.char norm
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
ipivPtr <- Call.array ipiv
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gbcon normPtr nPtr klPtr kuPtr abPtr ldabPtr ipivPtr anormPtr rcondPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
gbequ ::
Int ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Float, Float, Float, Int)
gbequ m kl ku ab = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
r <- Call.newArray1 m
c <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
rPtr <- Call.ioarray r
cPtr <- Call.ioarray c
rowcndPtr <- Call.alloca
colcndPtr <- Call.alloca
amaxPtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.gbequ mPtr nPtr klPtr kuPtr abPtr ldabPtr rPtr cPtr rowcndPtr colcndPtr amaxPtr infoPtr
liftIO $ pure (,,,,,)
<*> Call.freezeArray r
<*> Call.freezeArray c
<*> peek rowcndPtr
<*> peek colcndPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
gbequb ::
Int ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Float, Float, Float, Int)
gbequb m kl ku ab = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
r <- Call.newArray1 m
c <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
rPtr <- Call.ioarray r
cPtr <- Call.ioarray c
rowcndPtr <- Call.alloca
colcndPtr <- Call.alloca
amaxPtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.gbequb mPtr nPtr klPtr kuPtr abPtr ldabPtr rPtr cPtr rowcndPtr colcndPtr amaxPtr infoPtr
liftIO $ pure (,,,,,)
<*> Call.freezeArray r
<*> Call.freezeArray c
<*> peek rowcndPtr
<*> peek colcndPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
gbrfs ::
Char ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
gbrfs trans kl ku ab afb ipiv b x = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let (afbDim0,afbDim1) = Call.sizes2 $ Array.shape afb
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = abDim0
let ldab = abDim1
let ldafb = afbDim1
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "gbrfs: n == afbDim0" (n == afbDim0)
Call.assert "gbrfs: n == ipivDim0" (n == ipivDim0)
Call.assert "gbrfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
transPtr <- Call.char trans
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
nrhsPtr <- Call.cint nrhs
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
afbPtr <- Call.array afb
ldafbPtr <- Call.cint ldafb
ipivPtr <- Call.array ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gbrfs transPtr nPtr klPtr kuPtr nrhsPtr abPtr ldabPtr afbPtr ldafbPtr ipivPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
gbsv ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
gbsv kl ku ab b = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = abDim0
let ldab = abDim1
let nrhs = bDim0
let ldb = bDim1
ipiv <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
nrhsPtr <- Call.cint nrhs
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.gbsv nPtr klPtr kuPtr nrhsPtr abPtr ldabPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
gbsvx ::
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
Char ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Char, Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
gbsvx fact trans kl ku ab afb ipiv equed r c b ldx = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (afbDim0,afbDim1) = Call.sizes2 $ MutArray.shape afb
let ipivDim0 = Call.sizes1 $ MutArray.shape ipiv
let rDim0 = Call.sizes1 $ MutArray.shape r
let cDim0 = Call.sizes1 $ MutArray.shape c
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = abDim0
let ldab = abDim1
let ldafb = afbDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "gbsvx: n == afbDim0" (n == afbDim0)
Call.assert "gbsvx: n == ipivDim0" (n == ipivDim0)
Call.assert "gbsvx: n == rDim0" (n == rDim0)
Call.assert "gbsvx: n == cDim0" (n == cDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
transPtr <- Call.char trans
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
nrhsPtr <- Call.cint nrhs
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
afbPtr <- Call.ioarray afb
ldafbPtr <- Call.cint ldafb
ipivPtr <- Call.ioarray ipiv
equedPtr <- Call.char equed
rPtr <- Call.ioarray r
cPtr <- Call.ioarray c
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gbsvx factPtr transPtr nPtr klPtr kuPtr nrhsPtr abPtr ldabPtr afbPtr ldafbPtr ipivPtr equedPtr rPtr cPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> fmap castCCharToChar (peek equedPtr)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
gbtf2 ::
Int ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
gbtf2 m kl ku ab = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
ipiv <- Call.newArray1 (minimum[m,n])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.gbtf2 mPtr nPtr klPtr kuPtr abPtr ldabPtr ipivPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
gbtrf ::
Int ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
gbtrf m kl ku ab = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
ipiv <- Call.newArray1 (minimum[m,n])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.gbtrf mPtr nPtr klPtr kuPtr abPtr ldabPtr ipivPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
gbtrs ::
Char ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
gbtrs trans kl ku ab ipiv b = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = abDim0
let ldab = abDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "gbtrs: n == ipivDim0" (n == ipivDim0)
evalContT $ do
transPtr <- Call.char trans
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
nrhsPtr <- Call.cint nrhs
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.gbtrs transPtr nPtr klPtr kuPtr nrhsPtr abPtr ldabPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
gebak ::
Char ->
Char ->
Int ->
Int ->
Array ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
gebak job side ilo ihi scale v = do
let scaleDim0 = Call.sizes1 $ Array.shape scale
let (vDim0,vDim1) = Call.sizes2 $ MutArray.shape v
let n = scaleDim0
let m = vDim0
let ldv = vDim1
evalContT $ do
jobPtr <- Call.char job
sidePtr <- Call.char side
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
scalePtr <- Call.array scale
mPtr <- Call.cint m
vPtr <- Call.ioarray v
ldvPtr <- Call.cint ldv
infoPtr <- Call.alloca
liftIO $ FFI.gebak jobPtr sidePtr nPtr iloPtr ihiPtr scalePtr mPtr vPtr ldvPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
gebal ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int, Int, Array ZeroInt Float, Int)
gebal job a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
scale <- Call.newArray1 n
evalContT $ do
jobPtr <- Call.char job
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
iloPtr <- Call.alloca
ihiPtr <- Call.alloca
scalePtr <- Call.ioarray scale
infoPtr <- Call.alloca
liftIO $ FFI.gebal jobPtr nPtr aPtr ldaPtr iloPtr ihiPtr scalePtr infoPtr
liftIO $ pure (,,,)
<*> fmap fromIntegral (peek iloPtr)
<*> fmap fromIntegral (peek ihiPtr)
<*> Call.freezeArray scale
<*> fmap fromIntegral (peek infoPtr)
gebd2 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Int)
gebd2 m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
d <- Call.newArray1 (minimum[m,n])
e <- Call.newArray1 (minimum[m,n]-1)
tauq <- Call.newArray1 (minimum[m,n])
taup <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[m,n])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
tauqPtr <- Call.ioarray tauq
taupPtr <- Call.ioarray taup
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.gebd2 mPtr nPtr aPtr ldaPtr dPtr ePtr tauqPtr taupPtr workPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray d
<*> Call.freezeArray e
<*> Call.freezeArray tauq
<*> Call.freezeArray taup
<*> fmap fromIntegral (peek infoPtr)
gebrd ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Int)
gebrd m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
d <- Call.newArray1 (minimum[m,n])
e <- Call.newArray1 (minimum[m,n]-1)
tauq <- Call.newArray1 (minimum[m,n])
taup <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
tauqPtr <- Call.ioarray tauq
taupPtr <- Call.ioarray taup
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.gebrd mPtr nPtr aPtr ldaPtr dPtr ePtr tauqPtr taupPtr workPtr lworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray d
<*> Call.freezeArray e
<*> Call.freezeArray tauq
<*> Call.freezeArray taup
<*> fmap fromIntegral (peek infoPtr)
gecon ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
IO (Float, Int)
gecon norm a anorm = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 (2*n)
evalContT $ do
normPtr <- Call.char norm
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gecon normPtr nPtr aPtr ldaPtr anormPtr rcondPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
geequ ::
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Float, Float, Float, Int)
geequ m a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
r <- Call.newArray1 m
c <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
rPtr <- Call.ioarray r
cPtr <- Call.ioarray c
rowcndPtr <- Call.alloca
colcndPtr <- Call.alloca
amaxPtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.geequ mPtr nPtr aPtr ldaPtr rPtr cPtr rowcndPtr colcndPtr amaxPtr infoPtr
liftIO $ pure (,,,,,)
<*> Call.freezeArray r
<*> Call.freezeArray c
<*> peek rowcndPtr
<*> peek colcndPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
geequb ::
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Float, Float, Float, Int)
geequb m a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
r <- Call.newArray1 m
c <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
rPtr <- Call.ioarray r
cPtr <- Call.ioarray c
rowcndPtr <- Call.alloca
colcndPtr <- Call.alloca
amaxPtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.geequb mPtr nPtr aPtr ldaPtr rPtr cPtr rowcndPtr colcndPtr amaxPtr infoPtr
liftIO $ pure (,,,,,)
<*> Call.freezeArray r
<*> Call.freezeArray c
<*> peek rowcndPtr
<*> peek colcndPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
gees ::
Char ->
Char ->
FunPtr (Ptr (Complex Float) -> IO Bool) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Int, Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int)
gees jobvs sort select a ldvs lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
w <- Call.newArray1 n
vs <- Call.newArray2 n ldvs
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 n
bwork <- Call.newArray1 n
evalContT $ do
jobvsPtr <- Call.char jobvs
sortPtr <- Call.char sort
selectPtr <- pure select
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
sdimPtr <- Call.alloca
wPtr <- Call.ioarray w
vsPtr <- Call.ioarray vs
ldvsPtr <- Call.cint ldvs
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
bworkPtr <- Call.ioarray bwork
infoPtr <- Call.alloca
liftIO $ FFI.gees jobvsPtr sortPtr selectPtr nPtr aPtr ldaPtr sdimPtr wPtr vsPtr ldvsPtr workPtr lworkPtr rworkPtr bworkPtr infoPtr
liftIO $ pure (,,,)
<*> fmap fromIntegral (peek sdimPtr)
<*> Call.freezeArray w
<*> Call.freezeArray vs
<*> fmap fromIntegral (peek infoPtr)
geesx ::
Char ->
Char ->
FunPtr (Ptr (Complex Float) -> IO Bool) ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Int, Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Float, Float, Int)
geesx jobvs sort select sense a ldvs lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
w <- Call.newArray1 n
vs <- Call.newArray2 n ldvs
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 n
bwork <- Call.newArray1 n
evalContT $ do
jobvsPtr <- Call.char jobvs
sortPtr <- Call.char sort
selectPtr <- pure select
sensePtr <- Call.char sense
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
sdimPtr <- Call.alloca
wPtr <- Call.ioarray w
vsPtr <- Call.ioarray vs
ldvsPtr <- Call.cint ldvs
rcondePtr <- Call.alloca
rcondvPtr <- Call.alloca
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
bworkPtr <- Call.ioarray bwork
infoPtr <- Call.alloca
liftIO $ FFI.geesx jobvsPtr sortPtr selectPtr sensePtr nPtr aPtr ldaPtr sdimPtr wPtr vsPtr ldvsPtr rcondePtr rcondvPtr workPtr lworkPtr rworkPtr bworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> fmap fromIntegral (peek sdimPtr)
<*> Call.freezeArray w
<*> Call.freezeArray vs
<*> peek rcondePtr
<*> peek rcondvPtr
<*> fmap fromIntegral (peek infoPtr)
geev ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
IO (Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int)
geev jobvl jobvr a ldvl ldvr lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
w <- Call.newArray1 n
vl <- Call.newArray2 n ldvl
vr <- Call.newArray2 n ldvr
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (2*n)
evalContT $ do
jobvlPtr <- Call.char jobvl
jobvrPtr <- Call.char jobvr
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
wPtr <- Call.ioarray w
vlPtr <- Call.ioarray vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.ioarray vr
ldvrPtr <- Call.cint ldvr
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.geev jobvlPtr jobvrPtr nPtr aPtr ldaPtr wPtr vlPtr ldvlPtr vrPtr ldvrPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray w
<*> Call.freezeArray vl
<*> Call.freezeArray vr
<*> fmap fromIntegral (peek infoPtr)
geevx ::
Char ->
Char ->
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
IO (Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int, Int, Array ZeroInt Float, Float, Array ZeroInt Float, Array ZeroInt Float, Int)
geevx balanc jobvl jobvr sense a ldvl ldvr lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
w <- Call.newArray1 n
vl <- Call.newArray2 n ldvl
vr <- Call.newArray2 n ldvr
scale <- Call.newArray1 n
rconde <- Call.newArray1 n
rcondv <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (2*n)
evalContT $ do
balancPtr <- Call.char balanc
jobvlPtr <- Call.char jobvl
jobvrPtr <- Call.char jobvr
sensePtr <- Call.char sense
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
wPtr <- Call.ioarray w
vlPtr <- Call.ioarray vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.ioarray vr
ldvrPtr <- Call.cint ldvr
iloPtr <- Call.alloca
ihiPtr <- Call.alloca
scalePtr <- Call.ioarray scale
abnrmPtr <- Call.alloca
rcondePtr <- Call.ioarray rconde
rcondvPtr <- Call.ioarray rcondv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.geevx balancPtr jobvlPtr jobvrPtr sensePtr nPtr aPtr ldaPtr wPtr vlPtr ldvlPtr vrPtr ldvrPtr iloPtr ihiPtr scalePtr abnrmPtr rcondePtr rcondvPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,,,,,,,,,)
<*> Call.freezeArray w
<*> Call.freezeArray vl
<*> Call.freezeArray vr
<*> fmap fromIntegral (peek iloPtr)
<*> fmap fromIntegral (peek ihiPtr)
<*> Call.freezeArray scale
<*> peek abnrmPtr
<*> Call.freezeArray rconde
<*> Call.freezeArray rcondv
<*> fmap fromIntegral (peek infoPtr)
gehd2 ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
gehd2 ilo ihi a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (n-1)
work <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.gehd2 nPtr iloPtr ihiPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
gehrd ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
gehrd ilo ihi a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (n-1)
work <- Call.newArray1 lwork
evalContT $ do
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.gehrd nPtr iloPtr ihiPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
gelq2 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
gelq2 m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 m
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.gelq2 mPtr nPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
gelqf ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
gelqf m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.gelqf mPtr nPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
gels ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
gels trans m a b lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.gels transPtr mPtr nPtr nrhsPtr aPtr ldaPtr bPtr ldbPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
gelsd ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Int, Int)
gelsd m a b rcond lwork lrwork liwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
s <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,lrwork])
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
sPtr <- Call.ioarray s
rcondPtr <- Call.float rcond
rankPtr <- Call.alloca
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.gelsd mPtr nPtr nrhsPtr aPtr ldaPtr bPtr ldbPtr sPtr rcondPtr rankPtr workPtr lworkPtr rworkPtr iworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray s
<*> fmap fromIntegral (peek rankPtr)
<*> fmap fromIntegral (peek infoPtr)
gelss ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Int ->
IO (Array ZeroInt Float, Int, Int)
gelss m a b rcond lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
s <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (5*minimum[m,n])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
sPtr <- Call.ioarray s
rcondPtr <- Call.float rcond
rankPtr <- Call.alloca
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gelss mPtr nPtr nrhsPtr aPtr ldaPtr bPtr ldbPtr sPtr rcondPtr rankPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray s
<*> fmap fromIntegral (peek rankPtr)
<*> fmap fromIntegral (peek infoPtr)
gelsy ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
Float ->
Int ->
IO (Int, Int)
gelsy m a b jpvt rcond lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let jpvtDim0 = Call.sizes1 $ MutArray.shape jpvt
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "gelsy: n == jpvtDim0" (n == jpvtDim0)
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (2*n)
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
jpvtPtr <- Call.ioarray jpvt
rcondPtr <- Call.float rcond
rankPtr <- Call.alloca
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gelsy mPtr nPtr nrhsPtr aPtr ldaPtr bPtr ldbPtr jpvtPtr rcondPtr rankPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> fmap fromIntegral (peek rankPtr)
<*> fmap fromIntegral (peek infoPtr)
geql2 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
geql2 m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.geql2 mPtr nPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
geqlf ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
geqlf m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.geqlf mPtr nPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
geqp3 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
geqp3 m a jpvt lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let jpvtDim0 = Call.sizes1 $ MutArray.shape jpvt
let n = aDim0
let lda = aDim1
Call.assert "geqp3: n == jpvtDim0" (n == jpvtDim0)
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (2*n)
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
jpvtPtr <- Call.ioarray jpvt
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.geqp3 mPtr nPtr aPtr ldaPtr jpvtPtr tauPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
geqr2 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
geqr2 m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.geqr2 mPtr nPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
geqr2p ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
geqr2p m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.geqr2p mPtr nPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
geqrf ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
geqrf m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.geqrf mPtr nPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
geqrfp ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
geqrfp m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.geqrfp mPtr nPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
gerfs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
gerfs trans a af ipiv b x = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (afDim0,afDim1) = Call.sizes2 $ Array.shape af
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = aDim0
let lda = aDim1
let ldaf = afDim1
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "gerfs: n == afDim0" (n == afDim0)
Call.assert "gerfs: n == ipivDim0" (n == ipivDim0)
Call.assert "gerfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
transPtr <- Call.char trans
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
afPtr <- Call.array af
ldafPtr <- Call.cint ldaf
ipivPtr <- Call.array ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gerfs transPtr nPtr nrhsPtr aPtr ldaPtr afPtr ldafPtr ipivPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
gerq2 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
gerq2 m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 m
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.gerq2 mPtr nPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
gerqf ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
gerqf m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.gerqf mPtr nPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
gesc2 ::
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
Array ZeroInt CInt ->
IO (Float)
gesc2 a rhs ipiv jpiv = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let rhsDim0 = Call.sizes1 $ MutArray.shape rhs
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let jpivDim0 = Call.sizes1 $ Array.shape jpiv
let n = aDim0
let lda = aDim1
let _rhsSize = rhsDim0
Call.assert "gesc2: n == ipivDim0" (n == ipivDim0)
Call.assert "gesc2: n == jpivDim0" (n == jpivDim0)
evalContT $ do
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
rhsPtr <- Call.ioarray rhs
ipivPtr <- Call.array ipiv
jpivPtr <- Call.array jpiv
scalePtr <- Call.alloca
liftIO $ FFI.gesc2 nPtr aPtr ldaPtr rhsPtr ipivPtr jpivPtr scalePtr
liftIO $ peek scalePtr
gesdd ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int)
gesdd jobz m a ucol ldu ldvt lwork lrwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
s <- Call.newArray1 (minimum[m,n])
u <- Call.newArray2 ucol ldu
vt <- Call.newArray2 n ldvt
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,lrwork])
iwork <- Call.newArray1 (8*minimum[m,n])
evalContT $ do
jobzPtr <- Call.char jobz
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
sPtr <- Call.ioarray s
uPtr <- Call.ioarray u
lduPtr <- Call.cint ldu
vtPtr <- Call.ioarray vt
ldvtPtr <- Call.cint ldvt
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.gesdd jobzPtr mPtr nPtr aPtr ldaPtr sPtr uPtr lduPtr vtPtr ldvtPtr workPtr lworkPtr rworkPtr iworkPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> Call.freezeArray u
<*> Call.freezeArray vt
<*> fmap fromIntegral (peek infoPtr)
gesv ::
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
gesv a b = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
ipiv <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.gesv nPtr nrhsPtr aPtr ldaPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
gesvd ::
Char ->
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int)
gesvd jobu jobvt m a ucol ldu ldvt lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
s <- Call.newArray1 (minimum[m,n])
u <- Call.newArray2 ucol ldu
vt <- Call.newArray2 n ldvt
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (5*minimum[m,n])
evalContT $ do
jobuPtr <- Call.char jobu
jobvtPtr <- Call.char jobvt
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
sPtr <- Call.ioarray s
uPtr <- Call.ioarray u
lduPtr <- Call.cint ldu
vtPtr <- Call.ioarray vt
ldvtPtr <- Call.cint ldvt
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gesvd jobuPtr jobvtPtr mPtr nPtr aPtr ldaPtr sPtr uPtr lduPtr vtPtr ldvtPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> Call.freezeArray u
<*> Call.freezeArray vt
<*> fmap fromIntegral (peek infoPtr)
gesvx ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
Char ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Char, Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
gesvx fact trans a af ipiv equed r c b ldx = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (afDim0,afDim1) = Call.sizes2 $ MutArray.shape af
let ipivDim0 = Call.sizes1 $ MutArray.shape ipiv
let rDim0 = Call.sizes1 $ MutArray.shape r
let cDim0 = Call.sizes1 $ MutArray.shape c
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldaf = afDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "gesvx: n == afDim0" (n == afDim0)
Call.assert "gesvx: n == ipivDim0" (n == ipivDim0)
Call.assert "gesvx: n == rDim0" (n == rDim0)
Call.assert "gesvx: n == cDim0" (n == cDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 (2*n)
evalContT $ do
factPtr <- Call.char fact
transPtr <- Call.char trans
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
afPtr <- Call.ioarray af
ldafPtr <- Call.cint ldaf
ipivPtr <- Call.ioarray ipiv
equedPtr <- Call.char equed
rPtr <- Call.ioarray r
cPtr <- Call.ioarray c
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gesvx factPtr transPtr nPtr nrhsPtr aPtr ldaPtr afPtr ldafPtr ipivPtr equedPtr rPtr cPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> fmap castCCharToChar (peek equedPtr)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
getc2 ::
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Array ZeroInt CInt, Int)
getc2 a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 n
jpiv <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
jpivPtr <- Call.ioarray jpiv
infoPtr <- Call.alloca
liftIO $ FFI.getc2 nPtr aPtr ldaPtr ipivPtr jpivPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ipiv
<*> Call.freezeArray jpiv
<*> fmap fromIntegral (peek infoPtr)
getf2 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
getf2 m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 (minimum[m,n])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.getf2 mPtr nPtr aPtr ldaPtr ipivPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
getrf ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
getrf m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 (minimum[m,n])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.getrf mPtr nPtr aPtr ldaPtr ipivPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
getri ::
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Int ->
IO (Int)
getri a ipiv lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "getri: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.getri nPtr aPtr ldaPtr ipivPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
getrs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
getrs trans a ipiv b = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "getrs: n == ipivDim0" (n == ipivDim0)
evalContT $ do
transPtr <- Call.char trans
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.getrs transPtr nPtr nrhsPtr aPtr ldaPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ggbak ::
Char ->
Char ->
Int ->
Int ->
Array ZeroInt Float ->
Array ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
ggbak job side ilo ihi lscale rscale v = do
let lscaleDim0 = Call.sizes1 $ Array.shape lscale
let rscaleDim0 = Call.sizes1 $ Array.shape rscale
let (vDim0,vDim1) = Call.sizes2 $ MutArray.shape v
let n = lscaleDim0
let m = vDim0
let ldv = vDim1
Call.assert "ggbak: n == rscaleDim0" (n == rscaleDim0)
evalContT $ do
jobPtr <- Call.char job
sidePtr <- Call.char side
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
lscalePtr <- Call.array lscale
rscalePtr <- Call.array rscale
mPtr <- Call.cint m
vPtr <- Call.ioarray v
ldvPtr <- Call.cint ldv
infoPtr <- Call.alloca
liftIO $ FFI.ggbak jobPtr sidePtr nPtr iloPtr ihiPtr lscalePtr rscalePtr mPtr vPtr ldvPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ggbal ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int, Int, Array ZeroInt Float, Array ZeroInt Float, Int)
ggbal job a b lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "ggbal: n == bDim0" (n == bDim0)
lscale <- Call.newArray1 n
rscale <- Call.newArray1 n
work <- Call.newArray1 lwork
evalContT $ do
jobPtr <- Call.char job
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
iloPtr <- Call.alloca
ihiPtr <- Call.alloca
lscalePtr <- Call.ioarray lscale
rscalePtr <- Call.ioarray rscale
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.ggbal jobPtr nPtr aPtr ldaPtr bPtr ldbPtr iloPtr ihiPtr lscalePtr rscalePtr workPtr infoPtr
liftIO $ pure (,,,,)
<*> fmap fromIntegral (peek iloPtr)
<*> fmap fromIntegral (peek ihiPtr)
<*> Call.freezeArray lscale
<*> Call.freezeArray rscale
<*> fmap fromIntegral (peek infoPtr)
gges ::
Char ->
Char ->
Char ->
FunPtr (Ptr (Complex Float) -> Ptr (Complex Float) -> IO Bool) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
IO (Int, Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int)
gges jobvsl jobvsr sort selctg a b ldvsl ldvsr lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "gges: n == bDim0" (n == bDim0)
alpha <- Call.newArray1 n
beta <- Call.newArray1 n
vsl <- Call.newArray2 n ldvsl
vsr <- Call.newArray2 n ldvsr
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (8*n)
bwork <- Call.newArray1 n
evalContT $ do
jobvslPtr <- Call.char jobvsl
jobvsrPtr <- Call.char jobvsr
sortPtr <- Call.char sort
selctgPtr <- pure selctg
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
sdimPtr <- Call.alloca
alphaPtr <- Call.ioarray alpha
betaPtr <- Call.ioarray beta
vslPtr <- Call.ioarray vsl
ldvslPtr <- Call.cint ldvsl
vsrPtr <- Call.ioarray vsr
ldvsrPtr <- Call.cint ldvsr
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
bworkPtr <- Call.ioarray bwork
infoPtr <- Call.alloca
liftIO $ FFI.gges jobvslPtr jobvsrPtr sortPtr selctgPtr nPtr aPtr ldaPtr bPtr ldbPtr sdimPtr alphaPtr betaPtr vslPtr ldvslPtr vsrPtr ldvsrPtr workPtr lworkPtr rworkPtr bworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> fmap fromIntegral (peek sdimPtr)
<*> Call.freezeArray alpha
<*> Call.freezeArray beta
<*> Call.freezeArray vsl
<*> Call.freezeArray vsr
<*> fmap fromIntegral (peek infoPtr)
ggesx ::
Char ->
Char ->
Char ->
FunPtr (Ptr (Complex Float) -> Ptr (Complex Float) -> IO Bool) ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
IO (Int, Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt Float, Array ZeroInt Float, Int)
ggesx jobvsl jobvsr sort selctg sense a b ldvsl ldvsr lwork liwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "ggesx: n == bDim0" (n == bDim0)
alpha <- Call.newArray1 n
beta <- Call.newArray1 n
vsl <- Call.newArray2 n ldvsl
vsr <- Call.newArray2 n ldvsr
rconde <- Call.newArray1 2
rcondv <- Call.newArray1 2
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (8*n)
iwork <- Call.newArray1 (maximum[1,liwork])
bwork <- Call.newArray1 n
evalContT $ do
jobvslPtr <- Call.char jobvsl
jobvsrPtr <- Call.char jobvsr
sortPtr <- Call.char sort
selctgPtr <- pure selctg
sensePtr <- Call.char sense
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
sdimPtr <- Call.alloca
alphaPtr <- Call.ioarray alpha
betaPtr <- Call.ioarray beta
vslPtr <- Call.ioarray vsl
ldvslPtr <- Call.cint ldvsl
vsrPtr <- Call.ioarray vsr
ldvsrPtr <- Call.cint ldvsr
rcondePtr <- Call.ioarray rconde
rcondvPtr <- Call.ioarray rcondv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
bworkPtr <- Call.ioarray bwork
infoPtr <- Call.alloca
liftIO $ FFI.ggesx jobvslPtr jobvsrPtr sortPtr selctgPtr sensePtr nPtr aPtr ldaPtr bPtr ldbPtr sdimPtr alphaPtr betaPtr vslPtr ldvslPtr vsrPtr ldvsrPtr rcondePtr rcondvPtr workPtr lworkPtr rworkPtr iworkPtr liworkPtr bworkPtr infoPtr
liftIO $ pure (,,,,,,,)
<*> fmap fromIntegral (peek sdimPtr)
<*> Call.freezeArray alpha
<*> Call.freezeArray beta
<*> Call.freezeArray vsl
<*> Call.freezeArray vsr
<*> Call.freezeArray rconde
<*> Call.freezeArray rcondv
<*> fmap fromIntegral (peek infoPtr)
ggev ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
IO (Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int)
ggev jobvl jobvr a b ldvl ldvr lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "ggev: n == bDim0" (n == bDim0)
alpha <- Call.newArray1 n
beta <- Call.newArray1 n
vl <- Call.newArray2 n ldvl
vr <- Call.newArray2 n ldvr
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (8*n)
evalContT $ do
jobvlPtr <- Call.char jobvl
jobvrPtr <- Call.char jobvr
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
alphaPtr <- Call.ioarray alpha
betaPtr <- Call.ioarray beta
vlPtr <- Call.ioarray vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.ioarray vr
ldvrPtr <- Call.cint ldvr
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.ggev jobvlPtr jobvrPtr nPtr aPtr ldaPtr bPtr ldbPtr alphaPtr betaPtr vlPtr ldvlPtr vrPtr ldvrPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray alpha
<*> Call.freezeArray beta
<*> Call.freezeArray vl
<*> Call.freezeArray vr
<*> fmap fromIntegral (peek infoPtr)
ggevx ::
Char ->
Char ->
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int, Int, Array ZeroInt Float, Array ZeroInt Float, Float, Float, Array ZeroInt Float, Array ZeroInt Float, Int)
ggevx balanc jobvl jobvr sense a b ldvl ldvr lwork lrwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "ggevx: n == bDim0" (n == bDim0)
alpha <- Call.newArray1 n
beta <- Call.newArray1 n
vl <- Call.newArray2 n ldvl
vr <- Call.newArray2 n ldvr
lscale <- Call.newArray1 n
rscale <- Call.newArray1 n
rconde <- Call.newArray1 n
rcondv <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 lrwork
iwork <- Call.newArray1 (n+2)
bwork <- Call.newArray1 n
evalContT $ do
balancPtr <- Call.char balanc
jobvlPtr <- Call.char jobvl
jobvrPtr <- Call.char jobvr
sensePtr <- Call.char sense
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
alphaPtr <- Call.ioarray alpha
betaPtr <- Call.ioarray beta
vlPtr <- Call.ioarray vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.ioarray vr
ldvrPtr <- Call.cint ldvr
iloPtr <- Call.alloca
ihiPtr <- Call.alloca
lscalePtr <- Call.ioarray lscale
rscalePtr <- Call.ioarray rscale
abnrmPtr <- Call.alloca
bbnrmPtr <- Call.alloca
rcondePtr <- Call.ioarray rconde
rcondvPtr <- Call.ioarray rcondv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
bworkPtr <- Call.ioarray bwork
infoPtr <- Call.alloca
liftIO $ FFI.ggevx balancPtr jobvlPtr jobvrPtr sensePtr nPtr aPtr ldaPtr bPtr ldbPtr alphaPtr betaPtr vlPtr ldvlPtr vrPtr ldvrPtr iloPtr ihiPtr lscalePtr rscalePtr abnrmPtr bbnrmPtr rcondePtr rcondvPtr workPtr lworkPtr rworkPtr iworkPtr bworkPtr infoPtr
liftIO $ pure (,,,,,,,,,,,,)
<*> Call.freezeArray alpha
<*> Call.freezeArray beta
<*> Call.freezeArray vl
<*> Call.freezeArray vr
<*> fmap fromIntegral (peek iloPtr)
<*> fmap fromIntegral (peek ihiPtr)
<*> Call.freezeArray lscale
<*> Call.freezeArray rscale
<*> peek abnrmPtr
<*> peek bbnrmPtr
<*> Call.freezeArray rconde
<*> Call.freezeArray rcondv
<*> fmap fromIntegral (peek infoPtr)
ggglm ::
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Int)
ggglm a b d lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let dDim0 = Call.sizes1 $ MutArray.shape d
let m = aDim0
let lda = aDim1
let p = bDim0
let ldb = bDim1
let n = dDim0
x <- Call.newArray1 m
y <- Call.newArray1 p
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
nPtr <- Call.cint n
mPtr <- Call.cint m
pPtr <- Call.cint p
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
dPtr <- Call.ioarray d
xPtr <- Call.ioarray x
yPtr <- Call.ioarray y
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.ggglm nPtr mPtr pPtr aPtr ldaPtr bPtr ldbPtr dPtr xPtr yPtr workPtr lworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray x
<*> Call.freezeArray y
<*> fmap fromIntegral (peek infoPtr)
gghrd ::
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
gghrd compq compz ilo ihi a b q z = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = aDim0
let lda = aDim1
let ldb = bDim1
let ldq = qDim1
let ldz = zDim1
Call.assert "gghrd: n == bDim0" (n == bDim0)
Call.assert "gghrd: n == qDim0" (n == qDim0)
Call.assert "gghrd: n == zDim0" (n == zDim0)
evalContT $ do
compqPtr <- Call.char compq
compzPtr <- Call.char compz
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
infoPtr <- Call.alloca
liftIO $ FFI.gghrd compqPtr compzPtr nPtr iloPtr ihiPtr aPtr ldaPtr bPtr ldbPtr qPtr ldqPtr zPtr ldzPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
gglse ::
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
gglse a b c d lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let cDim0 = Call.sizes1 $ MutArray.shape c
let dDim0 = Call.sizes1 $ MutArray.shape d
let n = aDim0
let lda = aDim1
let ldb = bDim1
let m = cDim0
let p = dDim0
Call.assert "gglse: n == bDim0" (n == bDim0)
x <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
pPtr <- Call.cint p
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
cPtr <- Call.ioarray c
dPtr <- Call.ioarray d
xPtr <- Call.ioarray x
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.gglse mPtr nPtr pPtr aPtr ldaPtr bPtr ldbPtr cPtr dPtr xPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray x
<*> fmap fromIntegral (peek infoPtr)
ggqrf ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Int)
ggqrf n a b lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let m = aDim0
let lda = aDim1
let p = bDim0
let ldb = bDim1
taua <- Call.newArray1 (minimum[n,m])
taub <- Call.newArray1 (minimum[n,p])
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
nPtr <- Call.cint n
mPtr <- Call.cint m
pPtr <- Call.cint p
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauaPtr <- Call.ioarray taua
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
taubPtr <- Call.ioarray taub
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.ggqrf nPtr mPtr pPtr aPtr ldaPtr tauaPtr bPtr ldbPtr taubPtr workPtr lworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray taua
<*> Call.freezeArray taub
<*> fmap fromIntegral (peek infoPtr)
ggrqf ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Int)
ggrqf m p a b lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "ggrqf: n == bDim0" (n == bDim0)
taua <- Call.newArray1 (minimum[m,n])
taub <- Call.newArray1 (minimum[p,n])
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
pPtr <- Call.cint p
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauaPtr <- Call.ioarray taua
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
taubPtr <- Call.ioarray taub
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.ggrqf mPtr pPtr nPtr aPtr ldaPtr tauaPtr bPtr ldbPtr taubPtr workPtr lworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray taua
<*> Call.freezeArray taub
<*> fmap fromIntegral (peek infoPtr)
gtcon ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
Float ->
IO (Float, Int)
gtcon norm dl d du du2 ipiv anorm = do
let dlDim0 = Call.sizes1 $ Array.shape dl
let dDim0 = Call.sizes1 $ Array.shape d
let duDim0 = Call.sizes1 $ Array.shape du
let du2Dim0 = Call.sizes1 $ Array.shape du2
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = dDim0
Call.assert "gtcon: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "gtcon: n-1 == duDim0" (n-1 == duDim0)
Call.assert "gtcon: n-2 == du2Dim0" (n-2 == du2Dim0)
Call.assert "gtcon: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 (2*n)
evalContT $ do
normPtr <- Call.char norm
nPtr <- Call.cint n
dlPtr <- Call.array dl
dPtr <- Call.array d
duPtr <- Call.array du
du2Ptr <- Call.array du2
ipivPtr <- Call.array ipiv
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.gtcon normPtr nPtr dlPtr dPtr duPtr du2Ptr ipivPtr anormPtr rcondPtr workPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
gtrfs ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
gtrfs trans dl d du dlf df duf du2 ipiv b x = do
let dlDim0 = Call.sizes1 $ Array.shape dl
let dDim0 = Call.sizes1 $ Array.shape d
let duDim0 = Call.sizes1 $ Array.shape du
let dlfDim0 = Call.sizes1 $ Array.shape dlf
let dfDim0 = Call.sizes1 $ Array.shape df
let dufDim0 = Call.sizes1 $ Array.shape duf
let du2Dim0 = Call.sizes1 $ Array.shape du2
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "gtrfs: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "gtrfs: n-1 == duDim0" (n-1 == duDim0)
Call.assert "gtrfs: n-1 == dlfDim0" (n-1 == dlfDim0)
Call.assert "gtrfs: n == dfDim0" (n == dfDim0)
Call.assert "gtrfs: n-1 == dufDim0" (n-1 == dufDim0)
Call.assert "gtrfs: n-2 == du2Dim0" (n-2 == du2Dim0)
Call.assert "gtrfs: n == ipivDim0" (n == ipivDim0)
Call.assert "gtrfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
transPtr <- Call.char trans
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dlPtr <- Call.array dl
dPtr <- Call.array d
duPtr <- Call.array du
dlfPtr <- Call.array dlf
dfPtr <- Call.array df
dufPtr <- Call.array duf
du2Ptr <- Call.array du2
ipivPtr <- Call.array ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gtrfs transPtr nPtr nrhsPtr dlPtr dPtr duPtr dlfPtr dfPtr dufPtr du2Ptr ipivPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
gtsv ::
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
gtsv dl d du b = do
let dlDim0 = Call.sizes1 $ MutArray.shape dl
let dDim0 = Call.sizes1 $ MutArray.shape d
let duDim0 = Call.sizes1 $ MutArray.shape du
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "gtsv: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "gtsv: n-1 == duDim0" (n-1 == duDim0)
evalContT $ do
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dlPtr <- Call.ioarray dl
dPtr <- Call.ioarray d
duPtr <- Call.ioarray du
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.gtsv nPtr nrhsPtr dlPtr dPtr duPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
gtsvx ::
Char ->
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
gtsvx fact trans dl d du dlf df duf du2 ipiv b ldx = do
let dlDim0 = Call.sizes1 $ Array.shape dl
let dDim0 = Call.sizes1 $ Array.shape d
let duDim0 = Call.sizes1 $ Array.shape du
let dlfDim0 = Call.sizes1 $ MutArray.shape dlf
let dfDim0 = Call.sizes1 $ MutArray.shape df
let dufDim0 = Call.sizes1 $ MutArray.shape duf
let du2Dim0 = Call.sizes1 $ MutArray.shape du2
let ipivDim0 = Call.sizes1 $ MutArray.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "gtsvx: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "gtsvx: n-1 == duDim0" (n-1 == duDim0)
Call.assert "gtsvx: n-1 == dlfDim0" (n-1 == dlfDim0)
Call.assert "gtsvx: n == dfDim0" (n == dfDim0)
Call.assert "gtsvx: n-1 == dufDim0" (n-1 == dufDim0)
Call.assert "gtsvx: n-2 == du2Dim0" (n-2 == du2Dim0)
Call.assert "gtsvx: n == ipivDim0" (n == ipivDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
transPtr <- Call.char trans
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dlPtr <- Call.array dl
dPtr <- Call.array d
duPtr <- Call.array du
dlfPtr <- Call.ioarray dlf
dfPtr <- Call.ioarray df
dufPtr <- Call.ioarray duf
du2Ptr <- Call.ioarray du2
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.gtsvx factPtr transPtr nPtr nrhsPtr dlPtr dPtr duPtr dlfPtr dfPtr dufPtr du2Ptr ipivPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
gttrf ::
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IO (Array ZeroInt (Complex Float), Array ZeroInt CInt, Int)
gttrf dl d du = do
let dlDim0 = Call.sizes1 $ MutArray.shape dl
let dDim0 = Call.sizes1 $ MutArray.shape d
let duDim0 = Call.sizes1 $ MutArray.shape du
let n = dDim0
Call.assert "gttrf: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "gttrf: n-1 == duDim0" (n-1 == duDim0)
du2 <- Call.newArray1 (n-2)
ipiv <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
dlPtr <- Call.ioarray dl
dPtr <- Call.ioarray d
duPtr <- Call.ioarray du
du2Ptr <- Call.ioarray du2
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.gttrf nPtr dlPtr dPtr duPtr du2Ptr ipivPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray du2
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
gttrs ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
gttrs trans dl d du du2 ipiv b = do
let dlDim0 = Call.sizes1 $ Array.shape dl
let dDim0 = Call.sizes1 $ Array.shape d
let duDim0 = Call.sizes1 $ Array.shape du
let du2Dim0 = Call.sizes1 $ Array.shape du2
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "gttrs: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "gttrs: n-1 == duDim0" (n-1 == duDim0)
Call.assert "gttrs: n-2 == du2Dim0" (n-2 == du2Dim0)
Call.assert "gttrs: n == ipivDim0" (n == ipivDim0)
evalContT $ do
transPtr <- Call.char trans
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dlPtr <- Call.array dl
dPtr <- Call.array d
duPtr <- Call.array du
du2Ptr <- Call.array du2
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.gttrs transPtr nPtr nrhsPtr dlPtr dPtr duPtr du2Ptr ipivPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
gtts2 ::
Int ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO ()
gtts2 itrans dl d du du2 ipiv b = do
let dlDim0 = Call.sizes1 $ Array.shape dl
let dDim0 = Call.sizes1 $ Array.shape d
let duDim0 = Call.sizes1 $ Array.shape du
let du2Dim0 = Call.sizes1 $ Array.shape du2
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "gtts2: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "gtts2: n-1 == duDim0" (n-1 == duDim0)
Call.assert "gtts2: n-2 == du2Dim0" (n-2 == du2Dim0)
Call.assert "gtts2: n == ipivDim0" (n == ipivDim0)
evalContT $ do
itransPtr <- Call.cint itrans
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dlPtr <- Call.array dl
dPtr <- Call.array d
duPtr <- Call.array du
du2Ptr <- Call.array du2
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
liftIO $ FFI.gtts2 itransPtr nPtr nrhsPtr dlPtr dPtr duPtr du2Ptr ipivPtr bPtr ldbPtr
hbev ::
Char ->
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Int)
hbev jobz uplo kd ab ldz = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 n
rwork <- Call.newArray1 (maximum[1,3*n-2])
evalContT $ do
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hbev jobzPtr uploPtr nPtr kdPtr abPtr ldabPtr wPtr zPtr ldzPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> fmap fromIntegral (peek infoPtr)
hbevd ::
Char ->
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Int)
hbevd jobz uplo kd ab ldz lwork rworkSize lrwork liwork = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 rworkSize
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.hbevd jobzPtr uploPtr nPtr kdPtr abPtr ldabPtr wPtr zPtr ldzPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> fmap fromIntegral (peek infoPtr)
hbevx ::
Char ->
Char ->
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Float ->
Float ->
Int ->
Int ->
Float ->
Int ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
hbevx jobz range uplo kd ab ldq vl vu il iu abstol m ldz = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
q <- Call.newArray2 n ldq
w <- Call.newArray1 n
z <- Call.newArray2 (maximum[1,m]) ldz
work <- Call.newArray1 n
rwork <- Call.newArray1 (7*n)
iwork <- Call.newArray1 (5*n)
ifail <- Call.newArray1 n
evalContT $ do
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
abstolPtr <- Call.float abstol
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
ifailPtr <- Call.ioarray ifail
infoPtr <- Call.alloca
liftIO $ FFI.hbevx jobzPtr rangePtr uploPtr nPtr kdPtr abPtr ldabPtr qPtr ldqPtr vlPtr vuPtr ilPtr iuPtr abstolPtr mPtr wPtr zPtr ldzPtr workPtr rworkPtr iworkPtr ifailPtr infoPtr
liftIO $ pure (,,,,,)
<*> Call.freezeArray q
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray ifail
<*> fmap fromIntegral (peek infoPtr)
hbgst ::
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int)
hbgst vect uplo ka kb ab bb ldx = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (bbDim0,bbDim1) = Call.sizes2 $ Array.shape bb
let n = abDim0
let ldab = abDim1
let ldbb = bbDim1
Call.assert "hbgst: n == bbDim0" (n == bbDim0)
x <- Call.newArray2 n ldx
work <- Call.newArray1 n
rwork <- Call.newArray1 n
evalContT $ do
vectPtr <- Call.char vect
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kaPtr <- Call.cint ka
kbPtr <- Call.cint kb
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
bbPtr <- Call.array bb
ldbbPtr <- Call.cint ldbb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hbgst vectPtr uploPtr nPtr kaPtr kbPtr abPtr ldabPtr bbPtr ldbbPtr xPtr ldxPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray x
<*> fmap fromIntegral (peek infoPtr)
hbgv ::
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Int)
hbgv jobz uplo ka kb ab bb ldz = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (bbDim0,bbDim1) = Call.sizes2 $ MutArray.shape bb
let n = abDim0
let ldab = abDim1
let ldbb = bbDim1
Call.assert "hbgv: n == bbDim0" (n == bbDim0)
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 n
rwork <- Call.newArray1 (3*n)
evalContT $ do
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kaPtr <- Call.cint ka
kbPtr <- Call.cint kb
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
bbPtr <- Call.ioarray bb
ldbbPtr <- Call.cint ldbb
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hbgv jobzPtr uploPtr nPtr kaPtr kbPtr abPtr ldabPtr bbPtr ldbbPtr wPtr zPtr ldzPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> fmap fromIntegral (peek infoPtr)
hbgvd ::
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Int)
hbgvd jobz uplo ka kb ab bb ldz lwork lrwork liwork = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (bbDim0,bbDim1) = Call.sizes2 $ MutArray.shape bb
let n = abDim0
let ldab = abDim1
let ldbb = bbDim1
Call.assert "hbgvd: n == bbDim0" (n == bbDim0)
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,lrwork])
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kaPtr <- Call.cint ka
kbPtr <- Call.cint kb
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
bbPtr <- Call.ioarray bb
ldbbPtr <- Call.cint ldbb
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.hbgvd jobzPtr uploPtr nPtr kaPtr kbPtr abPtr ldabPtr bbPtr ldbbPtr wPtr zPtr ldzPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> fmap fromIntegral (peek infoPtr)
hbgvx ::
Char ->
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Float ->
Float ->
Int ->
Int ->
Float ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
hbgvx jobz range uplo ka kb ab bb ldq vl vu il iu abstol ldz = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (bbDim0,bbDim1) = Call.sizes2 $ MutArray.shape bb
let n = abDim0
let ldab = abDim1
let ldbb = bbDim1
Call.assert "hbgvx: n == bbDim0" (n == bbDim0)
q <- Call.newArray2 n ldq
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 n
rwork <- Call.newArray1 (7*n)
iwork <- Call.newArray1 (5*n)
ifail <- Call.newArray1 n
evalContT $ do
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kaPtr <- Call.cint ka
kbPtr <- Call.cint kb
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
bbPtr <- Call.ioarray bb
ldbbPtr <- Call.cint ldbb
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
abstolPtr <- Call.float abstol
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
ifailPtr <- Call.ioarray ifail
infoPtr <- Call.alloca
liftIO $ FFI.hbgvx jobzPtr rangePtr uploPtr nPtr kaPtr kbPtr abPtr ldabPtr bbPtr ldbbPtr qPtr ldqPtr vlPtr vuPtr ilPtr iuPtr abstolPtr mPtr wPtr zPtr ldzPtr workPtr rworkPtr iworkPtr ifailPtr infoPtr
liftIO $ pure (,,,,,)
<*> Call.freezeArray q
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray ifail
<*> fmap fromIntegral (peek infoPtr)
hbtrd ::
Char ->
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
hbtrd vect uplo kd ab q = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let n = abDim0
let ldab = abDim1
let ldq = qDim1
Call.assert "hbtrd: n == qDim0" (n == qDim0)
d <- Call.newArray1 n
e <- Call.newArray1 (n-1)
work <- Call.newArray1 n
evalContT $ do
vectPtr <- Call.char vect
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.hbtrd vectPtr uploPtr nPtr kdPtr abPtr ldabPtr dPtr ePtr qPtr ldqPtr workPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray d
<*> Call.freezeArray e
<*> fmap fromIntegral (peek infoPtr)
hecon ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Float ->
IO (Float, Int)
hecon uplo a ipiv anorm = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "hecon: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.hecon uploPtr nPtr aPtr ldaPtr ipivPtr anormPtr rcondPtr workPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
heequb ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Float, Float, Int)
heequb uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
s <- Call.newArray1 n
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
sPtr <- Call.ioarray s
scondPtr <- Call.alloca
amaxPtr <- Call.alloca
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.heequb uploPtr nPtr aPtr ldaPtr sPtr scondPtr amaxPtr workPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> peek scondPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
heev ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Int)
heev jobz uplo a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
w <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,3*n-2])
evalContT $ do
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
wPtr <- Call.ioarray w
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.heev jobzPtr uploPtr nPtr aPtr ldaPtr wPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
heevd ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Int)
heevd jobz uplo a lwork rworkSize lrwork liwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
w <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 rworkSize
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
wPtr <- Call.ioarray w
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.heevd jobzPtr uploPtr nPtr aPtr ldaPtr wPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
heevr ::
Char ->
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Float ->
Int ->
Int ->
Float ->
Int ->
Int ->
Int ->
Int ->
Int ->
IO (Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
heevr jobz range uplo a vl vu il iu abstol m ldz lwork lrwork liwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
w <- Call.newArray1 n
z <- Call.newArray2 (maximum[1,m]) ldz
isuppz <- Call.newArray1 (2*maximum[1,m])
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,lrwork])
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
abstolPtr <- Call.float abstol
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
isuppzPtr <- Call.ioarray isuppz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.heevr jobzPtr rangePtr uploPtr nPtr aPtr ldaPtr vlPtr vuPtr ilPtr iuPtr abstolPtr mPtr wPtr zPtr ldzPtr isuppzPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,,,,)
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray isuppz
<*> fmap fromIntegral (peek infoPtr)
heevx ::
Char ->
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Float ->
Int ->
Int ->
Float ->
Int ->
Int ->
Int ->
IO (Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
heevx jobz range uplo a vl vu il iu abstol m ldz lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
w <- Call.newArray1 n
z <- Call.newArray2 (maximum[1,m]) ldz
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (7*n)
iwork <- Call.newArray1 (5*n)
ifail <- Call.newArray1 n
evalContT $ do
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
abstolPtr <- Call.float abstol
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
ifailPtr <- Call.ioarray ifail
infoPtr <- Call.alloca
liftIO $ FFI.heevx jobzPtr rangePtr uploPtr nPtr aPtr ldaPtr vlPtr vuPtr ilPtr iuPtr abstolPtr mPtr wPtr zPtr ldzPtr workPtr lworkPtr rworkPtr iworkPtr ifailPtr infoPtr
liftIO $ pure (,,,,)
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray ifail
<*> fmap fromIntegral (peek infoPtr)
hegs2 ::
Int ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
hegs2 itype uplo a b = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "hegs2: n == bDim0" (n == bDim0)
evalContT $ do
itypePtr <- Call.cint itype
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.hegs2 itypePtr uploPtr nPtr aPtr ldaPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hegst ::
Int ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
hegst itype uplo a b = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "hegst: n == bDim0" (n == bDim0)
evalContT $ do
itypePtr <- Call.cint itype
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.hegst itypePtr uploPtr nPtr aPtr ldaPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hegv ::
Int ->
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Int)
hegv itype jobz uplo a b lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "hegv: n == bDim0" (n == bDim0)
w <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,3*n-2])
evalContT $ do
itypePtr <- Call.cint itype
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
wPtr <- Call.ioarray w
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hegv itypePtr jobzPtr uploPtr nPtr aPtr ldaPtr bPtr ldbPtr wPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
hegvd ::
Int ->
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Int)
hegvd itype jobz uplo a b lwork lrwork liwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "hegvd: n == bDim0" (n == bDim0)
w <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,lrwork])
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
itypePtr <- Call.cint itype
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
wPtr <- Call.ioarray w
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.hegvd itypePtr jobzPtr uploPtr nPtr aPtr ldaPtr bPtr ldbPtr wPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
hegvx ::
Int ->
Char ->
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Float ->
Int ->
Int ->
Float ->
Int ->
Int ->
Int ->
IO (Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
hegvx itype jobz range uplo a b vl vu il iu abstol m ldz lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "hegvx: n == bDim0" (n == bDim0)
w <- Call.newArray1 n
z <- Call.newArray2 (maximum[1,m]) ldz
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (7*n)
iwork <- Call.newArray1 (5*n)
ifail <- Call.newArray1 n
evalContT $ do
itypePtr <- Call.cint itype
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
abstolPtr <- Call.float abstol
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
ifailPtr <- Call.ioarray ifail
infoPtr <- Call.alloca
liftIO $ FFI.hegvx itypePtr jobzPtr rangePtr uploPtr nPtr aPtr ldaPtr bPtr ldbPtr vlPtr vuPtr ilPtr iuPtr abstolPtr mPtr wPtr zPtr ldzPtr workPtr lworkPtr rworkPtr iworkPtr ifailPtr infoPtr
liftIO $ pure (,,,,)
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray ifail
<*> fmap fromIntegral (peek infoPtr)
herfs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
herfs uplo a af ipiv b x = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (afDim0,afDim1) = Call.sizes2 $ Array.shape af
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = aDim0
let lda = aDim1
let ldaf = afDim1
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "herfs: n == afDim0" (n == afDim0)
Call.assert "herfs: n == ipivDim0" (n == ipivDim0)
Call.assert "herfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
afPtr <- Call.array af
ldafPtr <- Call.cint ldaf
ipivPtr <- Call.array ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.herfs uploPtr nPtr nrhsPtr aPtr ldaPtr afPtr ldafPtr ipivPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
hesv ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt CInt, Int)
hesv uplo a b lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
ipiv <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.hesv uploPtr nPtr nrhsPtr aPtr ldaPtr ipivPtr bPtr ldbPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
hesvx ::
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
hesvx fact uplo a af ipiv b ldx lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (afDim0,afDim1) = Call.sizes2 $ MutArray.shape af
let ipivDim0 = Call.sizes1 $ MutArray.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let n = aDim0
let lda = aDim1
let ldaf = afDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "hesvx: n == afDim0" (n == afDim0)
Call.assert "hesvx: n == ipivDim0" (n == ipivDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
afPtr <- Call.ioarray af
ldafPtr <- Call.cint ldaf
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hesvx factPtr uploPtr nPtr nrhsPtr aPtr ldaPtr afPtr ldafPtr ipivPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
heswapr ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO ()
heswapr uplo a i1 i2 = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
i1Ptr <- Call.cint i1
i2Ptr <- Call.cint i2
liftIO $ FFI.heswapr uploPtr nPtr aPtr ldaPtr i1Ptr i2Ptr
hetd2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt (Complex Float), Int)
hetd2 uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
d <- Call.newArray1 n
e <- Call.newArray1 (n-1)
tau <- Call.newArray1 (n-1)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
tauPtr <- Call.ioarray tau
infoPtr <- Call.alloca
liftIO $ FFI.hetd2 uploPtr nPtr aPtr ldaPtr dPtr ePtr tauPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray d
<*> Call.freezeArray e
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
hetf2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
hetf2 uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.hetf2 uploPtr nPtr aPtr ldaPtr ipivPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
hetrd ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt (Complex Float), Int)
hetrd uplo a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
d <- Call.newArray1 n
e <- Call.newArray1 (n-1)
tau <- Call.newArray1 (n-1)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.hetrd uploPtr nPtr aPtr ldaPtr dPtr ePtr tauPtr workPtr lworkPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray d
<*> Call.freezeArray e
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
hetrf ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt CInt, Int)
hetrf uplo a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.hetrf uploPtr nPtr aPtr ldaPtr ipivPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
hetri ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IO (Int)
hetri uplo a ipiv = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "hetri: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.hetri uploPtr nPtr aPtr ldaPtr ipivPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hetri2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Int ->
Int ->
IO (Int)
hetri2 uplo a ipiv nb lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "hetri2: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 (n+nb+1)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.hetri2 uploPtr nPtr aPtr ldaPtr ipivPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hetri2x ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Int ->
IO (Int)
hetri2x uplo a ipiv nb = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "hetri2x: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray2 (nb+3) (n+nb+1)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
nbPtr <- Call.cint nb
infoPtr <- Call.alloca
liftIO $ FFI.hetri2x uploPtr nPtr aPtr ldaPtr ipivPtr workPtr nbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hetrs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
hetrs uplo a ipiv b = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "hetrs: n == ipivDim0" (n == ipivDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.hetrs uploPtr nPtr nrhsPtr aPtr ldaPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hetrs2 ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
hetrs2 uplo a ipiv b = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "hetrs2: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.hetrs2 uploPtr nPtr nrhsPtr aPtr ldaPtr ipivPtr bPtr ldbPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hfrk ::
Char ->
Char ->
Char ->
Int ->
Int ->
Float ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
IOArray ZeroInt (Complex Float) ->
IO ()
hfrk transr uplo trans n k alpha a beta c = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let cDim0 = Call.sizes1 $ MutArray.shape c
let _ka = aDim0
let lda = aDim1
Call.assert "hfrk: n*(n+1)`div`2 == cDim0" (n*(n+1)`div`2 == cDim0)
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
transPtr <- Call.char trans
nPtr <- Call.cint n
kPtr <- Call.cint k
alphaPtr <- Call.float alpha
aPtr <- Call.array a
ldaPtr <- Call.cint lda
betaPtr <- Call.float beta
cPtr <- Call.ioarray c
liftIO $ FFI.hfrk transrPtr uploPtr transPtr nPtr kPtr alphaPtr aPtr ldaPtr betaPtr cPtr
hgeqz ::
Char ->
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Int)
hgeqz job compq compz ilo ihi h t q z lwork = do
let (hDim0,hDim1) = Call.sizes2 $ MutArray.shape h
let (tDim0,tDim1) = Call.sizes2 $ MutArray.shape t
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = hDim0
let ldh = hDim1
let ldt = tDim1
let ldq = qDim1
let ldz = zDim1
Call.assert "hgeqz: n == tDim0" (n == tDim0)
Call.assert "hgeqz: n == qDim0" (n == qDim0)
Call.assert "hgeqz: n == zDim0" (n == zDim0)
alpha <- Call.newArray1 n
beta <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 n
evalContT $ do
jobPtr <- Call.char job
compqPtr <- Call.char compq
compzPtr <- Call.char compz
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
hPtr <- Call.ioarray h
ldhPtr <- Call.cint ldh
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
alphaPtr <- Call.ioarray alpha
betaPtr <- Call.ioarray beta
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hgeqz jobPtr compqPtr compzPtr nPtr iloPtr ihiPtr hPtr ldhPtr tPtr ldtPtr alphaPtr betaPtr qPtr ldqPtr zPtr ldzPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray alpha
<*> Call.freezeArray beta
<*> fmap fromIntegral (peek infoPtr)
hpcon ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
Float ->
IO (Float, Int)
hpcon uplo ap ipiv anorm = do
let apDim0 = Call.sizes1 $ Array.shape ap
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = ipivDim0
Call.assert "hpcon: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
ipivPtr <- Call.array ipiv
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.hpcon uploPtr nPtr apPtr ipivPtr anormPtr rcondPtr workPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
hpev ::
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Int)
hpev jobz uplo n ap ldz = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "hpev: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 (maximum[1,2*n-1])
rwork <- Call.newArray1 (maximum[1,3*n-2])
evalContT $ do
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hpev jobzPtr uploPtr nPtr apPtr wPtr zPtr ldzPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> fmap fromIntegral (peek infoPtr)
hpevd ::
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Int)
hpevd jobz uplo n ap ldz lwork lrwork liwork = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "hpevd: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,lrwork])
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.hpevd jobzPtr uploPtr nPtr apPtr wPtr zPtr ldzPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> fmap fromIntegral (peek infoPtr)
hpevx ::
Char ->
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
Float ->
Float ->
Int ->
Int ->
Float ->
Int ->
Int ->
IO (Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
hpevx jobz range uplo n ap vl vu il iu abstol m ldz = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "hpevx: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
w <- Call.newArray1 n
z <- Call.newArray2 (maximum[1,m]) ldz
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 (7*n)
iwork <- Call.newArray1 (5*n)
ifail <- Call.newArray1 n
evalContT $ do
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
abstolPtr <- Call.float abstol
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
ifailPtr <- Call.ioarray ifail
infoPtr <- Call.alloca
liftIO $ FFI.hpevx jobzPtr rangePtr uploPtr nPtr apPtr vlPtr vuPtr ilPtr iuPtr abstolPtr mPtr wPtr zPtr ldzPtr workPtr rworkPtr iworkPtr ifailPtr infoPtr
liftIO $ pure (,,,,)
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray ifail
<*> fmap fromIntegral (peek infoPtr)
hpgst ::
Int ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
IO (Int)
hpgst itype uplo n ap bp = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let bpDim0 = Call.sizes1 $ Array.shape bp
Call.assert "hpgst: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "hpgst: n*(n+1)`div`2 == bpDim0" (n*(n+1)`div`2 == bpDim0)
evalContT $ do
itypePtr <- Call.cint itype
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
bpPtr <- Call.array bp
infoPtr <- Call.alloca
liftIO $ FFI.hpgst itypePtr uploPtr nPtr apPtr bpPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hpgv ::
Int ->
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Int)
hpgv itype jobz uplo n ap bp ldz = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let bpDim0 = Call.sizes1 $ MutArray.shape bp
Call.assert "hpgv: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "hpgv: n*(n+1)`div`2 == bpDim0" (n*(n+1)`div`2 == bpDim0)
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 (maximum[1,2*n-1])
rwork <- Call.newArray1 (maximum[1,3*n-2])
evalContT $ do
itypePtr <- Call.cint itype
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
bpPtr <- Call.ioarray bp
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hpgv itypePtr jobzPtr uploPtr nPtr apPtr bpPtr wPtr zPtr ldzPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> fmap fromIntegral (peek infoPtr)
hpgvd ::
Int ->
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Int)
hpgvd itype jobz uplo n ap bp ldz lwork lrwork liwork = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let bpDim0 = Call.sizes1 $ MutArray.shape bp
Call.assert "hpgvd: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "hpgvd: n*(n+1)`div`2 == bpDim0" (n*(n+1)`div`2 == bpDim0)
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,lrwork])
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
itypePtr <- Call.cint itype
jobzPtr <- Call.char jobz
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
bpPtr <- Call.ioarray bp
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.hpgvd itypePtr jobzPtr uploPtr nPtr apPtr bpPtr wPtr zPtr ldzPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> fmap fromIntegral (peek infoPtr)
hpgvx ::
Int ->
Char ->
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Float ->
Float ->
Int ->
Int ->
Float ->
Int ->
IO (Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
hpgvx itype jobz range uplo n ap bp vl vu il iu abstol ldz = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let bpDim0 = Call.sizes1 $ MutArray.shape bp
Call.assert "hpgvx: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "hpgvx: n*(n+1)`div`2 == bpDim0" (n*(n+1)`div`2 == bpDim0)
w <- Call.newArray1 n
z <- Call.newArray2 n ldz
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 (7*n)
iwork <- Call.newArray1 (5*n)
ifail <- Call.newArray1 n
evalContT $ do
itypePtr <- Call.cint itype
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
bpPtr <- Call.ioarray bp
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
abstolPtr <- Call.float abstol
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
ifailPtr <- Call.ioarray ifail
infoPtr <- Call.alloca
liftIO $ FFI.hpgvx itypePtr jobzPtr rangePtr uploPtr nPtr apPtr bpPtr vlPtr vuPtr ilPtr iuPtr abstolPtr mPtr wPtr zPtr ldzPtr workPtr rworkPtr iworkPtr ifailPtr infoPtr
liftIO $ pure (,,,,)
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray ifail
<*> fmap fromIntegral (peek infoPtr)
hprfs ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
hprfs uplo ap afp ipiv b x = do
let apDim0 = Call.sizes1 $ Array.shape ap
let afpDim0 = Call.sizes1 $ Array.shape afp
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = ipivDim0
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "hprfs: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "hprfs: n*(n+1)`div`2 == afpDim0" (n*(n+1)`div`2 == afpDim0)
Call.assert "hprfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
afpPtr <- Call.array afp
ipivPtr <- Call.array ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hprfs uploPtr nPtr nrhsPtr apPtr afpPtr ipivPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
hpsv ::
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
hpsv uplo n ap b = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let nrhs = bDim0
let ldb = bDim1
Call.assert "hpsv: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
ipiv <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.ioarray ap
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.hpsv uploPtr nPtr nrhsPtr apPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
hpsvx ::
Char ->
Char ->
Array ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
hpsvx fact uplo ap afp ipiv b ldx = do
let apDim0 = Call.sizes1 $ Array.shape ap
let afpDim0 = Call.sizes1 $ MutArray.shape afp
let ipivDim0 = Call.sizes1 $ MutArray.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let n = ipivDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "hpsvx: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "hpsvx: n*(n+1)`div`2 == afpDim0" (n*(n+1)`div`2 == afpDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
afpPtr <- Call.ioarray afp
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.hpsvx factPtr uploPtr nPtr nrhsPtr apPtr afpPtr ipivPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
hptrd ::
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt (Complex Float), Int)
hptrd uplo n ap = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "hptrd: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
d <- Call.newArray1 n
e <- Call.newArray1 (n-1)
tau <- Call.newArray1 (n-1)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
tauPtr <- Call.ioarray tau
infoPtr <- Call.alloca
liftIO $ FFI.hptrd uploPtr nPtr apPtr dPtr ePtr tauPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray d
<*> Call.freezeArray e
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
hptrf ::
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Array ZeroInt CInt, Int)
hptrf uplo n ap = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "hptrf: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
ipiv <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.hptrf uploPtr nPtr apPtr ipivPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
hptri ::
Char ->
IOArray ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
IO (Int)
hptri uplo ap ipiv = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = ipivDim0
Call.assert "hptri: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.hptri uploPtr nPtr apPtr ipivPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hptrs ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
hptrs uplo ap ipiv b = do
let apDim0 = Call.sizes1 $ Array.shape ap
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = ipivDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "hptrs: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.hptrs uploPtr nPtr nrhsPtr apPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
hsein ::
Char ->
Char ->
Char ->
Array ZeroInt Bool ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int, Array ZeroInt CInt, Array ZeroInt CInt, Int)
hsein side eigsrc initv select h w vl vr = do
let selectDim0 = Call.sizes1 $ Array.shape select
let (hDim0,hDim1) = Call.sizes2 $ Array.shape h
let wDim0 = Call.sizes1 $ MutArray.shape w
let (vlDim0,vlDim1) = Call.sizes2 $ MutArray.shape vl
let (vrDim0,vrDim1) = Call.sizes2 $ MutArray.shape vr
let n = selectDim0
let ldh = hDim1
let mm = vlDim0
let ldvl = vlDim1
let ldvr = vrDim1
Call.assert "hsein: n == hDim0" (n == hDim0)
Call.assert "hsein: n == wDim0" (n == wDim0)
Call.assert "hsein: mm == vrDim0" (mm == vrDim0)
work <- Call.newArray1 (n*n)
rwork <- Call.newArray1 n
ifaill <- Call.newArray1 mm
ifailr <- Call.newArray1 mm
evalContT $ do
sidePtr <- Call.char side
eigsrcPtr <- Call.char eigsrc
initvPtr <- Call.char initv
selectPtr <- Call.array select
nPtr <- Call.cint n
hPtr <- Call.array h
ldhPtr <- Call.cint ldh
wPtr <- Call.ioarray w
vlPtr <- Call.ioarray vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.ioarray vr
ldvrPtr <- Call.cint ldvr
mmPtr <- Call.cint mm
mPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
ifaillPtr <- Call.ioarray ifaill
ifailrPtr <- Call.ioarray ifailr
infoPtr <- Call.alloca
liftIO $ FFI.hsein sidePtr eigsrcPtr initvPtr selectPtr nPtr hPtr ldhPtr wPtr vlPtr ldvlPtr vrPtr ldvrPtr mmPtr mPtr workPtr rworkPtr ifaillPtr ifailrPtr infoPtr
liftIO $ pure (,,,)
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray ifaill
<*> Call.freezeArray ifailr
<*> fmap fromIntegral (peek infoPtr)
hseqr ::
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
hseqr job compz ilo ihi h z lwork = do
let (hDim0,hDim1) = Call.sizes2 $ MutArray.shape h
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = hDim0
let ldh = hDim1
let ldz = zDim1
Call.assert "hseqr: n == zDim0" (n == zDim0)
w <- Call.newArray1 n
work <- Call.newArray1 lwork
evalContT $ do
jobPtr <- Call.char job
compzPtr <- Call.char compz
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
hPtr <- Call.ioarray h
ldhPtr <- Call.cint ldh
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.hseqr jobPtr compzPtr nPtr iloPtr ihiPtr hPtr ldhPtr wPtr zPtr ldzPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
ilalc ::
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO CInt
ilalc m a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
liftIO $ FFI.ilalc mPtr nPtr aPtr ldaPtr
ilalr ::
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO CInt
ilalr m a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
liftIO $ FFI.ilalr mPtr nPtr aPtr ldaPtr
imax1 ::
Array ZeroInt (Complex Float) ->
Int ->
IO CInt
imax1 cx incx = do
let cxDim0 = Call.sizes1 $ Array.shape cx
let n = cxDim0
evalContT $ do
nPtr <- Call.cint n
cxPtr <- Call.array cx
incxPtr <- Call.cint incx
liftIO $ FFI.imax1 nPtr cxPtr incxPtr
labrd ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float))
labrd m nb a ldx ldy = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
d <- Call.newArray1 nb
e <- Call.newArray1 nb
tauq <- Call.newArray1 nb
taup <- Call.newArray1 nb
x <- Call.newArray2 nb ldx
y <- Call.newArray2 nb ldy
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
nbPtr <- Call.cint nb
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
tauqPtr <- Call.ioarray tauq
taupPtr <- Call.ioarray taup
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
yPtr <- Call.ioarray y
ldyPtr <- Call.cint ldy
liftIO $ FFI.labrd mPtr nPtr nbPtr aPtr ldaPtr dPtr ePtr tauqPtr taupPtr xPtr ldxPtr yPtr ldyPtr
liftIO $ pure (,,,,,)
<*> Call.freezeArray d
<*> Call.freezeArray e
<*> Call.freezeArray tauq
<*> Call.freezeArray taup
<*> Call.freezeArray x
<*> Call.freezeArray y
lacgv ::
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO ()
lacgv n x incx = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let _xSize = xDim0
evalContT $ do
nPtr <- Call.cint n
xPtr <- Call.ioarray x
incxPtr <- Call.cint incx
liftIO $ FFI.lacgv nPtr xPtr incxPtr
lacn2 ::
IOArray ZeroInt (Complex Float) ->
Float ->
Int ->
IOArray ZeroInt CInt ->
IO (Array ZeroInt (Complex Float), Float, Int)
lacn2 x est kase isave = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let isaveDim0 = Call.sizes1 $ MutArray.shape isave
let n = xDim0
Call.assert "lacn2: 3 == isaveDim0" (3 == isaveDim0)
v <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
vPtr <- Call.ioarray v
xPtr <- Call.ioarray x
estPtr <- Call.float est
kasePtr <- Call.cint kase
isavePtr <- Call.ioarray isave
liftIO $ FFI.lacn2 nPtr vPtr xPtr estPtr kasePtr isavePtr
liftIO $ pure (,,)
<*> Call.freezeArray v
<*> peek estPtr
<*> fmap fromIntegral (peek kasePtr)
lacon ::
IOArray ZeroInt (Complex Float) ->
Float ->
Int ->
IO (Array ZeroInt (Complex Float), Float, Int)
lacon x est kase = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let n = xDim0
v <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
vPtr <- Call.ioarray v
xPtr <- Call.ioarray x
estPtr <- Call.float est
kasePtr <- Call.cint kase
liftIO $ FFI.lacon nPtr vPtr xPtr estPtr kasePtr
liftIO $ pure (,,)
<*> Call.freezeArray v
<*> peek estPtr
<*> fmap fromIntegral (peek kasePtr)
lacp2 ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) Float ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float))
lacp2 uplo m a ldb = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
b <- Call.newArray2 n ldb
evalContT $ do
uploPtr <- Call.char uplo
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
liftIO $ FFI.lacp2 uploPtr mPtr nPtr aPtr ldaPtr bPtr ldbPtr
liftIO $ Call.freezeArray b
lacpy ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float))
lacpy uplo m a ldb = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
b <- Call.newArray2 n ldb
evalContT $ do
uploPtr <- Call.char uplo
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
liftIO $ FFI.lacpy uploPtr mPtr nPtr aPtr ldaPtr bPtr ldbPtr
liftIO $ Call.freezeArray b
lacrm ::
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) Float ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float))
lacrm m a b ldc = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let n = aDim0
let lda = aDim1
let ldb = bDim1
Call.assert "lacrm: n == bDim0" (n == bDim0)
c <- Call.newArray2 n ldc
rwork <- Call.newArray1 (2*m*n)
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
rworkPtr <- Call.ioarray rwork
liftIO $ FFI.lacrm mPtr nPtr aPtr ldaPtr bPtr ldbPtr cPtr ldcPtr rworkPtr
liftIO $ Call.freezeArray c
lacrt ::
IOArray ZeroInt (Complex Float) ->
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
Complex Float ->
Complex Float ->
IO ()
lacrt cx incx cy incy c s = do
let cxDim0 = Call.sizes1 $ MutArray.shape cx
let cyDim0 = Call.sizes1 $ MutArray.shape cy
let n = cxDim0
Call.assert "lacrt: n == cyDim0" (n == cyDim0)
evalContT $ do
nPtr <- Call.cint n
cxPtr <- Call.ioarray cx
incxPtr <- Call.cint incx
cyPtr <- Call.ioarray cy
incyPtr <- Call.cint incy
cPtr <- Call.complexFloat c
sPtr <- Call.complexFloat s
liftIO $ FFI.lacrt nPtr cxPtr incxPtr cyPtr incyPtr cPtr sPtr
laed0 ::
Int ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int)
laed0 qsiz d e q ldqs rworkSize iworkSize = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let n = dDim0
let ldq = qDim1
Call.assert "laed0: n-1 == eDim0" (n-1 == eDim0)
Call.assert "laed0: n == qDim0" (n == qDim0)
qstore <- Call.newArray2 n ldqs
rwork <- Call.newArray1 rworkSize
iwork <- Call.newArray1 iworkSize
evalContT $ do
qsizPtr <- Call.cint qsiz
nPtr <- Call.cint n
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
qstorePtr <- Call.ioarray qstore
ldqsPtr <- Call.cint ldqs
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.laed0 qsizPtr nPtr dPtr ePtr qPtr ldqPtr qstorePtr ldqsPtr rworkPtr iworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray qstore
<*> fmap fromIntegral (peek infoPtr)
laed7 ::
Int ->
Int ->
Int ->
Int ->
Int ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
IOArray ZeroInt Float ->
IOArray ZeroInt CInt ->
Array ZeroInt CInt ->
Array ZeroInt CInt ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) CInt ->
Array (ZeroInt,ZeroInt) Float ->
Int ->
IO (Array ZeroInt CInt, Int)
laed7 cutpnt qsiz tlvls curlvl curpbm d q rho qstore qptr prmptr perm givptr givcol givnum rworkSize = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let qstoreDim0 = Call.sizes1 $ MutArray.shape qstore
let qptrDim0 = Call.sizes1 $ MutArray.shape qptr
let prmptrDim0 = Call.sizes1 $ Array.shape prmptr
let permDim0 = Call.sizes1 $ Array.shape perm
let givptrDim0 = Call.sizes1 $ Array.shape givptr
let (givcolDim0,givcolDim1) = Call.sizes2 $ Array.shape givcol
let (givnumDim0,givnumDim1) = Call.sizes2 $ Array.shape givnum
let n = dDim0
let ldq = qDim1
let nlgn = prmptrDim0
Call.assert "laed7: n == qDim0" (n == qDim0)
Call.assert "laed7: n^!2+1 == qstoreDim0" (n^!2+1 == qstoreDim0)
Call.assert "laed7: n+2 == qptrDim0" (n+2 == qptrDim0)
Call.assert "laed7: nlgn == permDim0" (nlgn == permDim0)
Call.assert "laed7: nlgn == givptrDim0" (nlgn == givptrDim0)
Call.assert "laed7: nlgn == givcolDim0" (nlgn == givcolDim0)
Call.assert "laed7: 2 == givcolDim1" (2 == givcolDim1)
Call.assert "laed7: nlgn == givnumDim0" (nlgn == givnumDim0)
Call.assert "laed7: 2 == givnumDim1" (2 == givnumDim1)
indxq <- Call.newArray1 n
work <- Call.newArray1 (qsiz*n)
rwork <- Call.newArray1 rworkSize
iwork <- Call.newArray1 (4*n)
evalContT $ do
nPtr <- Call.cint n
cutpntPtr <- Call.cint cutpnt
qsizPtr <- Call.cint qsiz
tlvlsPtr <- Call.cint tlvls
curlvlPtr <- Call.cint curlvl
curpbmPtr <- Call.cint curpbm
dPtr <- Call.ioarray d
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
rhoPtr <- Call.float rho
indxqPtr <- Call.ioarray indxq
qstorePtr <- Call.ioarray qstore
qptrPtr <- Call.ioarray qptr
prmptrPtr <- Call.array prmptr
permPtr <- Call.array perm
givptrPtr <- Call.array givptr
givcolPtr <- Call.array givcol
givnumPtr <- Call.array givnum
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.laed7 nPtr cutpntPtr qsizPtr tlvlsPtr curlvlPtr curpbmPtr dPtr qPtr ldqPtr rhoPtr indxqPtr qstorePtr qptrPtr prmptrPtr permPtr givptrPtr givcolPtr givnumPtr workPtr rworkPtr iworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray indxq
<*> fmap fromIntegral (peek infoPtr)
laed8 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt Float ->
Float ->
Int ->
Array ZeroInt Float ->
Int ->
Array ZeroInt CInt ->
IO (Int, Float, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt Float, Array ZeroInt CInt, Array ZeroInt CInt, Array ZeroInt CInt, Int, Array (ZeroInt,ZeroInt) CInt, Array (ZeroInt,ZeroInt) Float, Int)
laed8 qsiz q d rho cutpnt z ldq2 indxq = do
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let dDim0 = Call.sizes1 $ MutArray.shape d
let zDim0 = Call.sizes1 $ Array.shape z
let indxqDim0 = Call.sizes1 $ Array.shape indxq
let n = qDim0
let ldq = qDim1
Call.assert "laed8: n == dDim0" (n == dDim0)
Call.assert "laed8: n == zDim0" (n == zDim0)
Call.assert "laed8: n == indxqDim0" (n == indxqDim0)
dlamda <- Call.newArray1 n
q2 <- Call.newArray2 n ldq2
w <- Call.newArray1 n
indxp <- Call.newArray1 n
indx <- Call.newArray1 n
perm <- Call.newArray1 n
givcol <- Call.newArray2 n 2
givnum <- Call.newArray2 n 2
evalContT $ do
kPtr <- Call.alloca
nPtr <- Call.cint n
qsizPtr <- Call.cint qsiz
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
dPtr <- Call.ioarray d
rhoPtr <- Call.float rho
cutpntPtr <- Call.cint cutpnt
zPtr <- Call.array z
dlamdaPtr <- Call.ioarray dlamda
q2Ptr <- Call.ioarray q2
ldq2Ptr <- Call.cint ldq2
wPtr <- Call.ioarray w
indxpPtr <- Call.ioarray indxp
indxPtr <- Call.ioarray indx
indxqPtr <- Call.array indxq
permPtr <- Call.ioarray perm
givptrPtr <- Call.alloca
givcolPtr <- Call.ioarray givcol
givnumPtr <- Call.ioarray givnum
infoPtr <- Call.alloca
liftIO $ FFI.laed8 kPtr nPtr qsizPtr qPtr ldqPtr dPtr rhoPtr cutpntPtr zPtr dlamdaPtr q2Ptr ldq2Ptr wPtr indxpPtr indxPtr indxqPtr permPtr givptrPtr givcolPtr givnumPtr infoPtr
liftIO $ pure (,,,,,,,,,,,)
<*> fmap fromIntegral (peek kPtr)
<*> peek rhoPtr
<*> Call.freezeArray dlamda
<*> Call.freezeArray q2
<*> Call.freezeArray w
<*> Call.freezeArray indxp
<*> Call.freezeArray indx
<*> Call.freezeArray perm
<*> fmap fromIntegral (peek givptrPtr)
<*> Call.freezeArray givcol
<*> Call.freezeArray givnum
<*> fmap fromIntegral (peek infoPtr)
laein ::
Bool ->
Bool ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Complex Float ->
IOArray ZeroInt (Complex Float) ->
Int ->
Float ->
Float ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int)
laein rightv noinit h w v ldb eps3 smlnum = do
let (hDim0,hDim1) = Call.sizes2 $ Array.shape h
let vDim0 = Call.sizes1 $ MutArray.shape v
let n = hDim0
let ldh = hDim1
Call.assert "laein: n == vDim0" (n == vDim0)
b <- Call.newArray2 n ldb
rwork <- Call.newArray1 n
evalContT $ do
rightvPtr <- Call.bool rightv
noinitPtr <- Call.bool noinit
nPtr <- Call.cint n
hPtr <- Call.array h
ldhPtr <- Call.cint ldh
wPtr <- Call.complexFloat w
vPtr <- Call.ioarray v
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
rworkPtr <- Call.ioarray rwork
eps3Ptr <- Call.float eps3
smlnumPtr <- Call.float smlnum
infoPtr <- Call.alloca
liftIO $ FFI.laein rightvPtr noinitPtr nPtr hPtr ldhPtr wPtr vPtr bPtr ldbPtr rworkPtr eps3Ptr smlnumPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray b
<*> fmap fromIntegral (peek infoPtr)
laesy ::
Complex Float ->
Complex Float ->
Complex Float ->
IO (Complex Float, Complex Float, Complex Float, Complex Float, Complex Float)
laesy a b c = do
evalContT $ do
aPtr <- Call.complexFloat a
bPtr <- Call.complexFloat b
cPtr <- Call.complexFloat c
rt1Ptr <- Call.alloca
rt2Ptr <- Call.alloca
evscalPtr <- Call.alloca
cs1Ptr <- Call.alloca
sn1Ptr <- Call.alloca
liftIO $ FFI.laesy aPtr bPtr cPtr rt1Ptr rt2Ptr evscalPtr cs1Ptr sn1Ptr
liftIO $ pure (,,,,)
<*> peek rt1Ptr
<*> peek rt2Ptr
<*> peek evscalPtr
<*> peek cs1Ptr
<*> peek sn1Ptr
laev2 ::
Complex Float ->
Complex Float ->
Complex Float ->
IO (Float, Float, Float, Complex Float)
laev2 a b c = do
evalContT $ do
aPtr <- Call.complexFloat a
bPtr <- Call.complexFloat b
cPtr <- Call.complexFloat c
rt1Ptr <- Call.alloca
rt2Ptr <- Call.alloca
cs1Ptr <- Call.alloca
sn1Ptr <- Call.alloca
liftIO $ FFI.laev2 aPtr bPtr cPtr rt1Ptr rt2Ptr cs1Ptr sn1Ptr
liftIO $ pure (,,,)
<*> peek rt1Ptr
<*> peek rt2Ptr
<*> peek cs1Ptr
<*> peek sn1Ptr
lag2z ::
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Double), Int)
lag2z m sa lda = do
let (saDim0,saDim1) = Call.sizes2 $ Array.shape sa
let n = saDim0
let ldsa = saDim1
a <- Call.newArray2 n lda
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
saPtr <- Call.array sa
ldsaPtr <- Call.cint ldsa
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.lag2z mPtr nPtr saPtr ldsaPtr aPtr ldaPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray a
<*> fmap fromIntegral (peek infoPtr)
lags2 ::
Bool ->
Float ->
Complex Float ->
Float ->
Float ->
Complex Float ->
Float ->
IO (Float, Complex Float, Float, Complex Float, Float, Complex Float)
lags2 upper a1 a2 a3 b1 b2 b3 = do
evalContT $ do
upperPtr <- Call.bool upper
a1Ptr <- Call.float a1
a2Ptr <- Call.complexFloat a2
a3Ptr <- Call.float a3
b1Ptr <- Call.float b1
b2Ptr <- Call.complexFloat b2
b3Ptr <- Call.float b3
csuPtr <- Call.alloca
snuPtr <- Call.alloca
csvPtr <- Call.alloca
snvPtr <- Call.alloca
csqPtr <- Call.alloca
snqPtr <- Call.alloca
liftIO $ FFI.lags2 upperPtr a1Ptr a2Ptr a3Ptr b1Ptr b2Ptr b3Ptr csuPtr snuPtr csvPtr snvPtr csqPtr snqPtr
liftIO $ pure (,,,,,)
<*> peek csuPtr
<*> peek snuPtr
<*> peek csvPtr
<*> peek snvPtr
<*> peek csqPtr
<*> peek snqPtr
lagtm ::
Char ->
Float ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO ()
lagtm trans alpha dl d du x beta b = do
let dlDim0 = Call.sizes1 $ Array.shape dl
let dDim0 = Call.sizes1 $ Array.shape d
let duDim0 = Call.sizes1 $ Array.shape du
let (xDim0,xDim1) = Call.sizes2 $ Array.shape x
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = dDim0
let nrhs = xDim0
let ldx = xDim1
let ldb = bDim1
Call.assert "lagtm: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "lagtm: n-1 == duDim0" (n-1 == duDim0)
Call.assert "lagtm: nrhs == bDim0" (nrhs == bDim0)
evalContT $ do
transPtr <- Call.char trans
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
alphaPtr <- Call.float alpha
dlPtr <- Call.array dl
dPtr <- Call.array d
duPtr <- Call.array du
xPtr <- Call.array x
ldxPtr <- Call.cint ldx
betaPtr <- Call.float beta
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
liftIO $ FFI.lagtm transPtr nPtr nrhsPtr alphaPtr dlPtr dPtr duPtr xPtr ldxPtr betaPtr bPtr ldbPtr
lahef ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int, Array ZeroInt CInt, Array (ZeroInt,ZeroInt) (Complex Float), Int)
lahef uplo nb a ldw = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 n
w <- Call.newArray2 nb ldw
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nbPtr <- Call.cint nb
kbPtr <- Call.alloca
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
wPtr <- Call.ioarray w
ldwPtr <- Call.cint ldw
infoPtr <- Call.alloca
liftIO $ FFI.lahef uploPtr nPtr nbPtr kbPtr aPtr ldaPtr ipivPtr wPtr ldwPtr infoPtr
liftIO $ pure (,,,)
<*> fmap fromIntegral (peek kbPtr)
<*> Call.freezeArray ipiv
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
lahqr ::
Bool ->
Bool ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
lahqr wantt wantz ilo ihi h iloz ihiz z = do
let (hDim0,hDim1) = Call.sizes2 $ MutArray.shape h
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = hDim0
let ldh = hDim1
let ldz = zDim1
Call.assert "lahqr: n == zDim0" (n == zDim0)
w <- Call.newArray1 n
evalContT $ do
wanttPtr <- Call.bool wantt
wantzPtr <- Call.bool wantz
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
hPtr <- Call.ioarray h
ldhPtr <- Call.cint ldh
wPtr <- Call.ioarray w
ilozPtr <- Call.cint iloz
ihizPtr <- Call.cint ihiz
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
infoPtr <- Call.alloca
liftIO $ FFI.lahqr wanttPtr wantzPtr nPtr iloPtr ihiPtr hPtr ldhPtr wPtr ilozPtr ihizPtr zPtr ldzPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
lahr2 ::
Int ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float))
lahr2 n k nb a ldt ldy = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let lda = aDim1
Call.assert "lahr2: n-k+1 == aDim0" (n-k+1 == aDim0)
tau <- Call.newArray1 nb
t <- Call.newArray2 nb ldt
y <- Call.newArray2 nb ldy
evalContT $ do
nPtr <- Call.cint n
kPtr <- Call.cint k
nbPtr <- Call.cint nb
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
yPtr <- Call.ioarray y
ldyPtr <- Call.cint ldy
liftIO $ FFI.lahr2 nPtr kPtr nbPtr aPtr ldaPtr tauPtr tPtr ldtPtr yPtr ldyPtr
liftIO $ pure (,,)
<*> Call.freezeArray tau
<*> Call.freezeArray t
<*> Call.freezeArray y
laic1 ::
Int ->
Array ZeroInt (Complex Float) ->
Float ->
Array ZeroInt (Complex Float) ->
Complex Float ->
IO (Float, Complex Float, Complex Float)
laic1 job x sest w gamma = do
let xDim0 = Call.sizes1 $ Array.shape x
let wDim0 = Call.sizes1 $ Array.shape w
let j = xDim0
Call.assert "laic1: j == wDim0" (j == wDim0)
evalContT $ do
jobPtr <- Call.cint job
jPtr <- Call.cint j
xPtr <- Call.array x
sestPtr <- Call.float sest
wPtr <- Call.array w
gammaPtr <- Call.complexFloat gamma
sestprPtr <- Call.alloca
sPtr <- Call.alloca
cPtr <- Call.alloca
liftIO $ FFI.laic1 jobPtr jPtr xPtr sestPtr wPtr gammaPtr sestprPtr sPtr cPtr
liftIO $ pure (,,)
<*> peek sestprPtr
<*> peek sPtr
<*> peek cPtr
lals0 ::
Int ->
Int ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Array ZeroInt CInt ->
Int ->
Array (ZeroInt,ZeroInt) CInt ->
Array (ZeroInt,ZeroInt) Float ->
Array (ZeroInt,ZeroInt) Float ->
Array ZeroInt Float ->
Array (ZeroInt,ZeroInt) Float ->
Array ZeroInt Float ->
Float ->
Float ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int)
lals0 icompq nl nr sqre b ldbx perm givptr givcol givnum poles difl difr z c s rworkSize = do
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let permDim0 = Call.sizes1 $ Array.shape perm
let (givcolDim0,givcolDim1) = Call.sizes2 $ Array.shape givcol
let (givnumDim0,givnumDim1) = Call.sizes2 $ Array.shape givnum
let (polesDim0,polesDim1) = Call.sizes2 $ Array.shape poles
let diflDim0 = Call.sizes1 $ Array.shape difl
let (difrDim0,difrDim1) = Call.sizes2 $ Array.shape difr
let zDim0 = Call.sizes1 $ Array.shape z
let nrhs = bDim0
let ldb = bDim1
let _n = permDim0
let ldgcol = givcolDim1
let ldgnum = givnumDim1
let k = diflDim0
Call.assert "lals0: 2 == givcolDim0" (2 == givcolDim0)
Call.assert "lals0: 2 == givnumDim0" (2 == givnumDim0)
Call.assert "lals0: 2 == polesDim0" (2 == polesDim0)
Call.assert "lals0: ldgnum == polesDim1" (ldgnum == polesDim1)
Call.assert "lals0: 2 == difrDim0" (2 == difrDim0)
Call.assert "lals0: ldgnum == difrDim1" (ldgnum == difrDim1)
Call.assert "lals0: k == zDim0" (k == zDim0)
bx <- Call.newArray2 nrhs ldbx
rwork <- Call.newArray1 rworkSize
evalContT $ do
icompqPtr <- Call.cint icompq
nlPtr <- Call.cint nl
nrPtr <- Call.cint nr
sqrePtr <- Call.cint sqre
nrhsPtr <- Call.cint nrhs
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
bxPtr <- Call.ioarray bx
ldbxPtr <- Call.cint ldbx
permPtr <- Call.array perm
givptrPtr <- Call.cint givptr
givcolPtr <- Call.array givcol
ldgcolPtr <- Call.cint ldgcol
givnumPtr <- Call.array givnum
ldgnumPtr <- Call.cint ldgnum
polesPtr <- Call.array poles
diflPtr <- Call.array difl
difrPtr <- Call.array difr
zPtr <- Call.array z
kPtr <- Call.cint k
cPtr <- Call.float c
sPtr <- Call.float s
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.lals0 icompqPtr nlPtr nrPtr sqrePtr nrhsPtr bPtr ldbPtr bxPtr ldbxPtr permPtr givptrPtr givcolPtr ldgcolPtr givnumPtr ldgnumPtr polesPtr diflPtr difrPtr zPtr kPtr cPtr sPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray bx
<*> fmap fromIntegral (peek infoPtr)
lalsa ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Array (ZeroInt,ZeroInt) Float ->
Array (ZeroInt,ZeroInt) Float ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) Float ->
Array (ZeroInt,ZeroInt) Float ->
Array (ZeroInt,ZeroInt) Float ->
Array (ZeroInt,ZeroInt) Float ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) CInt ->
Array (ZeroInt,ZeroInt) CInt ->
Array (ZeroInt,ZeroInt) Float ->
Array ZeroInt Float ->
Array ZeroInt Float ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int)
lalsa icompq b ldbx u vt k difl difr z poles givptr givcol perm givnum c s rworkSize = do
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let (uDim0,uDim1) = Call.sizes2 $ Array.shape u
let (vtDim0,vtDim1) = Call.sizes2 $ Array.shape vt
let kDim0 = Call.sizes1 $ Array.shape k
let (diflDim0,diflDim1) = Call.sizes2 $ Array.shape difl
let (difrDim0,difrDim1) = Call.sizes2 $ Array.shape difr
let (zDim0,zDim1) = Call.sizes2 $ Array.shape z
let (polesDim0,polesDim1) = Call.sizes2 $ Array.shape poles
let givptrDim0 = Call.sizes1 $ Array.shape givptr
let (givcolDim0,givcolDim1) = Call.sizes2 $ Array.shape givcol
let (permDim0,permDim1) = Call.sizes2 $ Array.shape perm
let (givnumDim0,givnumDim1) = Call.sizes2 $ Array.shape givnum
let cDim0 = Call.sizes1 $ Array.shape c
let sDim0 = Call.sizes1 $ Array.shape s
let nrhs = bDim0
let ldb = bDim1
let smlsiz = uDim0
let ldu = uDim1
let n = kDim0
let nlvl = diflDim0
let ldgcol = givcolDim1
Call.assert "lalsa: smlsiz+1 == vtDim0" (smlsiz+1 == vtDim0)
Call.assert "lalsa: ldu == vtDim1" (ldu == vtDim1)
Call.assert "lalsa: ldu == diflDim1" (ldu == diflDim1)
Call.assert "lalsa: 2*nlvl == difrDim0" (2*nlvl == difrDim0)
Call.assert "lalsa: ldu == difrDim1" (ldu == difrDim1)
Call.assert "lalsa: nlvl == zDim0" (nlvl == zDim0)
Call.assert "lalsa: ldu == zDim1" (ldu == zDim1)
Call.assert "lalsa: 2*nlvl == polesDim0" (2*nlvl == polesDim0)
Call.assert "lalsa: ldu == polesDim1" (ldu == polesDim1)
Call.assert "lalsa: n == givptrDim0" (n == givptrDim0)
Call.assert "lalsa: 2*nlvl == givcolDim0" (2*nlvl == givcolDim0)
Call.assert "lalsa: nlvl == permDim0" (nlvl == permDim0)
Call.assert "lalsa: ldgcol == permDim1" (ldgcol == permDim1)
Call.assert "lalsa: 2*nlvl == givnumDim0" (2*nlvl == givnumDim0)
Call.assert "lalsa: ldu == givnumDim1" (ldu == givnumDim1)
Call.assert "lalsa: n == cDim0" (n == cDim0)
Call.assert "lalsa: n == sDim0" (n == sDim0)
bx <- Call.newArray2 nrhs ldbx
rwork <- Call.newArray1 rworkSize
iwork <- Call.newArray1 (3*n)
evalContT $ do
icompqPtr <- Call.cint icompq
smlsizPtr <- Call.cint smlsiz
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
bxPtr <- Call.ioarray bx
ldbxPtr <- Call.cint ldbx
uPtr <- Call.array u
lduPtr <- Call.cint ldu
vtPtr <- Call.array vt
kPtr <- Call.array k
diflPtr <- Call.array difl
difrPtr <- Call.array difr
zPtr <- Call.array z
polesPtr <- Call.array poles
givptrPtr <- Call.array givptr
givcolPtr <- Call.array givcol
ldgcolPtr <- Call.cint ldgcol
permPtr <- Call.array perm
givnumPtr <- Call.array givnum
cPtr <- Call.array c
sPtr <- Call.array s
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.lalsa icompqPtr smlsizPtr nPtr nrhsPtr bPtr ldbPtr bxPtr ldbxPtr uPtr lduPtr vtPtr kPtr diflPtr difrPtr zPtr polesPtr givptrPtr givcolPtr ldgcolPtr permPtr givnumPtr cPtr sPtr rworkPtr iworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray bx
<*> fmap fromIntegral (peek infoPtr)
lalsd ::
Char ->
Int ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Int ->
Int ->
IO (Int, Int)
lalsd uplo smlsiz d e b rcond rworkSize nlvl = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "lalsd: n-1 == eDim0" (n-1 == eDim0)
work <- Call.newArray1 (n*nrhs)
rwork <- Call.newArray1 rworkSize
iwork <- Call.newArray1 (3*n*nlvl+11*n)
evalContT $ do
uploPtr <- Call.char uplo
smlsizPtr <- Call.cint smlsiz
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
rcondPtr <- Call.float rcond
rankPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.lalsd uploPtr smlsizPtr nPtr nrhsPtr dPtr ePtr bPtr ldbPtr rcondPtr rankPtr workPtr rworkPtr iworkPtr infoPtr
liftIO $ pure (,)
<*> fmap fromIntegral (peek rankPtr)
<*> fmap fromIntegral (peek infoPtr)
langb ::
Char ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
langb norm kl ku ab lwork = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
workPtr <- Call.ioarray work
liftIO $ FFI.langb normPtr nPtr klPtr kuPtr abPtr ldabPtr workPtr
lange ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
lange norm m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
workPtr <- Call.ioarray work
liftIO $ FFI.lange normPtr mPtr nPtr aPtr ldaPtr workPtr
langt ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
IO Float
langt norm dl d du = do
let dlDim0 = Call.sizes1 $ Array.shape dl
let dDim0 = Call.sizes1 $ Array.shape d
let duDim0 = Call.sizes1 $ Array.shape du
let n = dDim0
Call.assert "langt: n-1 == dlDim0" (n-1 == dlDim0)
Call.assert "langt: n-1 == duDim0" (n-1 == duDim0)
evalContT $ do
normPtr <- Call.char norm
nPtr <- Call.cint n
dlPtr <- Call.array dl
dPtr <- Call.array d
duPtr <- Call.array du
liftIO $ FFI.langt normPtr nPtr dlPtr dPtr duPtr
lanhb ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
lanhb norm uplo k ab lwork = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kPtr <- Call.cint k
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
workPtr <- Call.ioarray work
liftIO $ FFI.lanhb normPtr uploPtr nPtr kPtr abPtr ldabPtr workPtr
lanhe ::
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
lanhe norm uplo a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
workPtr <- Call.ioarray work
liftIO $ FFI.lanhe normPtr uploPtr nPtr aPtr ldaPtr workPtr
lanhf ::
Char ->
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Int ->
IO Float
lanhf norm transr uplo n a lwork = do
let aDim0 = Call.sizes1 $ Array.shape a
Call.assert "lanhf: n*(n+1)`div`2 == aDim0" (n*(n+1)`div`2 == aDim0)
work <- Call.newArray1 lwork
evalContT $ do
normPtr <- Call.char norm
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
workPtr <- Call.ioarray work
liftIO $ FFI.lanhf normPtr transrPtr uploPtr nPtr aPtr workPtr
lanhp ::
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Int ->
IO Float
lanhp norm uplo n ap lwork = do
let apDim0 = Call.sizes1 $ Array.shape ap
Call.assert "lanhp: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
workPtr <- Call.ioarray work
liftIO $ FFI.lanhp normPtr uploPtr nPtr apPtr workPtr
lanhs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
lanhs norm a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
workPtr <- Call.ioarray work
liftIO $ FFI.lanhs normPtr nPtr aPtr ldaPtr workPtr
lanht ::
Char ->
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
IO Float
lanht norm d e = do
let dDim0 = Call.sizes1 $ Array.shape d
let eDim0 = Call.sizes1 $ Array.shape e
let n = dDim0
Call.assert "lanht: n-1 == eDim0" (n-1 == eDim0)
evalContT $ do
normPtr <- Call.char norm
nPtr <- Call.cint n
dPtr <- Call.array d
ePtr <- Call.array e
liftIO $ FFI.lanht normPtr nPtr dPtr ePtr
lansb ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
lansb norm uplo k ab lwork = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kPtr <- Call.cint k
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
workPtr <- Call.ioarray work
liftIO $ FFI.lansb normPtr uploPtr nPtr kPtr abPtr ldabPtr workPtr
lansp ::
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Int ->
IO Float
lansp norm uplo n ap lwork = do
let apDim0 = Call.sizes1 $ Array.shape ap
Call.assert "lansp: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
workPtr <- Call.ioarray work
liftIO $ FFI.lansp normPtr uploPtr nPtr apPtr workPtr
lansy ::
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
lansy norm uplo a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
workPtr <- Call.ioarray work
liftIO $ FFI.lansy normPtr uploPtr nPtr aPtr ldaPtr workPtr
lantb ::
Char ->
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
lantb norm uplo diag k ab lwork = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
kPtr <- Call.cint k
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
workPtr <- Call.ioarray work
liftIO $ FFI.lantb normPtr uploPtr diagPtr nPtr kPtr abPtr ldabPtr workPtr
lantp ::
Char ->
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Int ->
IO Float
lantp norm uplo diag n ap lwork = do
let apDim0 = Call.sizes1 $ Array.shape ap
Call.assert "lantp: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
apPtr <- Call.array ap
workPtr <- Call.ioarray work
liftIO $ FFI.lantp normPtr uploPtr diagPtr nPtr apPtr workPtr
lantr ::
Char ->
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO Float
lantr norm uplo diag m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
workPtr <- Call.ioarray work
liftIO $ FFI.lantr normPtr uploPtr diagPtr mPtr nPtr aPtr ldaPtr workPtr
lapll ::
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO (Float)
lapll n x incx y incy = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let yDim0 = Call.sizes1 $ MutArray.shape y
Call.assert "lapll: 1+(n-1)*incx == xDim0" (1+(n-1)*incx == xDim0)
Call.assert "lapll: 1+(n-1)*incy == yDim0" (1+(n-1)*incy == yDim0)
evalContT $ do
nPtr <- Call.cint n
xPtr <- Call.ioarray x
incxPtr <- Call.cint incx
yPtr <- Call.ioarray y
incyPtr <- Call.cint incy
ssminPtr <- Call.alloca
liftIO $ FFI.lapll nPtr xPtr incxPtr yPtr incyPtr ssminPtr
liftIO $ peek ssminPtr
lapmr ::
Bool ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
IO ()
lapmr forwrd x k = do
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let kDim0 = Call.sizes1 $ MutArray.shape k
let n = xDim0
let ldx = xDim1
let m = kDim0
evalContT $ do
forwrdPtr <- Call.bool forwrd
mPtr <- Call.cint m
nPtr <- Call.cint n
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
kPtr <- Call.ioarray k
liftIO $ FFI.lapmr forwrdPtr mPtr nPtr xPtr ldxPtr kPtr
lapmt ::
Bool ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
IO ()
lapmt forwrd m x k = do
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let kDim0 = Call.sizes1 $ MutArray.shape k
let n = xDim0
let ldx = xDim1
Call.assert "lapmt: n == kDim0" (n == kDim0)
evalContT $ do
forwrdPtr <- Call.bool forwrd
mPtr <- Call.cint m
nPtr <- Call.cint n
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
kPtr <- Call.ioarray k
liftIO $ FFI.lapmt forwrdPtr mPtr nPtr xPtr ldxPtr kPtr
laqgb ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt Float ->
Array ZeroInt Float ->
Float ->
Float ->
Float ->
IO (Char)
laqgb kl ku ab r c rowcnd colcnd amax = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let rDim0 = Call.sizes1 $ Array.shape r
let cDim0 = Call.sizes1 $ Array.shape c
let n = abDim0
let ldab = abDim1
let m = rDim0
Call.assert "laqgb: n == cDim0" (n == cDim0)
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
rPtr <- Call.array r
cPtr <- Call.array c
rowcndPtr <- Call.float rowcnd
colcndPtr <- Call.float colcnd
amaxPtr <- Call.float amax
equedPtr <- Call.alloca
liftIO $ FFI.laqgb mPtr nPtr klPtr kuPtr abPtr ldabPtr rPtr cPtr rowcndPtr colcndPtr amaxPtr equedPtr
liftIO $ fmap castCCharToChar (peek equedPtr)
laqge ::
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt Float ->
Array ZeroInt Float ->
Float ->
Float ->
Float ->
IO (Char)
laqge a r c rowcnd colcnd amax = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let rDim0 = Call.sizes1 $ Array.shape r
let cDim0 = Call.sizes1 $ Array.shape c
let n = aDim0
let lda = aDim1
let m = rDim0
Call.assert "laqge: n == cDim0" (n == cDim0)
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
rPtr <- Call.array r
cPtr <- Call.array c
rowcndPtr <- Call.float rowcnd
colcndPtr <- Call.float colcnd
amaxPtr <- Call.float amax
equedPtr <- Call.alloca
liftIO $ FFI.laqge mPtr nPtr aPtr ldaPtr rPtr cPtr rowcndPtr colcndPtr amaxPtr equedPtr
liftIO $ fmap castCCharToChar (peek equedPtr)
laqhb ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Float ->
IO (Array ZeroInt Float, Char)
laqhb uplo kd ab scond amax = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
s <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
sPtr <- Call.ioarray s
scondPtr <- Call.float scond
amaxPtr <- Call.float amax
equedPtr <- Call.alloca
liftIO $ FFI.laqhb uploPtr nPtr kdPtr abPtr ldabPtr sPtr scondPtr amaxPtr equedPtr
liftIO $ pure (,)
<*> Call.freezeArray s
<*> fmap castCCharToChar (peek equedPtr)
laqhe ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt Float ->
Float ->
Float ->
IO (Char)
laqhe uplo a s scond amax = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let sDim0 = Call.sizes1 $ Array.shape s
let n = aDim0
let lda = aDim1
Call.assert "laqhe: n == sDim0" (n == sDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
sPtr <- Call.array s
scondPtr <- Call.float scond
amaxPtr <- Call.float amax
equedPtr <- Call.alloca
liftIO $ FFI.laqhe uploPtr nPtr aPtr ldaPtr sPtr scondPtr amaxPtr equedPtr
liftIO $ fmap castCCharToChar (peek equedPtr)
laqhp ::
Char ->
IOArray ZeroInt (Complex Float) ->
Array ZeroInt Float ->
Float ->
Float ->
IO (Char)
laqhp uplo ap s scond amax = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let sDim0 = Call.sizes1 $ Array.shape s
let n = sDim0
Call.assert "laqhp: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
sPtr <- Call.array s
scondPtr <- Call.float scond
amaxPtr <- Call.float amax
equedPtr <- Call.alloca
liftIO $ FFI.laqhp uploPtr nPtr apPtr sPtr scondPtr amaxPtr equedPtr
liftIO $ fmap castCCharToChar (peek equedPtr)
laqp2 ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IO (Array ZeroInt (Complex Float))
laqp2 m offset a jpvt vn1 vn2 = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let jpvtDim0 = Call.sizes1 $ MutArray.shape jpvt
let vn1Dim0 = Call.sizes1 $ MutArray.shape vn1
let vn2Dim0 = Call.sizes1 $ MutArray.shape vn2
let n = aDim0
let lda = aDim1
Call.assert "laqp2: n == jpvtDim0" (n == jpvtDim0)
Call.assert "laqp2: n == vn1Dim0" (n == vn1Dim0)
Call.assert "laqp2: n == vn2Dim0" (n == vn2Dim0)
tau <- Call.newArray1 (minimum[m,n])
work <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
offsetPtr <- Call.cint offset
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
jpvtPtr <- Call.ioarray jpvt
tauPtr <- Call.ioarray tau
vn1Ptr <- Call.ioarray vn1
vn2Ptr <- Call.ioarray vn2
workPtr <- Call.ioarray work
liftIO $ FFI.laqp2 mPtr nPtr offsetPtr aPtr ldaPtr jpvtPtr tauPtr vn1Ptr vn2Ptr workPtr
liftIO $ Call.freezeArray tau
laqps ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
Int ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int, Array ZeroInt (Complex Float))
laqps m offset a jpvt kb vn1 vn2 auxv f = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let jpvtDim0 = Call.sizes1 $ MutArray.shape jpvt
let vn1Dim0 = Call.sizes1 $ MutArray.shape vn1
let vn2Dim0 = Call.sizes1 $ MutArray.shape vn2
let auxvDim0 = Call.sizes1 $ MutArray.shape auxv
let (fDim0,fDim1) = Call.sizes2 $ MutArray.shape f
let n = aDim0
let lda = aDim1
let nb = auxvDim0
let ldf = fDim1
Call.assert "laqps: n == jpvtDim0" (n == jpvtDim0)
Call.assert "laqps: n == vn1Dim0" (n == vn1Dim0)
Call.assert "laqps: n == vn2Dim0" (n == vn2Dim0)
Call.assert "laqps: nb == fDim0" (nb == fDim0)
tau <- Call.newArray1 kb
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
offsetPtr <- Call.cint offset
nbPtr <- Call.cint nb
kbPtr <- Call.alloca
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
jpvtPtr <- Call.ioarray jpvt
tauPtr <- Call.ioarray tau
vn1Ptr <- Call.ioarray vn1
vn2Ptr <- Call.ioarray vn2
auxvPtr <- Call.ioarray auxv
fPtr <- Call.ioarray f
ldfPtr <- Call.cint ldf
liftIO $ FFI.laqps mPtr nPtr offsetPtr nbPtr kbPtr aPtr ldaPtr jpvtPtr tauPtr vn1Ptr vn2Ptr auxvPtr fPtr ldfPtr
liftIO $ pure (,)
<*> fmap fromIntegral (peek kbPtr)
<*> Call.freezeArray tau
laqr0 ::
Bool ->
Bool ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
laqr0 wantt wantz ilo h iloz ihiz z workSize lwork = do
let (hDim0,hDim1) = Call.sizes2 $ MutArray.shape h
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = hDim0
let ldh = hDim1
let ihi = zDim0
let ldz = zDim1
w <- Call.newArray1 n
work <- Call.newArray1 workSize
evalContT $ do
wanttPtr <- Call.bool wantt
wantzPtr <- Call.bool wantz
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
hPtr <- Call.ioarray h
ldhPtr <- Call.cint ldh
wPtr <- Call.ioarray w
ilozPtr <- Call.cint iloz
ihizPtr <- Call.cint ihiz
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.laqr0 wanttPtr wantzPtr nPtr iloPtr ihiPtr hPtr ldhPtr wPtr ilozPtr ihizPtr zPtr ldzPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
laqr1 ::
Array (ZeroInt,ZeroInt) (Complex Float) ->
Complex Float ->
Complex Float ->
IO (Array ZeroInt (Complex Float))
laqr1 h s1 s2 = do
let (hDim0,hDim1) = Call.sizes2 $ Array.shape h
let n = hDim0
let ldh = hDim1
v <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
hPtr <- Call.array h
ldhPtr <- Call.cint ldh
s1Ptr <- Call.complexFloat s1
s2Ptr <- Call.complexFloat s2
vPtr <- Call.ioarray v
liftIO $ FFI.laqr1 nPtr hPtr ldhPtr s1Ptr s2Ptr vPtr
liftIO $ Call.freezeArray v
laqr2 ::
Bool ->
Bool ->
Int ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
Int ->
Int ->
IO (Int, Int, Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float))
laqr2 wantt wantz ktop kbot nw h iloz ihiz z ldv nh ldt nv ldwv lwork = do
let (hDim0,hDim1) = Call.sizes2 $ MutArray.shape h
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = hDim0
let ldh = hDim1
let ldz = zDim1
Call.assert "laqr2: n == zDim0" (n == zDim0)
sh <- Call.newArray1 kbot
v <- Call.newArray2 nw ldv
t <- Call.newArray2 nw ldt
wv <- Call.newArray2 nw ldwv
work <- Call.newArray1 lwork
evalContT $ do
wanttPtr <- Call.bool wantt
wantzPtr <- Call.bool wantz
nPtr <- Call.cint n
ktopPtr <- Call.cint ktop
kbotPtr <- Call.cint kbot
nwPtr <- Call.cint nw
hPtr <- Call.ioarray h
ldhPtr <- Call.cint ldh
ilozPtr <- Call.cint iloz
ihizPtr <- Call.cint ihiz
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
nsPtr <- Call.alloca
ndPtr <- Call.alloca
shPtr <- Call.ioarray sh
vPtr <- Call.ioarray v
ldvPtr <- Call.cint ldv
nhPtr <- Call.cint nh
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
nvPtr <- Call.cint nv
wvPtr <- Call.ioarray wv
ldwvPtr <- Call.cint ldwv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
liftIO $ FFI.laqr2 wanttPtr wantzPtr nPtr ktopPtr kbotPtr nwPtr hPtr ldhPtr ilozPtr ihizPtr zPtr ldzPtr nsPtr ndPtr shPtr vPtr ldvPtr nhPtr tPtr ldtPtr nvPtr wvPtr ldwvPtr workPtr lworkPtr
liftIO $ pure (,,,,,)
<*> fmap fromIntegral (peek nsPtr)
<*> fmap fromIntegral (peek ndPtr)
<*> Call.freezeArray sh
<*> Call.freezeArray v
<*> Call.freezeArray t
<*> Call.freezeArray wv
laqr3 ::
Bool ->
Bool ->
Int ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
Int ->
Int ->
IO (Int, Int, Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float))
laqr3 wantt wantz ktop kbot nw h iloz ihiz z ldv nh ldt nv ldwv lwork = do
let (hDim0,hDim1) = Call.sizes2 $ MutArray.shape h
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = hDim0
let ldh = hDim1
let ldz = zDim1
Call.assert "laqr3: n == zDim0" (n == zDim0)
sh <- Call.newArray1 kbot
v <- Call.newArray2 nw ldv
t <- Call.newArray2 nw ldt
wv <- Call.newArray2 nw ldwv
work <- Call.newArray1 lwork
evalContT $ do
wanttPtr <- Call.bool wantt
wantzPtr <- Call.bool wantz
nPtr <- Call.cint n
ktopPtr <- Call.cint ktop
kbotPtr <- Call.cint kbot
nwPtr <- Call.cint nw
hPtr <- Call.ioarray h
ldhPtr <- Call.cint ldh
ilozPtr <- Call.cint iloz
ihizPtr <- Call.cint ihiz
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
nsPtr <- Call.alloca
ndPtr <- Call.alloca
shPtr <- Call.ioarray sh
vPtr <- Call.ioarray v
ldvPtr <- Call.cint ldv
nhPtr <- Call.cint nh
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
nvPtr <- Call.cint nv
wvPtr <- Call.ioarray wv
ldwvPtr <- Call.cint ldwv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
liftIO $ FFI.laqr3 wanttPtr wantzPtr nPtr ktopPtr kbotPtr nwPtr hPtr ldhPtr ilozPtr ihizPtr zPtr ldzPtr nsPtr ndPtr shPtr vPtr ldvPtr nhPtr tPtr ldtPtr nvPtr wvPtr ldwvPtr workPtr lworkPtr
liftIO $ pure (,,,,,)
<*> fmap fromIntegral (peek nsPtr)
<*> fmap fromIntegral (peek ndPtr)
<*> Call.freezeArray sh
<*> Call.freezeArray v
<*> Call.freezeArray t
<*> Call.freezeArray wv
laqr4 ::
Bool ->
Bool ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
laqr4 wantt wantz ilo h iloz ihiz z workSize lwork = do
let (hDim0,hDim1) = Call.sizes2 $ MutArray.shape h
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = hDim0
let ldh = hDim1
let ihi = zDim0
let ldz = zDim1
w <- Call.newArray1 n
work <- Call.newArray1 workSize
evalContT $ do
wanttPtr <- Call.bool wantt
wantzPtr <- Call.bool wantz
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
hPtr <- Call.ioarray h
ldhPtr <- Call.cint ldh
wPtr <- Call.ioarray w
ilozPtr <- Call.cint iloz
ihizPtr <- Call.cint ihiz
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.laqr4 wanttPtr wantzPtr nPtr iloPtr ihiPtr hPtr ldhPtr wPtr ilozPtr ihizPtr zPtr ldzPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
laqr5 ::
Bool ->
Bool ->
Int ->
Int ->
Int ->
IOArray ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
Int ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float))
laqr5 wantt wantz kacc22 ktop kbot s h iloz z ldv ldu nv ldwv nh ldwh = do
let sDim0 = Call.sizes1 $ MutArray.shape s
let (hDim0,hDim1) = Call.sizes2 $ MutArray.shape h
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let nshfts = sDim0
let n = hDim0
let ldh = hDim1
let ihiz = zDim0
let ldz = zDim1
v <- Call.newArray2 (nshfts`div`2) ldv
u <- Call.newArray2 (3*nshfts-3) ldu
wv <- Call.newArray2 (3*nshfts-3) ldwv
wh <- Call.newArray2 nh ldwh
evalContT $ do
wanttPtr <- Call.bool wantt
wantzPtr <- Call.bool wantz
kacc22Ptr <- Call.cint kacc22
nPtr <- Call.cint n
ktopPtr <- Call.cint ktop
kbotPtr <- Call.cint kbot
nshftsPtr <- Call.cint nshfts
sPtr <- Call.ioarray s
hPtr <- Call.ioarray h
ldhPtr <- Call.cint ldh
ilozPtr <- Call.cint iloz
ihizPtr <- Call.cint ihiz
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
vPtr <- Call.ioarray v
ldvPtr <- Call.cint ldv
uPtr <- Call.ioarray u
lduPtr <- Call.cint ldu
nvPtr <- Call.cint nv
wvPtr <- Call.ioarray wv
ldwvPtr <- Call.cint ldwv
nhPtr <- Call.cint nh
whPtr <- Call.ioarray wh
ldwhPtr <- Call.cint ldwh
liftIO $ FFI.laqr5 wanttPtr wantzPtr kacc22Ptr nPtr ktopPtr kbotPtr nshftsPtr sPtr hPtr ldhPtr ilozPtr ihizPtr zPtr ldzPtr vPtr ldvPtr uPtr lduPtr nvPtr wvPtr ldwvPtr nhPtr whPtr ldwhPtr
liftIO $ pure (,,,)
<*> Call.freezeArray v
<*> Call.freezeArray u
<*> Call.freezeArray wv
<*> Call.freezeArray wh
laqsb ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt Float ->
Float ->
Float ->
IO (Char)
laqsb uplo kd ab s scond amax = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let sDim0 = Call.sizes1 $ Array.shape s
let n = abDim0
let ldab = abDim1
Call.assert "laqsb: n == sDim0" (n == sDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
sPtr <- Call.array s
scondPtr <- Call.float scond
amaxPtr <- Call.float amax
equedPtr <- Call.alloca
liftIO $ FFI.laqsb uploPtr nPtr kdPtr abPtr ldabPtr sPtr scondPtr amaxPtr equedPtr
liftIO $ fmap castCCharToChar (peek equedPtr)
laqsp ::
Char ->
IOArray ZeroInt (Complex Float) ->
Array ZeroInt Float ->
Float ->
Float ->
IO (Char)
laqsp uplo ap s scond amax = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let sDim0 = Call.sizes1 $ Array.shape s
let n = sDim0
Call.assert "laqsp: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
sPtr <- Call.array s
scondPtr <- Call.float scond
amaxPtr <- Call.float amax
equedPtr <- Call.alloca
liftIO $ FFI.laqsp uploPtr nPtr apPtr sPtr scondPtr amaxPtr equedPtr
liftIO $ fmap castCCharToChar (peek equedPtr)
laqsy ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt Float ->
Float ->
Float ->
IO (Char)
laqsy uplo a s scond amax = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let sDim0 = Call.sizes1 $ Array.shape s
let n = aDim0
let lda = aDim1
Call.assert "laqsy: n == sDim0" (n == sDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
sPtr <- Call.array s
scondPtr <- Call.float scond
amaxPtr <- Call.float amax
equedPtr <- Call.alloca
liftIO $ FFI.laqsy uploPtr nPtr aPtr ldaPtr sPtr scondPtr amaxPtr equedPtr
liftIO $ fmap castCCharToChar (peek equedPtr)
lar1v ::
Int ->
Int ->
Float ->
Array ZeroInt Float ->
Array ZeroInt Float ->
Array ZeroInt Float ->
Array ZeroInt Float ->
Float ->
Float ->
IOArray ZeroInt (Complex Float) ->
Bool ->
Int ->
IO (Int, Float, Float, Int, Array ZeroInt CInt, Float, Float, Float)
lar1v b1 bn lambda d l ld lld pivmin gaptol z wantnc r = do
let dDim0 = Call.sizes1 $ Array.shape d
let lDim0 = Call.sizes1 $ Array.shape l
let ldDim0 = Call.sizes1 $ Array.shape ld
let lldDim0 = Call.sizes1 $ Array.shape lld
let zDim0 = Call.sizes1 $ MutArray.shape z
let n = dDim0
Call.assert "lar1v: n-1 == lDim0" (n-1 == lDim0)
Call.assert "lar1v: n-1 == ldDim0" (n-1 == ldDim0)
Call.assert "lar1v: n-1 == lldDim0" (n-1 == lldDim0)
Call.assert "lar1v: n == zDim0" (n == zDim0)
isuppz <- Call.newArray1 2
work <- Call.newArray1 (4*n)
evalContT $ do
nPtr <- Call.cint n
b1Ptr <- Call.cint b1
bnPtr <- Call.cint bn
lambdaPtr <- Call.float lambda
dPtr <- Call.array d
lPtr <- Call.array l
ldPtr <- Call.array ld
lldPtr <- Call.array lld
pivminPtr <- Call.float pivmin
gaptolPtr <- Call.float gaptol
zPtr <- Call.ioarray z
wantncPtr <- Call.bool wantnc
negcntPtr <- Call.alloca
ztzPtr <- Call.alloca
mingmaPtr <- Call.alloca
rPtr <- Call.cint r
isuppzPtr <- Call.ioarray isuppz
nrminvPtr <- Call.alloca
residPtr <- Call.alloca
rqcorrPtr <- Call.alloca
workPtr <- Call.ioarray work
liftIO $ FFI.lar1v nPtr b1Ptr bnPtr lambdaPtr dPtr lPtr ldPtr lldPtr pivminPtr gaptolPtr zPtr wantncPtr negcntPtr ztzPtr mingmaPtr rPtr isuppzPtr nrminvPtr residPtr rqcorrPtr workPtr
liftIO $ pure (,,,,,,,)
<*> fmap fromIntegral (peek negcntPtr)
<*> peek ztzPtr
<*> peek mingmaPtr
<*> fmap fromIntegral (peek rPtr)
<*> Call.freezeArray isuppz
<*> peek nrminvPtr
<*> peek residPtr
<*> peek rqcorrPtr
lar2v ::
Int ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Int ->
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
Int ->
IO ()
lar2v n x y z incx c s incc = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let yDim0 = Call.sizes1 $ MutArray.shape y
let zDim0 = Call.sizes1 $ MutArray.shape z
let cDim0 = Call.sizes1 $ Array.shape c
let sDim0 = Call.sizes1 $ Array.shape s
Call.assert "lar2v: 1+(n-1)*incx == xDim0" (1+(n-1)*incx == xDim0)
Call.assert "lar2v: 1+(n-1)*incx == yDim0" (1+(n-1)*incx == yDim0)
Call.assert "lar2v: 1+(n-1)*incx == zDim0" (1+(n-1)*incx == zDim0)
Call.assert "lar2v: 1+(n-1)*incc == cDim0" (1+(n-1)*incc == cDim0)
Call.assert "lar2v: 1+(n-1)*incc == sDim0" (1+(n-1)*incc == sDim0)
evalContT $ do
nPtr <- Call.cint n
xPtr <- Call.ioarray x
yPtr <- Call.ioarray y
zPtr <- Call.ioarray z
incxPtr <- Call.cint incx
cPtr <- Call.array c
sPtr <- Call.array s
inccPtr <- Call.cint incc
liftIO $ FFI.lar2v nPtr xPtr yPtr zPtr incxPtr cPtr sPtr inccPtr
larcm ::
Array (ZeroInt,ZeroInt) Float ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float))
larcm a b ldc = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let m = aDim0
let lda = aDim1
let n = bDim0
let ldb = bDim1
c <- Call.newArray2 n ldc
rwork <- Call.newArray1 (2*m*n)
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
rworkPtr <- Call.ioarray rwork
liftIO $ FFI.larcm mPtr nPtr aPtr ldaPtr bPtr ldbPtr cPtr ldcPtr rworkPtr
liftIO $ Call.freezeArray c
larf ::
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Int ->
Complex Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO ()
larf side m v incv tau c workSize = do
let vDim0 = Call.sizes1 $ Array.shape v
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _vSize = vDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
mPtr <- Call.cint m
nPtr <- Call.cint n
vPtr <- Call.array v
incvPtr <- Call.cint incv
tauPtr <- Call.complexFloat tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
liftIO $ FFI.larf sidePtr mPtr nPtr vPtr incvPtr tauPtr cPtr ldcPtr workPtr
larfb ::
Char ->
Char ->
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO ()
larfb side trans direct storev m v t c ldwork = do
let (vDim0,vDim1) = Call.sizes2 $ Array.shape v
let (tDim0,tDim1) = Call.sizes2 $ Array.shape t
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _vSize = vDim0
let ldv = vDim1
let k = tDim0
let ldt = tDim1
let n = cDim0
let ldc = cDim1
work <- Call.newArray2 k ldwork
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
directPtr <- Call.char direct
storevPtr <- Call.char storev
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
vPtr <- Call.array v
ldvPtr <- Call.cint ldv
tPtr <- Call.array t
ldtPtr <- Call.cint ldt
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
ldworkPtr <- Call.cint ldwork
liftIO $ FFI.larfb sidePtr transPtr directPtr storevPtr mPtr nPtr kPtr vPtr ldvPtr tPtr ldtPtr cPtr ldcPtr workPtr ldworkPtr
larfg ::
Int ->
Complex Float ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO (Complex Float, Complex Float)
larfg n alpha x incx = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let _xSize = xDim0
evalContT $ do
nPtr <- Call.cint n
alphaPtr <- Call.complexFloat alpha
xPtr <- Call.ioarray x
incxPtr <- Call.cint incx
tauPtr <- Call.alloca
liftIO $ FFI.larfg nPtr alphaPtr xPtr incxPtr tauPtr
liftIO $ pure (,)
<*> peek alphaPtr
<*> peek tauPtr
larfgp ::
Int ->
Complex Float ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO (Complex Float, Complex Float)
larfgp n alpha x incx = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let _xSize = xDim0
evalContT $ do
nPtr <- Call.cint n
alphaPtr <- Call.complexFloat alpha
xPtr <- Call.ioarray x
incxPtr <- Call.cint incx
tauPtr <- Call.alloca
liftIO $ FFI.larfgp nPtr alphaPtr xPtr incxPtr tauPtr
liftIO $ pure (,)
<*> peek alphaPtr
<*> peek tauPtr
larft ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float))
larft direct storev n v tau ldt = do
let (vDim0,vDim1) = Call.sizes2 $ Array.shape v
let tauDim0 = Call.sizes1 $ Array.shape tau
let _vSize = vDim0
let ldv = vDim1
let k = tauDim0
t <- Call.newArray2 k ldt
evalContT $ do
directPtr <- Call.char direct
storevPtr <- Call.char storev
nPtr <- Call.cint n
kPtr <- Call.cint k
vPtr <- Call.array v
ldvPtr <- Call.cint ldv
tauPtr <- Call.array tau
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
liftIO $ FFI.larft directPtr storevPtr nPtr kPtr vPtr ldvPtr tauPtr tPtr ldtPtr
liftIO $ Call.freezeArray t
larfx ::
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Complex Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO ()
larfx side m v tau c workSize = do
let vDim0 = Call.sizes1 $ Array.shape v
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _vSize = vDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
mPtr <- Call.cint m
nPtr <- Call.cint n
vPtr <- Call.array v
tauPtr <- Call.complexFloat tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
liftIO $ FFI.larfx sidePtr mPtr nPtr vPtr tauPtr cPtr ldcPtr workPtr
largv ::
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
Int ->
IO (Array ZeroInt Float)
largv n x incx y incy incc = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let yDim0 = Call.sizes1 $ MutArray.shape y
Call.assert "largv: 1+(n-1)*incx == xDim0" (1+(n-1)*incx == xDim0)
Call.assert "largv: 1+(n-1)*incy == yDim0" (1+(n-1)*incy == yDim0)
c <- Call.newArray1 (1+(n-1)*incc)
evalContT $ do
nPtr <- Call.cint n
xPtr <- Call.ioarray x
incxPtr <- Call.cint incx
yPtr <- Call.ioarray y
incyPtr <- Call.cint incy
cPtr <- Call.ioarray c
inccPtr <- Call.cint incc
liftIO $ FFI.largv nPtr xPtr incxPtr yPtr incyPtr cPtr inccPtr
liftIO $ Call.freezeArray c
larnv ::
Int ->
IOArray ZeroInt CInt ->
Int ->
IO (Array ZeroInt (Complex Float))
larnv idist iseed n = do
let iseedDim0 = Call.sizes1 $ MutArray.shape iseed
Call.assert "larnv: 4 == iseedDim0" (4 == iseedDim0)
x <- Call.newArray1 n
evalContT $ do
idistPtr <- Call.cint idist
iseedPtr <- Call.ioarray iseed
nPtr <- Call.cint n
xPtr <- Call.ioarray x
liftIO $ FFI.larnv idistPtr iseedPtr nPtr xPtr
liftIO $ Call.freezeArray x
larrv ::
Float ->
Float ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
Float ->
Array ZeroInt CInt ->
Int ->
Int ->
Int ->
Float ->
Float ->
Float ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
Array ZeroInt CInt ->
Array ZeroInt CInt ->
Array ZeroInt Float ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
larrv vl vu d l pivmin isplit m dol dou minrgp rtol1 rtol2 w werr wgap iblock indexw gers ldz = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let lDim0 = Call.sizes1 $ MutArray.shape l
let isplitDim0 = Call.sizes1 $ Array.shape isplit
let wDim0 = Call.sizes1 $ MutArray.shape w
let werrDim0 = Call.sizes1 $ MutArray.shape werr
let wgapDim0 = Call.sizes1 $ MutArray.shape wgap
let iblockDim0 = Call.sizes1 $ Array.shape iblock
let indexwDim0 = Call.sizes1 $ Array.shape indexw
let gersDim0 = Call.sizes1 $ Array.shape gers
let n = dDim0
Call.assert "larrv: n == lDim0" (n == lDim0)
Call.assert "larrv: n == isplitDim0" (n == isplitDim0)
Call.assert "larrv: n == wDim0" (n == wDim0)
Call.assert "larrv: n == werrDim0" (n == werrDim0)
Call.assert "larrv: n == wgapDim0" (n == wgapDim0)
Call.assert "larrv: n == iblockDim0" (n == iblockDim0)
Call.assert "larrv: n == indexwDim0" (n == indexwDim0)
Call.assert "larrv: 2*n == gersDim0" (2*n == gersDim0)
z <- Call.newArray2 (maximum[1,m]) ldz
isuppz <- Call.newArray1 (2*maximum[1,m])
work <- Call.newArray1 (12*n)
iwork <- Call.newArray1 (7*n)
evalContT $ do
nPtr <- Call.cint n
vlPtr <- Call.float vl
vuPtr <- Call.float vu
dPtr <- Call.ioarray d
lPtr <- Call.ioarray l
pivminPtr <- Call.float pivmin
isplitPtr <- Call.array isplit
mPtr <- Call.cint m
dolPtr <- Call.cint dol
douPtr <- Call.cint dou
minrgpPtr <- Call.float minrgp
rtol1Ptr <- Call.float rtol1
rtol2Ptr <- Call.float rtol2
wPtr <- Call.ioarray w
werrPtr <- Call.ioarray werr
wgapPtr <- Call.ioarray wgap
iblockPtr <- Call.array iblock
indexwPtr <- Call.array indexw
gersPtr <- Call.array gers
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
isuppzPtr <- Call.ioarray isuppz
workPtr <- Call.ioarray work
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.larrv nPtr vlPtr vuPtr dPtr lPtr pivminPtr isplitPtr mPtr dolPtr douPtr minrgpPtr rtol1Ptr rtol2Ptr wPtr werrPtr wgapPtr iblockPtr indexwPtr gersPtr zPtr ldzPtr isuppzPtr workPtr iworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray z
<*> Call.freezeArray isuppz
<*> fmap fromIntegral (peek infoPtr)
lartg ::
Complex Float ->
Complex Float ->
IO (Float, Complex Float, Complex Float)
lartg f g = do
evalContT $ do
fPtr <- Call.complexFloat f
gPtr <- Call.complexFloat g
csPtr <- Call.alloca
snPtr <- Call.alloca
rPtr <- Call.alloca
liftIO $ FFI.lartg fPtr gPtr csPtr snPtr rPtr
liftIO $ pure (,,)
<*> peek csPtr
<*> peek snPtr
<*> peek rPtr
lartv ::
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
Int ->
IO ()
lartv n x incx y incy c s incc = do
let xDim0 = Call.sizes1 $ MutArray.shape x
let yDim0 = Call.sizes1 $ MutArray.shape y
let cDim0 = Call.sizes1 $ Array.shape c
let sDim0 = Call.sizes1 $ Array.shape s
Call.assert "lartv: 1+(n-1)*incx == xDim0" (1+(n-1)*incx == xDim0)
Call.assert "lartv: 1+(n-1)*incy == yDim0" (1+(n-1)*incy == yDim0)
Call.assert "lartv: 1+(n-1)*incc == cDim0" (1+(n-1)*incc == cDim0)
Call.assert "lartv: 1+(n-1)*incc == sDim0" (1+(n-1)*incc == sDim0)
evalContT $ do
nPtr <- Call.cint n
xPtr <- Call.ioarray x
incxPtr <- Call.cint incx
yPtr <- Call.ioarray y
incyPtr <- Call.cint incy
cPtr <- Call.array c
sPtr <- Call.array s
inccPtr <- Call.cint incc
liftIO $ FFI.lartv nPtr xPtr incxPtr yPtr incyPtr cPtr sPtr inccPtr
larz ::
Char ->
Int ->
Int ->
Array ZeroInt (Complex Float) ->
Int ->
Complex Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO ()
larz side m l v incv tau c workSize = do
let vDim0 = Call.sizes1 $ Array.shape v
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let n = cDim0
let ldc = cDim1
Call.assert "larz: 1+(l-1)*abs(incv) == vDim0" (1+(l-1)*abs(incv) == vDim0)
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
mPtr <- Call.cint m
nPtr <- Call.cint n
lPtr <- Call.cint l
vPtr <- Call.array v
incvPtr <- Call.cint incv
tauPtr <- Call.complexFloat tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
liftIO $ FFI.larz sidePtr mPtr nPtr lPtr vPtr incvPtr tauPtr cPtr ldcPtr workPtr
larzb ::
Char ->
Char ->
Char ->
Char ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO ()
larzb side trans direct storev m l v t c ldwork = do
let (vDim0,vDim1) = Call.sizes2 $ Array.shape v
let (tDim0,tDim1) = Call.sizes2 $ Array.shape t
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _nv = vDim0
let ldv = vDim1
let k = tDim0
let ldt = tDim1
let n = cDim0
let ldc = cDim1
work <- Call.newArray2 k ldwork
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
directPtr <- Call.char direct
storevPtr <- Call.char storev
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
lPtr <- Call.cint l
vPtr <- Call.array v
ldvPtr <- Call.cint ldv
tPtr <- Call.array t
ldtPtr <- Call.cint ldt
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
ldworkPtr <- Call.cint ldwork
liftIO $ FFI.larzb sidePtr transPtr directPtr storevPtr mPtr nPtr kPtr lPtr vPtr ldvPtr tPtr ldtPtr cPtr ldcPtr workPtr ldworkPtr
larzt ::
Char ->
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float))
larzt direct storev n v tau ldt = do
let (vDim0,vDim1) = Call.sizes2 $ MutArray.shape v
let tauDim0 = Call.sizes1 $ Array.shape tau
let _vSize = vDim0
let ldv = vDim1
let k = tauDim0
t <- Call.newArray2 k ldt
evalContT $ do
directPtr <- Call.char direct
storevPtr <- Call.char storev
nPtr <- Call.cint n
kPtr <- Call.cint k
vPtr <- Call.ioarray v
ldvPtr <- Call.cint ldv
tauPtr <- Call.array tau
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
liftIO $ FFI.larzt directPtr storevPtr nPtr kPtr vPtr ldvPtr tauPtr tPtr ldtPtr
liftIO $ Call.freezeArray t
lascl ::
Char ->
Int ->
Int ->
Float ->
Float ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
lascl type_ kl ku cfrom cto m a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
type_Ptr <- Call.char type_
klPtr <- Call.cint kl
kuPtr <- Call.cint ku
cfromPtr <- Call.float cfrom
ctoPtr <- Call.float cto
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.lascl type_Ptr klPtr kuPtr cfromPtr ctoPtr mPtr nPtr aPtr ldaPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
laset ::
Char ->
Int ->
Int ->
Complex Float ->
Complex Float ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float))
laset uplo m n alpha beta lda = do
a <- Call.newArray2 n lda
evalContT $ do
uploPtr <- Call.char uplo
mPtr <- Call.cint m
nPtr <- Call.cint n
alphaPtr <- Call.complexFloat alpha
betaPtr <- Call.complexFloat beta
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
liftIO $ FFI.laset uploPtr mPtr nPtr alphaPtr betaPtr aPtr ldaPtr
liftIO $ Call.freezeArray a
lasr ::
Char ->
Char ->
Char ->
Int ->
Array ZeroInt Float ->
Array ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO ()
lasr side pivot direct m c s a = do
let cDim0 = Call.sizes1 $ Array.shape c
let sDim0 = Call.sizes1 $ Array.shape s
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let _cSize = cDim0
let _sSize = sDim0
let n = aDim0
let lda = aDim1
evalContT $ do
sidePtr <- Call.char side
pivotPtr <- Call.char pivot
directPtr <- Call.char direct
mPtr <- Call.cint m
nPtr <- Call.cint n
cPtr <- Call.array c
sPtr <- Call.array s
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
liftIO $ FFI.lasr sidePtr pivotPtr directPtr mPtr nPtr cPtr sPtr aPtr ldaPtr
lassq ::
Array ZeroInt (Complex Float) ->
Int ->
Float ->
Float ->
IO (Float, Float)
lassq x incx scale sumsq = do
let xDim0 = Call.sizes1 $ Array.shape x
let n = xDim0
evalContT $ do
nPtr <- Call.cint n
xPtr <- Call.array x
incxPtr <- Call.cint incx
scalePtr <- Call.float scale
sumsqPtr <- Call.float sumsq
liftIO $ FFI.lassq nPtr xPtr incxPtr scalePtr sumsqPtr
liftIO $ pure (,)
<*> peek scalePtr
<*> peek sumsqPtr
laswp ::
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Array ZeroInt CInt ->
Int ->
IO ()
laswp a k1 k2 ipiv incx = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "laswp: k1+(k2-k1)*abs(incx) == ipivDim0" (k1+(k2-k1)*abs(incx) == ipivDim0)
evalContT $ do
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
k1Ptr <- Call.cint k1
k2Ptr <- Call.cint k2
ipivPtr <- Call.array ipiv
incxPtr <- Call.cint incx
liftIO $ FFI.laswp nPtr aPtr ldaPtr k1Ptr k2Ptr ipivPtr incxPtr
lasyf ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int, Array ZeroInt CInt, Array (ZeroInt,ZeroInt) (Complex Float), Int)
lasyf uplo nb a ldw = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 n
w <- Call.newArray2 nb ldw
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nbPtr <- Call.cint nb
kbPtr <- Call.alloca
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
wPtr <- Call.ioarray w
ldwPtr <- Call.cint ldw
infoPtr <- Call.alloca
liftIO $ FFI.lasyf uploPtr nPtr nbPtr kbPtr aPtr ldaPtr ipivPtr wPtr ldwPtr infoPtr
liftIO $ pure (,,,)
<*> fmap fromIntegral (peek kbPtr)
<*> Call.freezeArray ipiv
<*> Call.freezeArray w
<*> fmap fromIntegral (peek infoPtr)
latbs ::
Char ->
Char ->
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt Float ->
IO (Float, Int)
latbs uplo trans diag normin kd ab x cnorm = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let xDim0 = Call.sizes1 $ MutArray.shape x
let cnormDim0 = Call.sizes1 $ MutArray.shape cnorm
let n = abDim0
let ldab = abDim1
Call.assert "latbs: n == xDim0" (n == xDim0)
Call.assert "latbs: n == cnormDim0" (n == cnormDim0)
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
norminPtr <- Call.char normin
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
xPtr <- Call.ioarray x
scalePtr <- Call.alloca
cnormPtr <- Call.ioarray cnorm
infoPtr <- Call.alloca
liftIO $ FFI.latbs uploPtr transPtr diagPtr norminPtr nPtr kdPtr abPtr ldabPtr xPtr scalePtr cnormPtr infoPtr
liftIO $ pure (,)
<*> peek scalePtr
<*> fmap fromIntegral (peek infoPtr)
latdf ::
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Float ->
Float ->
Array ZeroInt CInt ->
Array ZeroInt CInt ->
IO (Float, Float)
latdf ijob z rhs rdsum rdscal ipiv jpiv = do
let (zDim0,zDim1) = Call.sizes2 $ Array.shape z
let rhsDim0 = Call.sizes1 $ MutArray.shape rhs
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let jpivDim0 = Call.sizes1 $ Array.shape jpiv
let n = zDim0
let ldz = zDim1
Call.assert "latdf: n == rhsDim0" (n == rhsDim0)
Call.assert "latdf: n == ipivDim0" (n == ipivDim0)
Call.assert "latdf: n == jpivDim0" (n == jpivDim0)
evalContT $ do
ijobPtr <- Call.cint ijob
nPtr <- Call.cint n
zPtr <- Call.array z
ldzPtr <- Call.cint ldz
rhsPtr <- Call.ioarray rhs
rdsumPtr <- Call.float rdsum
rdscalPtr <- Call.float rdscal
ipivPtr <- Call.array ipiv
jpivPtr <- Call.array jpiv
liftIO $ FFI.latdf ijobPtr nPtr zPtr ldzPtr rhsPtr rdsumPtr rdscalPtr ipivPtr jpivPtr
liftIO $ pure (,)
<*> peek rdsumPtr
<*> peek rdscalPtr
latps ::
Char ->
Char ->
Char ->
Char ->
Array ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt Float ->
IO (Float, Int)
latps uplo trans diag normin ap x cnorm = do
let apDim0 = Call.sizes1 $ Array.shape ap
let xDim0 = Call.sizes1 $ MutArray.shape x
let cnormDim0 = Call.sizes1 $ MutArray.shape cnorm
let n = xDim0
Call.assert "latps: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "latps: n == cnormDim0" (n == cnormDim0)
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
norminPtr <- Call.char normin
nPtr <- Call.cint n
apPtr <- Call.array ap
xPtr <- Call.ioarray x
scalePtr <- Call.alloca
cnormPtr <- Call.ioarray cnorm
infoPtr <- Call.alloca
liftIO $ FFI.latps uploPtr transPtr diagPtr norminPtr nPtr apPtr xPtr scalePtr cnormPtr infoPtr
liftIO $ pure (,)
<*> peek scalePtr
<*> fmap fromIntegral (peek infoPtr)
latrd ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array ZeroInt (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float))
latrd uplo nb a ldw = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
e <- Call.newArray1 (n-1)
tau <- Call.newArray1 (n-1)
w <- Call.newArray2 nb ldw
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nbPtr <- Call.cint nb
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ePtr <- Call.ioarray e
tauPtr <- Call.ioarray tau
wPtr <- Call.ioarray w
ldwPtr <- Call.cint ldw
liftIO $ FFI.latrd uploPtr nPtr nbPtr aPtr ldaPtr ePtr tauPtr wPtr ldwPtr
liftIO $ pure (,,)
<*> Call.freezeArray e
<*> Call.freezeArray tau
<*> Call.freezeArray w
latrs ::
Char ->
Char ->
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt Float ->
IO (Float, Int)
latrs uplo trans diag normin a x cnorm = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let xDim0 = Call.sizes1 $ MutArray.shape x
let cnormDim0 = Call.sizes1 $ MutArray.shape cnorm
let n = aDim0
let lda = aDim1
Call.assert "latrs: n == xDim0" (n == xDim0)
Call.assert "latrs: n == cnormDim0" (n == cnormDim0)
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
norminPtr <- Call.char normin
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
xPtr <- Call.ioarray x
scalePtr <- Call.alloca
cnormPtr <- Call.ioarray cnorm
infoPtr <- Call.alloca
liftIO $ FFI.latrs uploPtr transPtr diagPtr norminPtr nPtr aPtr ldaPtr xPtr scalePtr cnormPtr infoPtr
liftIO $ pure (,)
<*> peek scalePtr
<*> fmap fromIntegral (peek infoPtr)
latrz ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float))
latrz m l a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 m
work <- Call.newArray1 m
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
lPtr <- Call.cint l
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
liftIO $ FFI.latrz mPtr nPtr lPtr aPtr ldaPtr tauPtr workPtr
liftIO $ Call.freezeArray tau
lauu2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
lauu2 uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.lauu2 uploPtr nPtr aPtr ldaPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
lauum ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
lauum uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.lauum uploPtr nPtr aPtr ldaPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pbcon ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
IO (Float, Int)
pbcon uplo kd ab anorm = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.pbcon uploPtr nPtr kdPtr abPtr ldabPtr anormPtr rcondPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
pbequ ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Float, Float, Int)
pbequ uplo kd ab = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
s <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
sPtr <- Call.ioarray s
scondPtr <- Call.alloca
amaxPtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.pbequ uploPtr nPtr kdPtr abPtr ldabPtr sPtr scondPtr amaxPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> peek scondPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
pbrfs ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
pbrfs uplo kd ab afb b x = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let (afbDim0,afbDim1) = Call.sizes2 $ Array.shape afb
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = abDim0
let ldab = abDim1
let ldafb = afbDim1
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "pbrfs: n == afbDim0" (n == afbDim0)
Call.assert "pbrfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
nrhsPtr <- Call.cint nrhs
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
afbPtr <- Call.array afb
ldafbPtr <- Call.cint ldafb
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.pbrfs uploPtr nPtr kdPtr nrhsPtr abPtr ldabPtr afbPtr ldafbPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
pbstf ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pbstf uplo kd ab = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
infoPtr <- Call.alloca
liftIO $ FFI.pbstf uploPtr nPtr kdPtr abPtr ldabPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pbsv ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pbsv uplo kd ab b = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = abDim0
let ldab = abDim1
let nrhs = bDim0
let ldb = bDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
nrhsPtr <- Call.cint nrhs
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.pbsv uploPtr nPtr kdPtr nrhsPtr abPtr ldabPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pbsvx ::
Char ->
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Char ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Char, Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
pbsvx fact uplo kd ab afb equed s b ldx = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let (afbDim0,afbDim1) = Call.sizes2 $ MutArray.shape afb
let sDim0 = Call.sizes1 $ MutArray.shape s
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = abDim0
let ldab = abDim1
let ldafb = afbDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "pbsvx: n == afbDim0" (n == afbDim0)
Call.assert "pbsvx: n == sDim0" (n == sDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
nrhsPtr <- Call.cint nrhs
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
afbPtr <- Call.ioarray afb
ldafbPtr <- Call.cint ldafb
equedPtr <- Call.char equed
sPtr <- Call.ioarray s
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.pbsvx factPtr uploPtr nPtr kdPtr nrhsPtr abPtr ldabPtr afbPtr ldafbPtr equedPtr sPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> fmap castCCharToChar (peek equedPtr)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
pbtf2 ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pbtf2 uplo kd ab = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
infoPtr <- Call.alloca
liftIO $ FFI.pbtf2 uploPtr nPtr kdPtr abPtr ldabPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pbtrf ::
Char ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pbtrf uplo kd ab = do
let (abDim0,abDim1) = Call.sizes2 $ MutArray.shape ab
let n = abDim0
let ldab = abDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.ioarray ab
ldabPtr <- Call.cint ldab
infoPtr <- Call.alloca
liftIO $ FFI.pbtrf uploPtr nPtr kdPtr abPtr ldabPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pbtrs ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pbtrs uplo kd ab b = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = abDim0
let ldab = abDim1
let nrhs = bDim0
let ldb = bDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
kdPtr <- Call.cint kd
nrhsPtr <- Call.cint nrhs
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.pbtrs uploPtr nPtr kdPtr nrhsPtr abPtr ldabPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pftrf ::
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Int)
pftrf transr uplo n a = do
let aDim0 = Call.sizes1 $ MutArray.shape a
Call.assert "pftrf: n*(n+1)`div`2 == aDim0" (n*(n+1)`div`2 == aDim0)
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
infoPtr <- Call.alloca
liftIO $ FFI.pftrf transrPtr uploPtr nPtr aPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pftri ::
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Int)
pftri transr uplo n a = do
let aDim0 = Call.sizes1 $ MutArray.shape a
Call.assert "pftri: n*(n+1)`div`2 == aDim0" (n*(n+1)`div`2 == aDim0)
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
infoPtr <- Call.alloca
liftIO $ FFI.pftri transrPtr uploPtr nPtr aPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pftrs ::
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pftrs transr uplo n a b = do
let aDim0 = Call.sizes1 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let nrhs = bDim0
let ldb = bDim1
Call.assert "pftrs: n*(n+1)`div`2 == aDim0" (n*(n+1)`div`2 == aDim0)
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.pftrs transrPtr uploPtr nPtr nrhsPtr aPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pocon ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
IO (Float, Int)
pocon uplo a anorm = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.pocon uploPtr nPtr aPtr ldaPtr anormPtr rcondPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
poequ ::
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Float, Float, Int)
poequ a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
s <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
sPtr <- Call.ioarray s
scondPtr <- Call.alloca
amaxPtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.poequ nPtr aPtr ldaPtr sPtr scondPtr amaxPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> peek scondPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
poequb ::
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Float, Float, Int)
poequb a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
s <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
sPtr <- Call.ioarray s
scondPtr <- Call.alloca
amaxPtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.poequb nPtr aPtr ldaPtr sPtr scondPtr amaxPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> peek scondPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
porfs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
porfs uplo a af b x = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (afDim0,afDim1) = Call.sizes2 $ Array.shape af
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = aDim0
let lda = aDim1
let ldaf = afDim1
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "porfs: n == afDim0" (n == afDim0)
Call.assert "porfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
afPtr <- Call.array af
ldafPtr <- Call.cint ldaf
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.porfs uploPtr nPtr nrhsPtr aPtr ldaPtr afPtr ldafPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
posv ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
posv uplo a b = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.posv uploPtr nPtr nrhsPtr aPtr ldaPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
posvx ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Char ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Char, Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
posvx fact uplo a af equed s b ldx = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (afDim0,afDim1) = Call.sizes2 $ MutArray.shape af
let sDim0 = Call.sizes1 $ MutArray.shape s
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let ldaf = afDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "posvx: n == afDim0" (n == afDim0)
Call.assert "posvx: n == sDim0" (n == sDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
afPtr <- Call.ioarray af
ldafPtr <- Call.cint ldaf
equedPtr <- Call.char equed
sPtr <- Call.ioarray s
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.posvx factPtr uploPtr nPtr nrhsPtr aPtr ldaPtr afPtr ldafPtr equedPtr sPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> fmap castCCharToChar (peek equedPtr)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
potf2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
potf2 uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.potf2 uploPtr nPtr aPtr ldaPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
potrf ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
potrf uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.potrf uploPtr nPtr aPtr ldaPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
potri ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
potri uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.potri uploPtr nPtr aPtr ldaPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
potrs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
potrs uplo a b = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.potrs uploPtr nPtr nrhsPtr aPtr ldaPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ppcon ::
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Float ->
IO (Float, Int)
ppcon uplo n ap anorm = do
let apDim0 = Call.sizes1 $ Array.shape ap
Call.assert "ppcon: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.ppcon uploPtr nPtr apPtr anormPtr rcondPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
ppequ ::
Char ->
Int ->
Array ZeroInt (Complex Float) ->
IO (Array ZeroInt Float, Float, Float, Int)
ppequ uplo n ap = do
let apDim0 = Call.sizes1 $ Array.shape ap
Call.assert "ppequ: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
s <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
sPtr <- Call.ioarray s
scondPtr <- Call.alloca
amaxPtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.ppequ uploPtr nPtr apPtr sPtr scondPtr amaxPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> peek scondPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
pprfs ::
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
pprfs uplo n ap afp b x = do
let apDim0 = Call.sizes1 $ Array.shape ap
let afpDim0 = Call.sizes1 $ Array.shape afp
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "pprfs: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "pprfs: n*(n+1)`div`2 == afpDim0" (n*(n+1)`div`2 == afpDim0)
Call.assert "pprfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
afpPtr <- Call.array afp
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.pprfs uploPtr nPtr nrhsPtr apPtr afpPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
ppsv ::
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
ppsv uplo n ap b = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let nrhs = bDim0
let ldb = bDim1
Call.assert "ppsv: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.ioarray ap
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.ppsv uploPtr nPtr nrhsPtr apPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ppsvx ::
Char ->
Char ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
Char ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Char, Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
ppsvx fact uplo ap afp equed s b ldx = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let afpDim0 = Call.sizes1 $ MutArray.shape afp
let sDim0 = Call.sizes1 $ MutArray.shape s
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = sDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "ppsvx: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "ppsvx: n*(n+1)`div`2 == afpDim0" (n*(n+1)`div`2 == afpDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.ioarray ap
afpPtr <- Call.ioarray afp
equedPtr <- Call.char equed
sPtr <- Call.ioarray s
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.ppsvx factPtr uploPtr nPtr nrhsPtr apPtr afpPtr equedPtr sPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> fmap castCCharToChar (peek equedPtr)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
pptrf ::
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Int)
pptrf uplo n ap = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "pptrf: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
infoPtr <- Call.alloca
liftIO $ FFI.pptrf uploPtr nPtr apPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pptri ::
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Int)
pptri uplo n ap = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "pptri: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
infoPtr <- Call.alloca
liftIO $ FFI.pptri uploPtr nPtr apPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pptrs ::
Char ->
Int ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pptrs uplo n ap b = do
let apDim0 = Call.sizes1 $ Array.shape ap
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let nrhs = bDim0
let ldb = bDim1
Call.assert "pptrs: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.pptrs uploPtr nPtr nrhsPtr apPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pstf2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
IO (Array ZeroInt CInt, Int, Int)
pstf2 uplo a tol = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
piv <- Call.newArray1 n
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
pivPtr <- Call.ioarray piv
rankPtr <- Call.alloca
tolPtr <- Call.float tol
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.pstf2 uploPtr nPtr aPtr ldaPtr pivPtr rankPtr tolPtr workPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray piv
<*> fmap fromIntegral (peek rankPtr)
<*> fmap fromIntegral (peek infoPtr)
pstrf ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
IO (Array ZeroInt CInt, Int, Int)
pstrf uplo a tol = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
piv <- Call.newArray1 n
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
pivPtr <- Call.ioarray piv
rankPtr <- Call.alloca
tolPtr <- Call.float tol
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.pstrf uploPtr nPtr aPtr ldaPtr pivPtr rankPtr tolPtr workPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray piv
<*> fmap fromIntegral (peek rankPtr)
<*> fmap fromIntegral (peek infoPtr)
ptcon ::
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
Float ->
IO (Float, Int)
ptcon d e anorm = do
let dDim0 = Call.sizes1 $ Array.shape d
let eDim0 = Call.sizes1 $ Array.shape e
let n = dDim0
Call.assert "ptcon: n-1 == eDim0" (n-1 == eDim0)
rwork <- Call.newArray1 n
evalContT $ do
nPtr <- Call.cint n
dPtr <- Call.array d
ePtr <- Call.array e
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.ptcon nPtr dPtr ePtr anormPtr rcondPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
pteqr ::
Char ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pteqr compz d e z = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = dDim0
let ldz = zDim1
Call.assert "pteqr: n-1 == eDim0" (n-1 == eDim0)
Call.assert "pteqr: n == zDim0" (n == zDim0)
work <- Call.newArray1 (4*n)
evalContT $ do
compzPtr <- Call.char compz
nPtr <- Call.cint n
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.pteqr compzPtr nPtr dPtr ePtr zPtr ldzPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ptrfs ::
Char ->
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
ptrfs uplo d e df ef b x = do
let dDim0 = Call.sizes1 $ Array.shape d
let eDim0 = Call.sizes1 $ Array.shape e
let dfDim0 = Call.sizes1 $ Array.shape df
let efDim0 = Call.sizes1 $ Array.shape ef
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "ptrfs: n-1 == eDim0" (n-1 == eDim0)
Call.assert "ptrfs: n == dfDim0" (n == dfDim0)
Call.assert "ptrfs: n-1 == efDim0" (n-1 == efDim0)
Call.assert "ptrfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 n
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dPtr <- Call.array d
ePtr <- Call.array e
dfPtr <- Call.array df
efPtr <- Call.array ef
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.ptrfs uploPtr nPtr nrhsPtr dPtr ePtr dfPtr efPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
ptsv ::
IOArray ZeroInt Float ->
IOArray ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
ptsv d e b = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "ptsv: n-1 == eDim0" (n-1 == eDim0)
evalContT $ do
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.ptsv nPtr nrhsPtr dPtr ePtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ptsvx ::
Char ->
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
IOArray ZeroInt Float ->
IOArray ZeroInt (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
ptsvx fact d e df ef b ldx = do
let dDim0 = Call.sizes1 $ Array.shape d
let eDim0 = Call.sizes1 $ Array.shape e
let dfDim0 = Call.sizes1 $ MutArray.shape df
let efDim0 = Call.sizes1 $ MutArray.shape ef
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "ptsvx: n-1 == eDim0" (n-1 == eDim0)
Call.assert "ptsvx: n == dfDim0" (n == dfDim0)
Call.assert "ptsvx: n-1 == efDim0" (n-1 == efDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 n
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dPtr <- Call.array d
ePtr <- Call.array e
dfPtr <- Call.ioarray df
efPtr <- Call.ioarray ef
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.ptsvx factPtr nPtr nrhsPtr dPtr ePtr dfPtr efPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
pttrf ::
IOArray ZeroInt Float ->
IOArray ZeroInt (Complex Float) ->
IO (Int)
pttrf d e = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let n = dDim0
Call.assert "pttrf: n-1 == eDim0" (n-1 == eDim0)
evalContT $ do
nPtr <- Call.cint n
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
infoPtr <- Call.alloca
liftIO $ FFI.pttrf nPtr dPtr ePtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
pttrs ::
Char ->
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
pttrs uplo d e b = do
let dDim0 = Call.sizes1 $ Array.shape d
let eDim0 = Call.sizes1 $ Array.shape e
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "pttrs: n-1 == eDim0" (n-1 == eDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dPtr <- Call.array d
ePtr <- Call.array e
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.pttrs uploPtr nPtr nrhsPtr dPtr ePtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ptts2 ::
Int ->
Array ZeroInt Float ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO ()
ptts2 iuplo d e b = do
let dDim0 = Call.sizes1 $ Array.shape d
let eDim0 = Call.sizes1 $ Array.shape e
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = dDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "ptts2: n-1 == eDim0" (n-1 == eDim0)
evalContT $ do
iuploPtr <- Call.cint iuplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
dPtr <- Call.array d
ePtr <- Call.array e
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
liftIO $ FFI.ptts2 iuploPtr nPtr nrhsPtr dPtr ePtr bPtr ldbPtr
rot ::
IOArray ZeroInt (Complex Float) ->
Int ->
IOArray ZeroInt (Complex Float) ->
Int ->
Float ->
Complex Float ->
IO ()
rot cx incx cy incy c s = do
let cxDim0 = Call.sizes1 $ MutArray.shape cx
let cyDim0 = Call.sizes1 $ MutArray.shape cy
let n = cxDim0
Call.assert "rot: n == cyDim0" (n == cyDim0)
evalContT $ do
nPtr <- Call.cint n
cxPtr <- Call.ioarray cx
incxPtr <- Call.cint incx
cyPtr <- Call.ioarray cy
incyPtr <- Call.cint incy
cPtr <- Call.float c
sPtr <- Call.complexFloat s
liftIO $ FFI.rot nPtr cxPtr incxPtr cyPtr incyPtr cPtr sPtr
rscl ::
Int ->
Float ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO ()
rscl n sa sx incx = do
let sxDim0 = Call.sizes1 $ MutArray.shape sx
let _sxSize = sxDim0
evalContT $ do
nPtr <- Call.cint n
saPtr <- Call.float sa
sxPtr <- Call.ioarray sx
incxPtr <- Call.cint incx
liftIO $ FFI.rscl nPtr saPtr sxPtr incxPtr
spcon ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
Float ->
IO (Float, Int)
spcon uplo ap ipiv anorm = do
let apDim0 = Call.sizes1 $ Array.shape ap
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = ipivDim0
Call.assert "spcon: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
ipivPtr <- Call.array ipiv
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.spcon uploPtr nPtr apPtr ipivPtr anormPtr rcondPtr workPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
spmv ::
Char ->
Int ->
Complex Float ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
Complex Float ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO ()
spmv uplo n alpha ap x incx beta y incy = do
let apDim0 = Call.sizes1 $ Array.shape ap
let xDim0 = Call.sizes1 $ Array.shape x
let yDim0 = Call.sizes1 $ MutArray.shape y
let _apSize = apDim0
let _xSize = xDim0
let _ySize = yDim0
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
alphaPtr <- Call.complexFloat alpha
apPtr <- Call.array ap
xPtr <- Call.array x
incxPtr <- Call.cint incx
betaPtr <- Call.complexFloat beta
yPtr <- Call.ioarray y
incyPtr <- Call.cint incy
liftIO $ FFI.spmv uploPtr nPtr alphaPtr apPtr xPtr incxPtr betaPtr yPtr incyPtr
spr ::
Char ->
Int ->
Complex Float ->
Array ZeroInt (Complex Float) ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO ()
spr uplo n alpha x incx ap = do
let xDim0 = Call.sizes1 $ Array.shape x
let apDim0 = Call.sizes1 $ MutArray.shape ap
let _xSize = xDim0
let _apSize = apDim0
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
alphaPtr <- Call.complexFloat alpha
xPtr <- Call.array x
incxPtr <- Call.cint incx
apPtr <- Call.ioarray ap
liftIO $ FFI.spr uploPtr nPtr alphaPtr xPtr incxPtr apPtr
sprfs ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
sprfs uplo ap afp ipiv b x = do
let apDim0 = Call.sizes1 $ Array.shape ap
let afpDim0 = Call.sizes1 $ Array.shape afp
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = ipivDim0
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "sprfs: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "sprfs: n*(n+1)`div`2 == afpDim0" (n*(n+1)`div`2 == afpDim0)
Call.assert "sprfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
afpPtr <- Call.array afp
ipivPtr <- Call.array ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.sprfs uploPtr nPtr nrhsPtr apPtr afpPtr ipivPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
spsv ::
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
spsv uplo n ap b = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let nrhs = bDim0
let ldb = bDim1
Call.assert "spsv: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
ipiv <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.ioarray ap
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.spsv uploPtr nPtr nrhsPtr apPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
spsvx ::
Char ->
Char ->
Array ZeroInt (Complex Float) ->
IOArray ZeroInt (Complex Float) ->
IOArray ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
spsvx fact uplo ap afp ipiv b ldx = do
let apDim0 = Call.sizes1 $ Array.shape ap
let afpDim0 = Call.sizes1 $ MutArray.shape afp
let ipivDim0 = Call.sizes1 $ MutArray.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let n = ipivDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "spsvx: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "spsvx: n*(n+1)`div`2 == afpDim0" (n*(n+1)`div`2 == afpDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
afpPtr <- Call.ioarray afp
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.spsvx factPtr uploPtr nPtr nrhsPtr apPtr afpPtr ipivPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
sptrf ::
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Array ZeroInt CInt, Int)
sptrf uplo n ap = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "sptrf: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
ipiv <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.sptrf uploPtr nPtr apPtr ipivPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
sptri ::
Char ->
IOArray ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
IO (Int)
sptri uplo ap ipiv = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = ipivDim0
Call.assert "sptri: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.sptri uploPtr nPtr apPtr ipivPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
sptrs ::
Char ->
Array ZeroInt (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
sptrs uplo ap ipiv b = do
let apDim0 = Call.sizes1 $ Array.shape ap
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = ipivDim0
let nrhs = bDim0
let ldb = bDim1
Call.assert "sptrs: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.sptrs uploPtr nPtr nrhsPtr apPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
stedc ::
Char ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
IO (Int)
stedc compz d e z lwork lrwork liwork = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = dDim0
let ldz = zDim1
Call.assert "stedc: n-1 == eDim0" (n-1 == eDim0)
Call.assert "stedc: n == zDim0" (n == zDim0)
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 (maximum[1,lrwork])
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
compzPtr <- Call.char compz
nPtr <- Call.cint n
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.stedc compzPtr nPtr dPtr ePtr zPtr ldzPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr liworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
stegr ::
Char ->
Char ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
Float ->
Float ->
Int ->
Int ->
Float ->
Int ->
Int ->
Int ->
Int ->
IO (Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
stegr jobz range d e vl vu il iu abstol m ldz lwork liwork = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let n = dDim0
Call.assert "stegr: n == eDim0" (n == eDim0)
w <- Call.newArray1 n
z <- Call.newArray2 (maximum[1,m]) ldz
isuppz <- Call.newArray1 (2*maximum[1,m])
work <- Call.newArray1 lwork
iwork <- Call.newArray1 liwork
evalContT $ do
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
nPtr <- Call.cint n
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
abstolPtr <- Call.float abstol
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
isuppzPtr <- Call.ioarray isuppz
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.stegr jobzPtr rangePtr nPtr dPtr ePtr vlPtr vuPtr ilPtr iuPtr abstolPtr mPtr wPtr zPtr ldzPtr isuppzPtr workPtr lworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,,,,)
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray isuppz
<*> fmap fromIntegral (peek infoPtr)
stein ::
Array ZeroInt Float ->
Array ZeroInt Float ->
Int ->
Array ZeroInt Float ->
Array ZeroInt CInt ->
Array ZeroInt CInt ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Int)
stein d e m w iblock isplit ldz = do
let dDim0 = Call.sizes1 $ Array.shape d
let eDim0 = Call.sizes1 $ Array.shape e
let wDim0 = Call.sizes1 $ Array.shape w
let iblockDim0 = Call.sizes1 $ Array.shape iblock
let isplitDim0 = Call.sizes1 $ Array.shape isplit
let n = dDim0
Call.assert "stein: n-1 == eDim0" (n-1 == eDim0)
Call.assert "stein: n == wDim0" (n == wDim0)
Call.assert "stein: n == iblockDim0" (n == iblockDim0)
Call.assert "stein: n == isplitDim0" (n == isplitDim0)
z <- Call.newArray2 m ldz
work <- Call.newArray1 (5*n)
iwork <- Call.newArray1 n
ifail <- Call.newArray1 m
evalContT $ do
nPtr <- Call.cint n
dPtr <- Call.array d
ePtr <- Call.array e
mPtr <- Call.cint m
wPtr <- Call.array w
iblockPtr <- Call.array iblock
isplitPtr <- Call.array isplit
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
iworkPtr <- Call.ioarray iwork
ifailPtr <- Call.ioarray ifail
infoPtr <- Call.alloca
liftIO $ FFI.stein nPtr dPtr ePtr mPtr wPtr iblockPtr isplitPtr zPtr ldzPtr workPtr iworkPtr ifailPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray z
<*> Call.freezeArray ifail
<*> fmap fromIntegral (peek infoPtr)
stemr ::
Char ->
Char ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
Float ->
Float ->
Int ->
Int ->
Int ->
Int ->
Int ->
Bool ->
Int ->
Int ->
IO (Int, Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array ZeroInt CInt, Bool, Int)
stemr jobz range d e vl vu il iu m ldz nzc tryrac lwork liwork = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let n = dDim0
Call.assert "stemr: n == eDim0" (n == eDim0)
w <- Call.newArray1 n
z <- Call.newArray2 (maximum[1,m]) ldz
isuppz <- Call.newArray1 (2*maximum[1,m])
work <- Call.newArray1 lwork
iwork <- Call.newArray1 liwork
evalContT $ do
jobzPtr <- Call.char jobz
rangePtr <- Call.char range
nPtr <- Call.cint n
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
vlPtr <- Call.float vl
vuPtr <- Call.float vu
ilPtr <- Call.cint il
iuPtr <- Call.cint iu
mPtr <- Call.alloca
wPtr <- Call.ioarray w
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
nzcPtr <- Call.cint nzc
isuppzPtr <- Call.ioarray isuppz
tryracPtr <- Call.bool tryrac
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.stemr jobzPtr rangePtr nPtr dPtr ePtr vlPtr vuPtr ilPtr iuPtr mPtr wPtr zPtr ldzPtr nzcPtr isuppzPtr tryracPtr workPtr lworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> fmap fromIntegral (peek mPtr)
<*> Call.freezeArray w
<*> Call.freezeArray z
<*> Call.freezeArray isuppz
<*> peek tryracPtr
<*> fmap fromIntegral (peek infoPtr)
steqr ::
Char ->
IOArray ZeroInt Float ->
IOArray ZeroInt Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
steqr compz d e z = do
let dDim0 = Call.sizes1 $ MutArray.shape d
let eDim0 = Call.sizes1 $ MutArray.shape e
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = dDim0
let ldz = zDim1
Call.assert "steqr: n-1 == eDim0" (n-1 == eDim0)
Call.assert "steqr: n == zDim0" (n == zDim0)
work <- Call.newArray1 (maximum[1,2*n-2])
evalContT $ do
compzPtr <- Call.char compz
nPtr <- Call.cint n
dPtr <- Call.ioarray d
ePtr <- Call.ioarray e
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.steqr compzPtr nPtr dPtr ePtr zPtr ldzPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
sum1 ::
Array ZeroInt (Complex Float) ->
Int ->
IO Float
sum1 cx incx = do
let cxDim0 = Call.sizes1 $ Array.shape cx
let n = cxDim0
evalContT $ do
nPtr <- Call.cint n
cxPtr <- Call.array cx
incxPtr <- Call.cint incx
liftIO $ FFI.sum1 nPtr cxPtr incxPtr
sycon ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Float ->
IO (Float, Int)
sycon uplo a ipiv anorm = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "sycon: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
anormPtr <- Call.float anorm
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.sycon uploPtr nPtr aPtr ldaPtr ipivPtr anormPtr rcondPtr workPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
syconv ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IO (Array ZeroInt (Complex Float), Int)
syconv uplo way a ipiv = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "syconv: n == ipivDim0" (n == ipivDim0)
e <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
wayPtr <- Call.char way
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
ePtr <- Call.ioarray e
infoPtr <- Call.alloca
liftIO $ FFI.syconv uploPtr wayPtr nPtr aPtr ldaPtr ipivPtr ePtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray e
<*> fmap fromIntegral (peek infoPtr)
syequb ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Float, Float, Int)
syequb uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
s <- Call.newArray1 n
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
sPtr <- Call.ioarray s
scondPtr <- Call.alloca
amaxPtr <- Call.alloca
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.syequb uploPtr nPtr aPtr ldaPtr sPtr scondPtr amaxPtr workPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> peek scondPtr
<*> peek amaxPtr
<*> fmap fromIntegral (peek infoPtr)
symv ::
Char ->
Complex Float ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
Complex Float ->
IOArray ZeroInt (Complex Float) ->
Int ->
IO ()
symv uplo alpha a x incx beta y incy = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let xDim0 = Call.sizes1 $ Array.shape x
let yDim0 = Call.sizes1 $ MutArray.shape y
let n = aDim0
let lda = aDim1
let _xSize = xDim0
let _ySize = yDim0
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
alphaPtr <- Call.complexFloat alpha
aPtr <- Call.array a
ldaPtr <- Call.cint lda
xPtr <- Call.array x
incxPtr <- Call.cint incx
betaPtr <- Call.complexFloat beta
yPtr <- Call.ioarray y
incyPtr <- Call.cint incy
liftIO $ FFI.symv uploPtr nPtr alphaPtr aPtr ldaPtr xPtr incxPtr betaPtr yPtr incyPtr
syr ::
Char ->
Complex Float ->
Array ZeroInt (Complex Float) ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO ()
syr uplo alpha x incx a = do
let xDim0 = Call.sizes1 $ Array.shape x
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let _xSize = xDim0
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
alphaPtr <- Call.complexFloat alpha
xPtr <- Call.array x
incxPtr <- Call.cint incx
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
liftIO $ FFI.syr uploPtr nPtr alphaPtr xPtr incxPtr aPtr ldaPtr
syrfs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
syrfs uplo a af ipiv b x = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (afDim0,afDim1) = Call.sizes2 $ Array.shape af
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ MutArray.shape x
let n = aDim0
let lda = aDim1
let ldaf = afDim1
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "syrfs: n == afDim0" (n == afDim0)
Call.assert "syrfs: n == ipivDim0" (n == ipivDim0)
Call.assert "syrfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
afPtr <- Call.array af
ldafPtr <- Call.cint ldaf
ipivPtr <- Call.array ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.syrfs uploPtr nPtr nrhsPtr aPtr ldaPtr afPtr ldafPtr ipivPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
sysv ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt CInt, Int)
sysv uplo a b lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
ipiv <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.sysv uploPtr nPtr nrhsPtr aPtr ldaPtr ipivPtr bPtr ldbPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
sysvx ::
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray ZeroInt CInt ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Float, Array ZeroInt Float, Array ZeroInt Float, Int)
sysvx fact uplo a af ipiv b ldx lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (afDim0,afDim1) = Call.sizes2 $ MutArray.shape af
let ipivDim0 = Call.sizes1 $ MutArray.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let n = aDim0
let lda = aDim1
let ldaf = afDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "sysvx: n == afDim0" (n == afDim0)
Call.assert "sysvx: n == ipivDim0" (n == ipivDim0)
x <- Call.newArray2 nrhs ldx
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 n
evalContT $ do
factPtr <- Call.char fact
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
afPtr <- Call.ioarray af
ldafPtr <- Call.cint ldaf
ipivPtr <- Call.ioarray ipiv
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.ioarray x
ldxPtr <- Call.cint ldx
rcondPtr <- Call.alloca
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.sysvx factPtr uploPtr nPtr nrhsPtr aPtr ldaPtr afPtr ldafPtr ipivPtr bPtr ldbPtr xPtr ldxPtr rcondPtr ferrPtr berrPtr workPtr lworkPtr rworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray x
<*> peek rcondPtr
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
syswapr ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO ()
syswapr uplo a i1 i2 = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
i1Ptr <- Call.cint i1
i2Ptr <- Call.cint i2
liftIO $ FFI.syswapr uploPtr nPtr aPtr ldaPtr i1Ptr i2Ptr
sytf2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt CInt, Int)
sytf2 uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
infoPtr <- Call.alloca
liftIO $ FFI.sytf2 uploPtr nPtr aPtr ldaPtr ipivPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
sytrf ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt CInt, Int)
sytrf uplo a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
ipiv <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.ioarray ipiv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.sytrf uploPtr nPtr aPtr ldaPtr ipivPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ipiv
<*> fmap fromIntegral (peek infoPtr)
sytri ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IO (Int)
sytri uplo a ipiv = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "sytri: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 (2*n)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.sytri uploPtr nPtr aPtr ldaPtr ipivPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
sytri2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Int ->
Int ->
IO (Int)
sytri2 uplo a ipiv nb lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "sytri2: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 (n+nb+1)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.sytri2 uploPtr nPtr aPtr ldaPtr ipivPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
sytri2x ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
Int ->
IO (Int)
sytri2x uplo a ipiv nb = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let n = aDim0
let lda = aDim1
Call.assert "sytri2x: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray2 (nb+3) (n+nb+1)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
workPtr <- Call.ioarray work
nbPtr <- Call.cint nb
infoPtr <- Call.alloca
liftIO $ FFI.sytri2x uploPtr nPtr aPtr ldaPtr ipivPtr workPtr nbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
sytrs ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
sytrs uplo a ipiv b = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "sytrs: n == ipivDim0" (n == ipivDim0)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.sytrs uploPtr nPtr nrhsPtr aPtr ldaPtr ipivPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
sytrs2 ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt CInt ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
sytrs2 uplo a ipiv b = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let ipivDim0 = Call.sizes1 $ Array.shape ipiv
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
Call.assert "sytrs2: n == ipivDim0" (n == ipivDim0)
work <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
ipivPtr <- Call.array ipiv
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.sytrs2 uploPtr nPtr nrhsPtr aPtr ldaPtr ipivPtr bPtr ldbPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
tbcon ::
Char ->
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Float, Int)
tbcon norm uplo diag kd ab = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let n = abDim0
let ldab = abDim1
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
kdPtr <- Call.cint kd
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.tbcon normPtr uploPtr diagPtr nPtr kdPtr abPtr ldabPtr rcondPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
tbrfs ::
Char ->
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
tbrfs uplo trans diag kd ab b x = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ Array.shape x
let n = abDim0
let ldab = abDim1
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "tbrfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
nPtr <- Call.cint n
kdPtr <- Call.cint kd
nrhsPtr <- Call.cint nrhs
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.array x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.tbrfs uploPtr transPtr diagPtr nPtr kdPtr nrhsPtr abPtr ldabPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
tbtrs ::
Char ->
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
tbtrs uplo trans diag kd ab b = do
let (abDim0,abDim1) = Call.sizes2 $ Array.shape ab
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = abDim0
let ldab = abDim1
let nrhs = bDim0
let ldb = bDim1
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
nPtr <- Call.cint n
kdPtr <- Call.cint kd
nrhsPtr <- Call.cint nrhs
abPtr <- Call.array ab
ldabPtr <- Call.cint ldab
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.tbtrs uploPtr transPtr diagPtr nPtr kdPtr nrhsPtr abPtr ldabPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
tfsm ::
Char ->
Char ->
Char ->
Char ->
Char ->
Int ->
Complex Float ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO ()
tfsm transr side uplo trans diag m alpha a b = do
let aDim0 = Call.sizes1 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = bDim0
let ldb = bDim1
Call.assert "tfsm: n*(n+1)`div`2 == aDim0" (n*(n+1)`div`2 == aDim0)
evalContT $ do
transrPtr <- Call.char transr
sidePtr <- Call.char side
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
mPtr <- Call.cint m
nPtr <- Call.cint n
alphaPtr <- Call.complexFloat alpha
aPtr <- Call.array a
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
liftIO $ FFI.tfsm transrPtr sidePtr uploPtr transPtr diagPtr mPtr nPtr alphaPtr aPtr bPtr ldbPtr
tftri ::
Char ->
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Int)
tftri transr uplo diag n a = do
let aDim0 = Call.sizes1 $ MutArray.shape a
Call.assert "tftri: n*(n+1)`div`2 == aDim0" (n*(n+1)`div`2 == aDim0)
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
aPtr <- Call.ioarray a
infoPtr <- Call.alloca
liftIO $ FFI.tftri transrPtr uploPtr diagPtr nPtr aPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
tfttp ::
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
tfttp transr uplo n arf = do
let arfDim0 = Call.sizes1 $ Array.shape arf
Call.assert "tfttp: n*(n+1)`div`2 == arfDim0" (n*(n+1)`div`2 == arfDim0)
ap <- Call.newArray1 (n*(n+1)`div`2)
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
nPtr <- Call.cint n
arfPtr <- Call.array arf
apPtr <- Call.ioarray ap
infoPtr <- Call.alloca
liftIO $ FFI.tfttp transrPtr uploPtr nPtr arfPtr apPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ap
<*> fmap fromIntegral (peek infoPtr)
tfttr ::
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int)
tfttr transr uplo n arf lda = do
let arfDim0 = Call.sizes1 $ Array.shape arf
Call.assert "tfttr: n*(n+1)`div`2 == arfDim0" (n*(n+1)`div`2 == arfDim0)
a <- Call.newArray2 n lda
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
nPtr <- Call.cint n
arfPtr <- Call.array arf
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.tfttr transrPtr uploPtr nPtr arfPtr aPtr ldaPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray a
<*> fmap fromIntegral (peek infoPtr)
tgevc ::
Char ->
Char ->
Array ZeroInt Bool ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int, Int)
tgevc side howmny select s p vl vr = do
let selectDim0 = Call.sizes1 $ Array.shape select
let (sDim0,sDim1) = Call.sizes2 $ Array.shape s
let (pDim0,pDim1) = Call.sizes2 $ Array.shape p
let (vlDim0,vlDim1) = Call.sizes2 $ MutArray.shape vl
let (vrDim0,vrDim1) = Call.sizes2 $ MutArray.shape vr
let n = selectDim0
let lds = sDim1
let ldp = pDim1
let mm = vlDim0
let ldvl = vlDim1
let ldvr = vrDim1
Call.assert "tgevc: n == sDim0" (n == sDim0)
Call.assert "tgevc: n == pDim0" (n == pDim0)
Call.assert "tgevc: mm == vrDim0" (mm == vrDim0)
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 (2*n)
evalContT $ do
sidePtr <- Call.char side
howmnyPtr <- Call.char howmny
selectPtr <- Call.array select
nPtr <- Call.cint n
sPtr <- Call.array s
ldsPtr <- Call.cint lds
pPtr <- Call.array p
ldpPtr <- Call.cint ldp
vlPtr <- Call.ioarray vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.ioarray vr
ldvrPtr <- Call.cint ldvr
mmPtr <- Call.cint mm
mPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.tgevc sidePtr howmnyPtr selectPtr nPtr sPtr ldsPtr pPtr ldpPtr vlPtr ldvlPtr vrPtr ldvrPtr mmPtr mPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> fmap fromIntegral (peek mPtr)
<*> fmap fromIntegral (peek infoPtr)
tgex2 ::
Bool ->
Bool ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
tgex2 wantq wantz a b q z j1 = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = aDim0
let lda = aDim1
let ldb = bDim1
let ldq = qDim1
let ldz = zDim1
Call.assert "tgex2: n == bDim0" (n == bDim0)
Call.assert "tgex2: n == qDim0" (n == qDim0)
Call.assert "tgex2: n == zDim0" (n == zDim0)
evalContT $ do
wantqPtr <- Call.bool wantq
wantzPtr <- Call.bool wantz
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
j1Ptr <- Call.cint j1
infoPtr <- Call.alloca
liftIO $ FFI.tgex2 wantqPtr wantzPtr nPtr aPtr ldaPtr bPtr ldbPtr qPtr ldqPtr zPtr ldzPtr j1Ptr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
tgexc ::
Bool ->
Bool ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Int, Int)
tgexc wantq wantz a b q z ifst ilst = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = aDim0
let lda = aDim1
let ldb = bDim1
let ldq = qDim1
let ldz = zDim1
Call.assert "tgexc: n == bDim0" (n == bDim0)
Call.assert "tgexc: n == qDim0" (n == qDim0)
Call.assert "tgexc: n == zDim0" (n == zDim0)
evalContT $ do
wantqPtr <- Call.bool wantq
wantzPtr <- Call.bool wantz
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
ifstPtr <- Call.cint ifst
ilstPtr <- Call.cint ilst
infoPtr <- Call.alloca
liftIO $ FFI.tgexc wantqPtr wantzPtr nPtr aPtr ldaPtr bPtr ldbPtr qPtr ldqPtr zPtr ldzPtr ifstPtr ilstPtr infoPtr
liftIO $ pure (,)
<*> fmap fromIntegral (peek ilstPtr)
<*> fmap fromIntegral (peek infoPtr)
tgsen ::
Int ->
Bool ->
Bool ->
Array ZeroInt Bool ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Int, Float, Float, Array ZeroInt Float, Int)
tgsen ijob wantq wantz select a b q z lwork liwork = do
let selectDim0 = Call.sizes1 $ Array.shape select
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let (zDim0,zDim1) = Call.sizes2 $ MutArray.shape z
let n = selectDim0
let lda = aDim1
let ldb = bDim1
let ldq = qDim1
let ldz = zDim1
Call.assert "tgsen: n == aDim0" (n == aDim0)
Call.assert "tgsen: n == bDim0" (n == bDim0)
Call.assert "tgsen: n == qDim0" (n == qDim0)
Call.assert "tgsen: n == zDim0" (n == zDim0)
alpha <- Call.newArray1 n
beta <- Call.newArray1 n
dif <- Call.newArray1 2
work <- Call.newArray1 (maximum[1,lwork])
iwork <- Call.newArray1 (maximum[1,liwork])
evalContT $ do
ijobPtr <- Call.cint ijob
wantqPtr <- Call.bool wantq
wantzPtr <- Call.bool wantz
selectPtr <- Call.array select
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
alphaPtr <- Call.ioarray alpha
betaPtr <- Call.ioarray beta
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
zPtr <- Call.ioarray z
ldzPtr <- Call.cint ldz
mPtr <- Call.alloca
plPtr <- Call.alloca
prPtr <- Call.alloca
difPtr <- Call.ioarray dif
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
iworkPtr <- Call.ioarray iwork
liworkPtr <- Call.cint liwork
infoPtr <- Call.alloca
liftIO $ FFI.tgsen ijobPtr wantqPtr wantzPtr selectPtr nPtr aPtr ldaPtr bPtr ldbPtr alphaPtr betaPtr qPtr ldqPtr zPtr ldzPtr mPtr plPtr prPtr difPtr workPtr lworkPtr iworkPtr liworkPtr infoPtr
liftIO $ pure (,,,,,,)
<*> Call.freezeArray alpha
<*> Call.freezeArray beta
<*> fmap fromIntegral (peek mPtr)
<*> peek plPtr
<*> peek prPtr
<*> Call.freezeArray dif
<*> fmap fromIntegral (peek infoPtr)
tgsja ::
Char ->
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Float ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int, Int)
tgsja jobu jobv jobq k l a b tola tolb u v q = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let (uDim0,uDim1) = Call.sizes2 $ MutArray.shape u
let (vDim0,vDim1) = Call.sizes2 $ MutArray.shape v
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let n = aDim0
let lda = aDim1
let ldb = bDim1
let m = uDim0
let ldu = uDim1
let p = vDim0
let ldv = vDim1
let ldq = qDim1
Call.assert "tgsja: n == bDim0" (n == bDim0)
Call.assert "tgsja: n == qDim0" (n == qDim0)
alpha <- Call.newArray1 n
beta <- Call.newArray1 n
work <- Call.newArray1 (2*n)
evalContT $ do
jobuPtr <- Call.char jobu
jobvPtr <- Call.char jobv
jobqPtr <- Call.char jobq
mPtr <- Call.cint m
pPtr <- Call.cint p
nPtr <- Call.cint n
kPtr <- Call.cint k
lPtr <- Call.cint l
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
tolaPtr <- Call.float tola
tolbPtr <- Call.float tolb
alphaPtr <- Call.ioarray alpha
betaPtr <- Call.ioarray beta
uPtr <- Call.ioarray u
lduPtr <- Call.cint ldu
vPtr <- Call.ioarray v
ldvPtr <- Call.cint ldv
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
workPtr <- Call.ioarray work
ncyclePtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.tgsja jobuPtr jobvPtr jobqPtr mPtr pPtr nPtr kPtr lPtr aPtr ldaPtr bPtr ldbPtr tolaPtr tolbPtr alphaPtr betaPtr uPtr lduPtr vPtr ldvPtr qPtr ldqPtr workPtr ncyclePtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray alpha
<*> Call.freezeArray beta
<*> fmap fromIntegral (peek ncyclePtr)
<*> fmap fromIntegral (peek infoPtr)
tgsna ::
Char ->
Char ->
Array ZeroInt Bool ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int, Int)
tgsna job howmny select a b vl vr mm lwork = do
let selectDim0 = Call.sizes1 $ Array.shape select
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (vlDim0,vlDim1) = Call.sizes2 $ Array.shape vl
let (vrDim0,vrDim1) = Call.sizes2 $ Array.shape vr
let n = selectDim0
let lda = aDim1
let ldb = bDim1
let m = vlDim0
let ldvl = vlDim1
let ldvr = vrDim1
Call.assert "tgsna: n == aDim0" (n == aDim0)
Call.assert "tgsna: n == bDim0" (n == bDim0)
Call.assert "tgsna: m == vrDim0" (m == vrDim0)
s <- Call.newArray1 mm
dif <- Call.newArray1 mm
work <- Call.newArray1 (maximum[1,lwork])
iwork <- Call.newArray1 (n+2)
evalContT $ do
jobPtr <- Call.char job
howmnyPtr <- Call.char howmny
selectPtr <- Call.array select
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
vlPtr <- Call.array vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.array vr
ldvrPtr <- Call.cint ldvr
sPtr <- Call.ioarray s
difPtr <- Call.ioarray dif
mmPtr <- Call.cint mm
mPtr <- Call.alloca
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.tgsna jobPtr howmnyPtr selectPtr nPtr aPtr ldaPtr bPtr ldbPtr vlPtr ldvlPtr vrPtr ldvrPtr sPtr difPtr mmPtr mPtr workPtr lworkPtr iworkPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> Call.freezeArray dif
<*> fmap fromIntegral (peek mPtr)
<*> fmap fromIntegral (peek infoPtr)
tgsy2 ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Float ->
Float ->
IO (Float, Float, Float, Int)
tgsy2 trans ijob a b c d e f rdsum rdscal = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let (dDim0,dDim1) = Call.sizes2 $ Array.shape d
let (eDim0,eDim1) = Call.sizes2 $ Array.shape e
let (fDim0,fDim1) = Call.sizes2 $ MutArray.shape f
let m = aDim0
let lda = aDim1
let n = bDim0
let ldb = bDim1
let ldc = cDim1
let ldd = dDim1
let lde = eDim1
let ldf = fDim1
Call.assert "tgsy2: n == cDim0" (n == cDim0)
Call.assert "tgsy2: m == dDim0" (m == dDim0)
Call.assert "tgsy2: n == eDim0" (n == eDim0)
Call.assert "tgsy2: n == fDim0" (n == fDim0)
evalContT $ do
transPtr <- Call.char trans
ijobPtr <- Call.cint ijob
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
dPtr <- Call.array d
lddPtr <- Call.cint ldd
ePtr <- Call.array e
ldePtr <- Call.cint lde
fPtr <- Call.ioarray f
ldfPtr <- Call.cint ldf
scalePtr <- Call.alloca
rdsumPtr <- Call.float rdsum
rdscalPtr <- Call.float rdscal
infoPtr <- Call.alloca
liftIO $ FFI.tgsy2 transPtr ijobPtr mPtr nPtr aPtr ldaPtr bPtr ldbPtr cPtr ldcPtr dPtr lddPtr ePtr ldePtr fPtr ldfPtr scalePtr rdsumPtr rdscalPtr infoPtr
liftIO $ pure (,,,)
<*> peek scalePtr
<*> peek rdsumPtr
<*> peek rdscalPtr
<*> fmap fromIntegral (peek infoPtr)
tgsyl ::
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Float, Float, Int)
tgsyl trans ijob a b c d e f lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let (dDim0,dDim1) = Call.sizes2 $ Array.shape d
let (eDim0,eDim1) = Call.sizes2 $ Array.shape e
let (fDim0,fDim1) = Call.sizes2 $ MutArray.shape f
let m = aDim0
let lda = aDim1
let n = bDim0
let ldb = bDim1
let ldc = cDim1
let ldd = dDim1
let lde = eDim1
let ldf = fDim1
Call.assert "tgsyl: n == cDim0" (n == cDim0)
Call.assert "tgsyl: m == dDim0" (m == dDim0)
Call.assert "tgsyl: n == eDim0" (n == eDim0)
Call.assert "tgsyl: n == fDim0" (n == fDim0)
work <- Call.newArray1 (maximum[1,lwork])
iwork <- Call.newArray1 (m+n+2)
evalContT $ do
transPtr <- Call.char trans
ijobPtr <- Call.cint ijob
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
dPtr <- Call.array d
lddPtr <- Call.cint ldd
ePtr <- Call.array e
ldePtr <- Call.cint lde
fPtr <- Call.ioarray f
ldfPtr <- Call.cint ldf
scalePtr <- Call.alloca
difPtr <- Call.alloca
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.tgsyl transPtr ijobPtr mPtr nPtr aPtr ldaPtr bPtr ldbPtr cPtr ldcPtr dPtr lddPtr ePtr ldePtr fPtr ldfPtr scalePtr difPtr workPtr lworkPtr iworkPtr infoPtr
liftIO $ pure (,,)
<*> peek scalePtr
<*> peek difPtr
<*> fmap fromIntegral (peek infoPtr)
tpcon ::
Char ->
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
IO (Float, Int)
tpcon norm uplo diag n ap = do
let apDim0 = Call.sizes1 $ Array.shape ap
Call.assert "tpcon: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
apPtr <- Call.array ap
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.tpcon normPtr uploPtr diagPtr nPtr apPtr rcondPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
tprfs ::
Char ->
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
tprfs uplo trans diag n ap b x = do
let apDim0 = Call.sizes1 $ Array.shape ap
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ Array.shape x
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "tprfs: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "tprfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.array x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.tprfs uploPtr transPtr diagPtr nPtr nrhsPtr apPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
tptri ::
Char ->
Char ->
Int ->
IOArray ZeroInt (Complex Float) ->
IO (Int)
tptri uplo diag n ap = do
let apDim0 = Call.sizes1 $ MutArray.shape ap
Call.assert "tptri: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
apPtr <- Call.ioarray ap
infoPtr <- Call.alloca
liftIO $ FFI.tptri uploPtr diagPtr nPtr apPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
tptrs ::
Char ->
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
tptrs uplo trans diag n ap b = do
let apDim0 = Call.sizes1 $ Array.shape ap
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let nrhs = bDim0
let ldb = bDim1
Call.assert "tptrs: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
apPtr <- Call.array ap
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.tptrs uploPtr transPtr diagPtr nPtr nrhsPtr apPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
tpttf ::
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
tpttf transr uplo n ap = do
let apDim0 = Call.sizes1 $ Array.shape ap
Call.assert "tpttf: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
arf <- Call.newArray1 (n*(n+1)`div`2)
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
arfPtr <- Call.ioarray arf
infoPtr <- Call.alloca
liftIO $ FFI.tpttf transrPtr uploPtr nPtr apPtr arfPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray arf
<*> fmap fromIntegral (peek infoPtr)
tpttr ::
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int)
tpttr uplo n ap lda = do
let apDim0 = Call.sizes1 $ Array.shape ap
Call.assert "tpttr: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
a <- Call.newArray2 n lda
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.tpttr uploPtr nPtr apPtr aPtr ldaPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray a
<*> fmap fromIntegral (peek infoPtr)
trcon ::
Char ->
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Float, Int)
trcon norm uplo diag a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
normPtr <- Call.char norm
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
rcondPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.trcon normPtr uploPtr diagPtr nPtr aPtr ldaPtr rcondPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> peek rcondPtr
<*> fmap fromIntegral (peek infoPtr)
trevc ::
Char ->
Char ->
Array ZeroInt Bool ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int, Int)
trevc side howmny select t vl vr = do
let selectDim0 = Call.sizes1 $ Array.shape select
let (tDim0,tDim1) = Call.sizes2 $ MutArray.shape t
let (vlDim0,vlDim1) = Call.sizes2 $ MutArray.shape vl
let (vrDim0,vrDim1) = Call.sizes2 $ MutArray.shape vr
let n = selectDim0
let ldt = tDim1
let mm = vlDim0
let ldvl = vlDim1
let ldvr = vrDim1
Call.assert "trevc: n == tDim0" (n == tDim0)
Call.assert "trevc: mm == vrDim0" (mm == vrDim0)
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
sidePtr <- Call.char side
howmnyPtr <- Call.char howmny
selectPtr <- Call.array select
nPtr <- Call.cint n
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
vlPtr <- Call.ioarray vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.ioarray vr
ldvrPtr <- Call.cint ldvr
mmPtr <- Call.cint mm
mPtr <- Call.alloca
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.trevc sidePtr howmnyPtr selectPtr nPtr tPtr ldtPtr vlPtr ldvlPtr vrPtr ldvrPtr mmPtr mPtr workPtr rworkPtr infoPtr
liftIO $ pure (,)
<*> fmap fromIntegral (peek mPtr)
<*> fmap fromIntegral (peek infoPtr)
trexc ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Int)
trexc compq t q ifst ilst = do
let (tDim0,tDim1) = Call.sizes2 $ MutArray.shape t
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let n = tDim0
let ldt = tDim1
let ldq = qDim1
Call.assert "trexc: n == qDim0" (n == qDim0)
evalContT $ do
compqPtr <- Call.char compq
nPtr <- Call.cint n
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
ifstPtr <- Call.cint ifst
ilstPtr <- Call.cint ilst
infoPtr <- Call.alloca
liftIO $ FFI.trexc compqPtr nPtr tPtr ldtPtr qPtr ldqPtr ifstPtr ilstPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
trrfs ::
Char ->
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int)
trrfs uplo trans diag a b x = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (xDim0,xDim1) = Call.sizes2 $ Array.shape x
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
let ldx = xDim1
Call.assert "trrfs: nrhs == xDim0" (nrhs == xDim0)
ferr <- Call.newArray1 nrhs
berr <- Call.newArray1 nrhs
work <- Call.newArray1 (2*n)
rwork <- Call.newArray1 n
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
xPtr <- Call.array x
ldxPtr <- Call.cint ldx
ferrPtr <- Call.ioarray ferr
berrPtr <- Call.ioarray berr
workPtr <- Call.ioarray work
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.trrfs uploPtr transPtr diagPtr nPtr nrhsPtr aPtr ldaPtr bPtr ldbPtr xPtr ldxPtr ferrPtr berrPtr workPtr rworkPtr infoPtr
liftIO $ pure (,,)
<*> Call.freezeArray ferr
<*> Call.freezeArray berr
<*> fmap fromIntegral (peek infoPtr)
trsen ::
Char ->
Char ->
Array ZeroInt Bool ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int, Float, Float, Int)
trsen job compq select t q lwork = do
let selectDim0 = Call.sizes1 $ Array.shape select
let (tDim0,tDim1) = Call.sizes2 $ MutArray.shape t
let (qDim0,qDim1) = Call.sizes2 $ MutArray.shape q
let n = selectDim0
let ldt = tDim1
let ldq = qDim1
Call.assert "trsen: n == tDim0" (n == tDim0)
Call.assert "trsen: n == qDim0" (n == qDim0)
w <- Call.newArray1 n
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
jobPtr <- Call.char job
compqPtr <- Call.char compq
selectPtr <- Call.array select
nPtr <- Call.cint n
tPtr <- Call.ioarray t
ldtPtr <- Call.cint ldt
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
wPtr <- Call.ioarray w
mPtr <- Call.alloca
sPtr <- Call.alloca
sepPtr <- Call.alloca
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.trsen jobPtr compqPtr selectPtr nPtr tPtr ldtPtr qPtr ldqPtr wPtr mPtr sPtr sepPtr workPtr lworkPtr infoPtr
liftIO $ pure (,,,,)
<*> Call.freezeArray w
<*> fmap fromIntegral (peek mPtr)
<*> peek sPtr
<*> peek sepPtr
<*> fmap fromIntegral (peek infoPtr)
trsna ::
Char ->
Char ->
Array ZeroInt Bool ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
IO (Array ZeroInt Float, Array ZeroInt Float, Int, Int)
trsna job howmny select t vl vr mm ldwork = do
let selectDim0 = Call.sizes1 $ Array.shape select
let (tDim0,tDim1) = Call.sizes2 $ Array.shape t
let (vlDim0,vlDim1) = Call.sizes2 $ Array.shape vl
let (vrDim0,vrDim1) = Call.sizes2 $ Array.shape vr
let n = selectDim0
let ldt = tDim1
let m = vlDim0
let ldvl = vlDim1
let ldvr = vrDim1
Call.assert "trsna: n == tDim0" (n == tDim0)
Call.assert "trsna: m == vrDim0" (m == vrDim0)
s <- Call.newArray1 mm
sep <- Call.newArray1 mm
work <- Call.newArray2 (n+6) ldwork
rwork <- Call.newArray1 n
evalContT $ do
jobPtr <- Call.char job
howmnyPtr <- Call.char howmny
selectPtr <- Call.array select
nPtr <- Call.cint n
tPtr <- Call.array t
ldtPtr <- Call.cint ldt
vlPtr <- Call.array vl
ldvlPtr <- Call.cint ldvl
vrPtr <- Call.array vr
ldvrPtr <- Call.cint ldvr
sPtr <- Call.ioarray s
sepPtr <- Call.ioarray sep
mmPtr <- Call.cint mm
mPtr <- Call.alloca
workPtr <- Call.ioarray work
ldworkPtr <- Call.cint ldwork
rworkPtr <- Call.ioarray rwork
infoPtr <- Call.alloca
liftIO $ FFI.trsna jobPtr howmnyPtr selectPtr nPtr tPtr ldtPtr vlPtr ldvlPtr vrPtr ldvrPtr sPtr sepPtr mmPtr mPtr workPtr ldworkPtr rworkPtr infoPtr
liftIO $ pure (,,,)
<*> Call.freezeArray s
<*> Call.freezeArray sep
<*> fmap fromIntegral (peek mPtr)
<*> fmap fromIntegral (peek infoPtr)
trsyl ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Float, Int)
trsyl trana tranb isgn a b c = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ Array.shape b
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let m = aDim0
let lda = aDim1
let n = bDim0
let ldb = bDim1
let ldc = cDim1
Call.assert "trsyl: n == cDim0" (n == cDim0)
evalContT $ do
tranaPtr <- Call.char trana
tranbPtr <- Call.char tranb
isgnPtr <- Call.cint isgn
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.array b
ldbPtr <- Call.cint ldb
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
scalePtr <- Call.alloca
infoPtr <- Call.alloca
liftIO $ FFI.trsyl tranaPtr tranbPtr isgnPtr mPtr nPtr aPtr ldaPtr bPtr ldbPtr cPtr ldcPtr scalePtr infoPtr
liftIO $ pure (,)
<*> peek scalePtr
<*> fmap fromIntegral (peek infoPtr)
trti2 ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
trti2 uplo diag a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.trti2 uploPtr diagPtr nPtr aPtr ldaPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
trtri ::
Char ->
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
trtri uplo diag a = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
evalContT $ do
uploPtr <- Call.char uplo
diagPtr <- Call.char diag
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
infoPtr <- Call.alloca
liftIO $ FFI.trtri uploPtr diagPtr nPtr aPtr ldaPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
trtrs ::
Char ->
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IO (Int)
trtrs uplo trans diag a b = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let (bDim0,bDim1) = Call.sizes2 $ MutArray.shape b
let n = aDim0
let lda = aDim1
let nrhs = bDim0
let ldb = bDim1
evalContT $ do
uploPtr <- Call.char uplo
transPtr <- Call.char trans
diagPtr <- Call.char diag
nPtr <- Call.cint n
nrhsPtr <- Call.cint nrhs
aPtr <- Call.array a
ldaPtr <- Call.cint lda
bPtr <- Call.ioarray b
ldbPtr <- Call.cint ldb
infoPtr <- Call.alloca
liftIO $ FFI.trtrs uploPtr transPtr diagPtr nPtr nrhsPtr aPtr ldaPtr bPtr ldbPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
trttf ::
Char ->
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
trttf transr uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
arf <- Call.newArray1 (n*(n+1)`div`2)
evalContT $ do
transrPtr <- Call.char transr
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
arfPtr <- Call.ioarray arf
infoPtr <- Call.alloca
liftIO $ FFI.trttf transrPtr uploPtr nPtr aPtr ldaPtr arfPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray arf
<*> fmap fromIntegral (peek infoPtr)
trttp ::
Char ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
IO (Array ZeroInt (Complex Float), Int)
trttp uplo a = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let n = aDim0
let lda = aDim1
ap <- Call.newArray1 (n*(n+1)`div`2)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
apPtr <- Call.ioarray ap
infoPtr <- Call.alloca
liftIO $ FFI.trttp uploPtr nPtr aPtr ldaPtr apPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray ap
<*> fmap fromIntegral (peek infoPtr)
tzrzf ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt (Complex Float), Int)
tzrzf m a lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let n = aDim0
let lda = aDim1
tau <- Call.newArray1 m
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.ioarray tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.tzrzf mPtr nPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray tau
<*> fmap fromIntegral (peek infoPtr)
unbdb ::
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Array ZeroInt Float, Array ZeroInt Float, Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Array ZeroInt (Complex Float), Int)
unbdb trans signs m p x11 x12 x21 x22 lwork = do
let (x11Dim0,x11Dim1) = Call.sizes2 $ MutArray.shape x11
let (x12Dim0,x12Dim1) = Call.sizes2 $ MutArray.shape x12
let (x21Dim0,x21Dim1) = Call.sizes2 $ MutArray.shape x21
let (x22Dim0,x22Dim1) = Call.sizes2 $ MutArray.shape x22
let q = x11Dim0
let ldx11 = x11Dim1
let ldx12 = x12Dim1
let ldx21 = x21Dim1
let ldx22 = x22Dim1
Call.assert "unbdb: m-q == x12Dim0" (m-q == x12Dim0)
Call.assert "unbdb: q == x21Dim0" (q == x21Dim0)
Call.assert "unbdb: m-q == x22Dim0" (m-q == x22Dim0)
theta <- Call.newArray1 q
phi <- Call.newArray1 (q-1)
taup1 <- Call.newArray1 p
taup2 <- Call.newArray1 (m-p)
tauq1 <- Call.newArray1 q
tauq2 <- Call.newArray1 (m-q)
work <- Call.newArray1 lwork
evalContT $ do
transPtr <- Call.char trans
signsPtr <- Call.char signs
mPtr <- Call.cint m
pPtr <- Call.cint p
qPtr <- Call.cint q
x11Ptr <- Call.ioarray x11
ldx11Ptr <- Call.cint ldx11
x12Ptr <- Call.ioarray x12
ldx12Ptr <- Call.cint ldx12
x21Ptr <- Call.ioarray x21
ldx21Ptr <- Call.cint ldx21
x22Ptr <- Call.ioarray x22
ldx22Ptr <- Call.cint ldx22
thetaPtr <- Call.ioarray theta
phiPtr <- Call.ioarray phi
taup1Ptr <- Call.ioarray taup1
taup2Ptr <- Call.ioarray taup2
tauq1Ptr <- Call.ioarray tauq1
tauq2Ptr <- Call.ioarray tauq2
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unbdb transPtr signsPtr mPtr pPtr qPtr x11Ptr ldx11Ptr x12Ptr ldx12Ptr x21Ptr ldx21Ptr x22Ptr ldx22Ptr thetaPtr phiPtr taup1Ptr taup2Ptr tauq1Ptr tauq2Ptr workPtr lworkPtr infoPtr
liftIO $ pure (,,,,,,)
<*> Call.freezeArray theta
<*> Call.freezeArray phi
<*> Call.freezeArray taup1
<*> Call.freezeArray taup2
<*> Call.freezeArray tauq1
<*> Call.freezeArray tauq2
<*> fmap fromIntegral (peek infoPtr)
uncsd ::
Char ->
Char ->
Char ->
Char ->
Char ->
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
Int ->
Int ->
Int ->
Int ->
Int ->
Int ->
Int ->
IO (Array ZeroInt Float, Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Array (ZeroInt,ZeroInt) (Complex Float), Int)
uncsd jobu1 jobu2 jobv1t jobv2t trans signs m p x11 x12 x21 x22 r ldu1 ldu2 ldv1t ldv2t lwork rworkSize lrwork = do
let (x11Dim0,x11Dim1) = Call.sizes2 $ MutArray.shape x11
let (x12Dim0,x12Dim1) = Call.sizes2 $ MutArray.shape x12
let (x21Dim0,x21Dim1) = Call.sizes2 $ MutArray.shape x21
let (x22Dim0,x22Dim1) = Call.sizes2 $ MutArray.shape x22
let q = x11Dim0
let ldx11 = x11Dim1
let ldx12 = x12Dim1
let ldx21 = x21Dim1
let ldx22 = x22Dim1
Call.assert "uncsd: m-q == x12Dim0" (m-q == x12Dim0)
Call.assert "uncsd: q == x21Dim0" (q == x21Dim0)
Call.assert "uncsd: m-q == x22Dim0" (m-q == x22Dim0)
theta <- Call.newArray1 r
u1 <- Call.newArray2 p ldu1
u2 <- Call.newArray2 (m-p) ldu2
v1t <- Call.newArray2 q ldv1t
v2t <- Call.newArray2 (m-q) ldv2t
work <- Call.newArray1 (maximum[1,lwork])
rwork <- Call.newArray1 rworkSize
iwork <- Call.newArray1 (m-minimum[p,m-p,q,m-q])
evalContT $ do
jobu1Ptr <- Call.char jobu1
jobu2Ptr <- Call.char jobu2
jobv1tPtr <- Call.char jobv1t
jobv2tPtr <- Call.char jobv2t
transPtr <- Call.char trans
signsPtr <- Call.char signs
mPtr <- Call.cint m
pPtr <- Call.cint p
qPtr <- Call.cint q
x11Ptr <- Call.ioarray x11
ldx11Ptr <- Call.cint ldx11
x12Ptr <- Call.ioarray x12
ldx12Ptr <- Call.cint ldx12
x21Ptr <- Call.ioarray x21
ldx21Ptr <- Call.cint ldx21
x22Ptr <- Call.ioarray x22
ldx22Ptr <- Call.cint ldx22
thetaPtr <- Call.ioarray theta
u1Ptr <- Call.ioarray u1
ldu1Ptr <- Call.cint ldu1
u2Ptr <- Call.ioarray u2
ldu2Ptr <- Call.cint ldu2
v1tPtr <- Call.ioarray v1t
ldv1tPtr <- Call.cint ldv1t
v2tPtr <- Call.ioarray v2t
ldv2tPtr <- Call.cint ldv2t
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
rworkPtr <- Call.ioarray rwork
lrworkPtr <- Call.cint lrwork
iworkPtr <- Call.ioarray iwork
infoPtr <- Call.alloca
liftIO $ FFI.uncsd jobu1Ptr jobu2Ptr jobv1tPtr jobv2tPtr transPtr signsPtr mPtr pPtr qPtr x11Ptr ldx11Ptr x12Ptr ldx12Ptr x21Ptr ldx21Ptr x22Ptr ldx22Ptr thetaPtr u1Ptr ldu1Ptr u2Ptr ldu2Ptr v1tPtr ldv1tPtr v2tPtr ldv2tPtr workPtr lworkPtr rworkPtr lrworkPtr iworkPtr infoPtr
liftIO $ pure (,,,,,)
<*> Call.freezeArray theta
<*> Call.freezeArray u1
<*> Call.freezeArray u2
<*> Call.freezeArray v1t
<*> Call.freezeArray v2t
<*> fmap fromIntegral (peek infoPtr)
ung2l ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IO (Int)
ung2l m a tau = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let k = tauDim0
work <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.ung2l mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ung2r ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IO (Int)
ung2r m a tau = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let k = tauDim0
work <- Call.newArray1 n
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.ung2r mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ungbr ::
Char ->
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Int)
ungbr vect m k a tau lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let _tauSize = tauDim0
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
vectPtr <- Call.char vect
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.ungbr vectPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unghr ::
Int ->
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Int)
unghr ilo ihi a tau lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
Call.assert "unghr: n-1 == tauDim0" (n-1 == tauDim0)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unghr nPtr iloPtr ihiPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ungl2 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IO (Int)
ungl2 m a tau = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let k = tauDim0
work <- Call.newArray1 m
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.ungl2 mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unglq ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Int)
unglq m a tau lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let k = tauDim0
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unglq mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ungql ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Int)
ungql m a tau lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let k = tauDim0
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.ungql mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ungqr ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Int)
ungqr m a tau lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let k = tauDim0
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.ungqr mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ungr2 ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IO (Int)
ungr2 m a tau = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let k = tauDim0
work <- Call.newArray1 m
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.ungr2 mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ungrq ::
Int ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Int)
ungrq m a tau lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
let k = tauDim0
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.ungrq mPtr nPtr kPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
ungtr ::
Char ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Int)
ungtr uplo a tau lwork = do
let (aDim0,aDim1) = Call.sizes2 $ MutArray.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let n = aDim0
let lda = aDim1
Call.assert "ungtr: n-1 == tauDim0" (n-1 == tauDim0)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
aPtr <- Call.ioarray a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.ungtr uploPtr nPtr aPtr ldaPtr tauPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unm2l ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unm2l side trans m a tau c workSize = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let k = aDim0
let lda = aDim1
let n = cDim0
let ldc = cDim1
Call.assert "unm2l: k == tauDim0" (k == tauDim0)
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.unm2l sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unm2r ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unm2r side trans m a tau c workSize = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let k = aDim0
let lda = aDim1
let n = cDim0
let ldc = cDim1
Call.assert "unm2r: k == tauDim0" (k == tauDim0)
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.unm2r sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmbr ::
Char ->
Char ->
Char ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmbr vect side trans m k a tau c lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let n = cDim0
let ldc = cDim1
Call.ignore "unmbr: minimum[nq,k] == tauDim0" tauDim0
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
vectPtr <- Call.char vect
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unmbr vectPtr sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmhr ::
Char ->
Char ->
Int ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmhr side trans m ilo ihi a tau c lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let _tauSize = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
iloPtr <- Call.cint ilo
ihiPtr <- Call.cint ihi
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unmhr sidePtr transPtr mPtr nPtr iloPtr ihiPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unml2 ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unml2 side trans m a tau c workSize = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let k = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.unml2 sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmlq ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmlq side trans m a tau c lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let k = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unmlq sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmql ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmql side trans m a tau c lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let k = aDim0
let lda = aDim1
let n = cDim0
let ldc = cDim1
Call.assert "unmql: k == tauDim0" (k == tauDim0)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unmql sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmqr ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmqr side trans m a tau c lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let k = aDim0
let lda = aDim1
let n = cDim0
let ldc = cDim1
Call.assert "unmqr: k == tauDim0" (k == tauDim0)
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unmqr sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmr2 ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmr2 side trans m a tau c workSize = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let k = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.unmr2 sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmr3 ::
Char ->
Char ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmr3 side trans m l a tau c workSize = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let k = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
lPtr <- Call.cint l
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.unmr3 sidePtr transPtr mPtr nPtr kPtr lPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmrq ::
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmrq side trans m a tau c lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let k = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unmrq sidePtr transPtr mPtr nPtr kPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmrz ::
Char ->
Char ->
Int ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmrz side trans m l a tau c lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let k = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
sidePtr <- Call.char side
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
kPtr <- Call.cint k
lPtr <- Call.cint l
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unmrz sidePtr transPtr mPtr nPtr kPtr lPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
unmtr ::
Char ->
Char ->
Char ->
Int ->
Array (ZeroInt,ZeroInt) (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
unmtr side uplo trans m a tau c lwork = do
let (aDim0,aDim1) = Call.sizes2 $ Array.shape a
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _aSize = aDim0
let lda = aDim1
let _tauSize = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 (maximum[1,lwork])
evalContT $ do
sidePtr <- Call.char side
uploPtr <- Call.char uplo
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
aPtr <- Call.array a
ldaPtr <- Call.cint lda
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
lworkPtr <- Call.cint lwork
infoPtr <- Call.alloca
liftIO $ FFI.unmtr sidePtr uploPtr transPtr mPtr nPtr aPtr ldaPtr tauPtr cPtr ldcPtr workPtr lworkPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)
upgtr ::
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
Int ->
IO (Array (ZeroInt,ZeroInt) (Complex Float), Int)
upgtr uplo n ap tau ldq = do
let apDim0 = Call.sizes1 $ Array.shape ap
let tauDim0 = Call.sizes1 $ Array.shape tau
Call.assert "upgtr: n*(n+1)`div`2 == apDim0" (n*(n+1)`div`2 == apDim0)
Call.assert "upgtr: n-1 == tauDim0" (n-1 == tauDim0)
q <- Call.newArray2 n ldq
work <- Call.newArray1 (n-1)
evalContT $ do
uploPtr <- Call.char uplo
nPtr <- Call.cint n
apPtr <- Call.array ap
tauPtr <- Call.array tau
qPtr <- Call.ioarray q
ldqPtr <- Call.cint ldq
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.upgtr uploPtr nPtr apPtr tauPtr qPtr ldqPtr workPtr infoPtr
liftIO $ pure (,)
<*> Call.freezeArray q
<*> fmap fromIntegral (peek infoPtr)
upmtr ::
Char ->
Char ->
Char ->
Int ->
Array ZeroInt (Complex Float) ->
Array ZeroInt (Complex Float) ->
IOArray (ZeroInt,ZeroInt) (Complex Float) ->
Int ->
IO (Int)
upmtr side uplo trans m ap tau c workSize = do
let apDim0 = Call.sizes1 $ Array.shape ap
let tauDim0 = Call.sizes1 $ Array.shape tau
let (cDim0,cDim1) = Call.sizes2 $ MutArray.shape c
let _apSize = apDim0
let _tauSize = tauDim0
let n = cDim0
let ldc = cDim1
work <- Call.newArray1 workSize
evalContT $ do
sidePtr <- Call.char side
uploPtr <- Call.char uplo
transPtr <- Call.char trans
mPtr <- Call.cint m
nPtr <- Call.cint n
apPtr <- Call.array ap
tauPtr <- Call.array tau
cPtr <- Call.ioarray c
ldcPtr <- Call.cint ldc
workPtr <- Call.ioarray work
infoPtr <- Call.alloca
liftIO $ FFI.upmtr sidePtr uploPtr transPtr mPtr nPtr apPtr tauPtr cPtr ldcPtr workPtr infoPtr
liftIO $ fmap fromIntegral (peek infoPtr)