module TensorSpec (spec) where
import Foreign
import Foreign.C.Types
import Test.Hspec
import Torch.FFI.Tests
import Torch.Types.THC (C'THCState)
import qualified Torch.FFI.THC.General as General
import qualified Torch.Types.THC.Byte as B
import qualified Torch.FFI.THC.Byte.Tensor as B
import qualified Torch.FFI.THC.Byte.TensorIndex as B
import qualified Torch.FFI.THC.Byte.TensorMasked as B
import qualified Torch.FFI.THC.Byte.TensorMathCompare as B
import qualified Torch.FFI.THC.Byte.TensorMathCompareT as B
import qualified Torch.FFI.THC.Byte.TensorMath as B
import qualified Torch.FFI.THC.Byte.TensorMathPairwise as B
import qualified Torch.FFI.THC.Byte.TensorMathPointwise as B
import qualified Torch.FFI.THC.Byte.TensorMathReduce as B
import qualified Torch.FFI.THC.Byte.TensorMathScan as B
import qualified Torch.FFI.THC.Byte.TensorMode as B
import qualified Torch.FFI.THC.Byte.TensorRandom as B
import qualified Torch.FFI.THC.Byte.TensorScatterGather as B
import qualified Torch.FFI.THC.Byte.TensorSort as B
import qualified Torch.FFI.THC.Byte.TensorTopK as B
-- import qualified Torch.Types.THC.Float as F
-- import qualified Torch.FFI.THC.Float.Tensor as F
-- import qualified Torch.FFI.THC.Float.TensorIndex as F
-- import qualified Torch.FFI.THC.Float.TensorMasked as F
-- import qualified Torch.FFI.THC.Float.TensorMathBlas as F
-- import qualified Torch.FFI.THC.Float.TensorMathCompare as F
-- import qualified Torch.FFI.THC.Float.TensorMathCompareT as F
-- import qualified Torch.FFI.THC.Float.TensorMath as F
-- import qualified Torch.FFI.THC.Float.TensorMathPairwise as F
-- import qualified Torch.FFI.THC.Float.TensorMathPointwise as F
-- import qualified Torch.FFI.THC.Float.TensorMathReduce as F
-- import qualified Torch.FFI.THC.Float.TensorMathScan as F
-- import qualified Torch.FFI.THC.Float.TensorMode as F
-- import qualified Torch.FFI.THC.Float.TensorRandom as F
-- import qualified Torch.FFI.THC.Float.TensorScatterGather as F
-- import qualified Torch.FFI.THC.Float.TensorSort as F
-- import qualified Torch.FFI.THC.Float.TensorTopK as F
import qualified Torch.Types.THC.Double as D
import qualified Torch.FFI.THC.Double.Tensor as D
import qualified Torch.FFI.THC.Double.TensorIndex as D
import qualified Torch.FFI.THC.Double.TensorMasked as D
import qualified Torch.FFI.THC.Double.TensorMathBlas as D
import qualified Torch.FFI.THC.Double.TensorMathCompare as D
import qualified Torch.FFI.THC.Double.TensorMathCompareT as D
import qualified Torch.FFI.THC.Double.TensorMath as D
import qualified Torch.FFI.THC.Double.TensorMathPairwise as D
import qualified Torch.FFI.THC.Double.TensorMathPointwise as D
import qualified Torch.FFI.THC.Double.TensorMathReduce as D
import qualified Torch.FFI.THC.Double.TensorMathScan as D
import qualified Torch.FFI.THC.Double.TensorMode as D
import qualified Torch.FFI.THC.Double.TensorRandom as D
import qualified Torch.FFI.THC.Double.TensorScatterGather as D
import qualified Torch.FFI.THC.Double.TensorSort as D
import qualified Torch.FFI.THC.Double.TensorTopK as D
import qualified Torch.Types.THC.Int as I
import qualified Torch.FFI.THC.Int.Tensor as I
import qualified Torch.FFI.THC.Int.TensorIndex as I
import qualified Torch.FFI.THC.Int.TensorMasked as I
import qualified Torch.FFI.THC.Int.TensorMathCompare as I
import qualified Torch.FFI.THC.Int.TensorMathCompareT as I
import qualified Torch.FFI.THC.Int.TensorMath as I
import qualified Torch.FFI.THC.Int.TensorMathPairwise as I
import qualified Torch.FFI.THC.Int.TensorMathPointwise as I
import qualified Torch.FFI.THC.Int.TensorMathReduce as I
import qualified Torch.FFI.THC.Int.TensorMathScan as I
import qualified Torch.FFI.THC.Int.TensorMode as I
import qualified Torch.FFI.THC.Int.TensorRandom as I
import qualified Torch.FFI.THC.Int.TensorScatterGather as I
import qualified Torch.FFI.THC.Int.TensorSort as I
import qualified Torch.FFI.THC.Int.TensorTopK as I
import qualified Torch.Types.THC.Short as S
import qualified Torch.FFI.THC.Short.Tensor as S
import qualified Torch.FFI.THC.Short.TensorIndex as S
import qualified Torch.FFI.THC.Short.TensorMasked as S
import qualified Torch.FFI.THC.Short.TensorMathCompare as S
import qualified Torch.FFI.THC.Short.TensorMathCompareT as S
import qualified Torch.FFI.THC.Short.TensorMath as S
import qualified Torch.FFI.THC.Short.TensorMathPairwise as S
import qualified Torch.FFI.THC.Short.TensorMathPointwise as S
import qualified Torch.FFI.THC.Short.TensorMathReduce as S
import qualified Torch.FFI.THC.Short.TensorMathScan as S
import qualified Torch.FFI.THC.Short.TensorMode as S
import qualified Torch.FFI.THC.Short.TensorRandom as S
import qualified Torch.FFI.THC.Short.TensorScatterGather as S
import qualified Torch.FFI.THC.Short.TensorSort as S
import qualified Torch.FFI.THC.Short.TensorTopK as S
import qualified Torch.Types.THC.Long as L
import qualified Torch.FFI.THC.Long.Tensor as L
import qualified Torch.FFI.THC.Long.TensorIndex as L
import qualified Torch.FFI.THC.Long.TensorMasked as L
import qualified Torch.FFI.THC.Long.TensorMathCompare as L
import qualified Torch.FFI.THC.Long.TensorMathCompareT as L
import qualified Torch.FFI.THC.Long.TensorMath as L
import qualified Torch.FFI.THC.Long.TensorMathPairwise as L
import qualified Torch.FFI.THC.Long.TensorMathPointwise as L
import qualified Torch.FFI.THC.Long.TensorMathReduce as L
import qualified Torch.FFI.THC.Long.TensorMathScan as L
import qualified Torch.FFI.THC.Long.TensorMode as L
import qualified Torch.FFI.THC.Long.TensorRandom as L
import qualified Torch.FFI.THC.Long.TensorScatterGather as L
import qualified Torch.FFI.THC.Long.TensorSort as L
import qualified Torch.FFI.THC.Long.TensorTopK as L
import Internal
main :: IO ()
main = hspec spec
spec :: Spec
spec = do
s <- runIO cudaState
afterAll_ (stopCuda s) $ do
-- describe "Float Tensor creation and access methods" $ signedSuite s floatBook
describe "Double Tensor creation and access methods" $ signedSuite s doubleBook
describe "Byte Tensor creation and access methods" $ signedSuite s byteBook
describe "Int Tensor creation and access methods" $ signedSuite s intBook
describe "Long Tensor creation and access methods" $ signedSuite s longBook
describe "Short Tensor creation and access methods" $ signedSuite s shortBook
describe "Double Tensor creation and access methods" $ signedSuite s doubleBook
longBook :: TestFunctions (Ptr C'THCState) (Ptr L.CTensor) L.CReal L.CAccReal
longBook = TestFunctions
{ _new = L.c_new
, _newWithSize1d = L.c_newWithSize1d
, _newWithSize2d = L.c_newWithSize2d
, _newWithSize3d = L.c_newWithSize3d
, _newWithSize4d = L.c_newWithSize4d
, _nDimension = L.c_nDimension
, _set1d = L.c_set1d
, _get1d = L.c_get1d
, _set2d = L.c_set2d
, _get2d = L.c_get2d
, _set3d = L.c_set3d
, _get3d = L.c_get3d
, _set4d = L.c_set4d
, _get4d = L.c_get4d
, _size = L.c_size
, _fill = L.c_fill
, _free = L.c_free
, _sumall = L.c_sumall
, _prodall = L.c_prodall
, _zero = L.c_zero
, _dot = Nothing
, _abs = Just L.c_abs
}
shortBook :: TestFunctions (Ptr C'THCState) (Ptr S.CTensor) S.CReal S.CAccReal
shortBook = TestFunctions
{ _new = S.c_new
, _newWithSize1d = S.c_newWithSize1d
, _newWithSize2d = S.c_newWithSize2d
, _newWithSize3d = S.c_newWithSize3d
, _newWithSize4d = S.c_newWithSize4d
, _nDimension = S.c_nDimension
, _set1d = S.c_set1d
, _get1d = S.c_get1d
, _set2d = S.c_set2d
, _get2d = S.c_get2d
, _set3d = S.c_set3d
, _get3d = S.c_get3d
, _set4d = S.c_set4d
, _get4d = S.c_get4d
, _size = S.c_size
, _fill = S.c_fill
, _free = S.c_free
, _sumall = S.c_sumall
, _prodall = S.c_prodall
, _zero = S.c_zero
, _dot = Nothing
, _abs = Just S.c_abs
}
{-
floatBook :: TestFunctions (Ptr C'THCState) (Ptr F.CTensor) F.CReal F.CAccReal
floatBook = TestFunctions
{ _new = F.c_new
, _newWithSize1d = F.c_newWithSize1d
, _newWithSize2d = F.c_newWithSize2d
, _newWithSize3d = F.c_newWithSize3d
, _newWithSize4d = F.c_newWithSize4d
, _nDimension = F.c_nDimension
, _set1d = F.c_set1d
, _get1d = F.c_get1d
, _set2d = F.c_set2d
, _get2d = F.c_get2d
, _set3d = F.c_set3d
, _get3d = F.c_get3d
, _set4d = F.c_set4d
, _get4d = F.c_get4d
, _size = F.c_size
, _fill = F.c_fill
, _free = F.c_free
, _sumall = F.c_sumall
, _prodall = F.c_prodall
, _zero = F.c_zero
, _dot = F.c_dot
, _abs = Just F.c_abs
}
-}
doubleBook :: TestFunctions (Ptr C'THCState) (Ptr D.CTensor) D.CReal D.CAccReal
doubleBook = TestFunctions
{ _new = D.c_new
, _newWithSize1d = D.c_newWithSize1d
, _newWithSize2d = D.c_newWithSize2d
, _newWithSize3d = D.c_newWithSize3d
, _newWithSize4d = D.c_newWithSize4d
, _nDimension = D.c_nDimension
, _set1d = D.c_set1d
, _get1d = D.c_get1d
, _set2d = D.c_set2d
, _get2d = D.c_get2d
, _set3d = D.c_set3d
, _get3d = D.c_get3d
, _set4d = D.c_set4d
, _get4d = D.c_get4d
, _size = D.c_size
, _fill = D.c_fill
, _free = D.c_free
, _sumall = D.c_sumall
, _prodall = D.c_prodall
, _zero = D.c_zero
, _dot = Just D.c_dot
, _abs = Just D.c_abs
}
byteBook :: TestFunctions (Ptr C'THCState) (Ptr B.CTensor) B.CReal B.CAccReal
byteBook = TestFunctions
{ _new = B.c_new
, _newWithSize1d = B.c_newWithSize1d
, _newWithSize2d = B.c_newWithSize2d
, _newWithSize3d = B.c_newWithSize3d
, _newWithSize4d = B.c_newWithSize4d
, _nDimension = B.c_nDimension
, _set1d = B.c_set1d
, _get1d = B.c_get1d
, _set2d = B.c_set2d
, _get2d = B.c_get2d
, _set3d = B.c_set3d
, _get3d = B.c_get3d
, _set4d = B.c_set4d
, _get4d = B.c_get4d
, _size = B.c_size
, _fill = B.c_fill
, _free = B.c_free
, _sumall = B.c_sumall
, _prodall = B.c_prodall
, _zero = B.c_zero
, _dot = Nothing
, _abs = Nothing
}
intBook :: TestFunctions (Ptr C'THCState) (Ptr I.CTensor) I.CReal I.CAccReal
intBook = TestFunctions
{ _new = I.c_new
, _newWithSize1d = I.c_newWithSize1d
, _newWithSize2d = I.c_newWithSize2d
, _newWithSize3d = I.c_newWithSize3d
, _newWithSize4d = I.c_newWithSize4d
, _nDimension = I.c_nDimension
, _set1d = I.c_set1d
, _get1d = I.c_get1d
, _set2d = I.c_set2d
, _get2d = I.c_get2d
, _set3d = I.c_set3d
, _get3d = I.c_get3d
, _set4d = I.c_set4d
, _get4d = I.c_get4d
, _size = I.c_size
, _fill = I.c_fill
, _free = I.c_free
, _sumall = I.c_sumall
, _prodall = I.c_prodall
, _zero = I.c_zero
, _dot = Nothing
, _abs = Just I.c_abs
}