tomoto 0.2.2 → 0.3.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (369) hide show
  1. checksums.yaml +4 -4
  2. data/CHANGELOG.md +10 -0
  3. data/README.md +8 -10
  4. data/ext/tomoto/ct.cpp +11 -11
  5. data/ext/tomoto/dmr.cpp +14 -13
  6. data/ext/tomoto/dt.cpp +14 -14
  7. data/ext/tomoto/extconf.rb +7 -5
  8. data/ext/tomoto/gdmr.cpp +7 -7
  9. data/ext/tomoto/hdp.cpp +9 -9
  10. data/ext/tomoto/hlda.cpp +13 -13
  11. data/ext/tomoto/hpa.cpp +5 -5
  12. data/ext/tomoto/lda.cpp +42 -39
  13. data/ext/tomoto/llda.cpp +6 -6
  14. data/ext/tomoto/mglda.cpp +15 -15
  15. data/ext/tomoto/pa.cpp +6 -6
  16. data/ext/tomoto/plda.cpp +6 -6
  17. data/ext/tomoto/slda.cpp +8 -8
  18. data/ext/tomoto/{ext.cpp → tomoto.cpp} +8 -8
  19. data/ext/tomoto/utils.h +16 -70
  20. data/lib/tomoto/version.rb +1 -1
  21. data/lib/tomoto.rb +5 -1
  22. data/vendor/EigenRand/EigenRand/Core.h +10 -10
  23. data/vendor/EigenRand/EigenRand/Dists/Basic.h +208 -9
  24. data/vendor/EigenRand/EigenRand/Dists/Discrete.h +52 -31
  25. data/vendor/EigenRand/EigenRand/Dists/GammaPoisson.h +9 -8
  26. data/vendor/EigenRand/EigenRand/Dists/NormalExp.h +28 -21
  27. data/vendor/EigenRand/EigenRand/EigenRand +11 -6
  28. data/vendor/EigenRand/EigenRand/Macro.h +13 -7
  29. data/vendor/EigenRand/EigenRand/MorePacketMath.h +348 -740
  30. data/vendor/EigenRand/EigenRand/MvDists/Multinomial.h +5 -3
  31. data/vendor/EigenRand/EigenRand/MvDists/MvNormal.h +9 -3
  32. data/vendor/EigenRand/EigenRand/PacketFilter.h +11 -253
  33. data/vendor/EigenRand/EigenRand/PacketRandomEngine.h +21 -47
  34. data/vendor/EigenRand/EigenRand/RandUtils.h +50 -344
  35. data/vendor/EigenRand/EigenRand/arch/AVX/MorePacketMath.h +619 -0
  36. data/vendor/EigenRand/EigenRand/arch/AVX/PacketFilter.h +149 -0
  37. data/vendor/EigenRand/EigenRand/arch/AVX/RandUtils.h +228 -0
  38. data/vendor/EigenRand/EigenRand/arch/NEON/MorePacketMath.h +473 -0
  39. data/vendor/EigenRand/EigenRand/arch/NEON/PacketFilter.h +142 -0
  40. data/vendor/EigenRand/EigenRand/arch/NEON/RandUtils.h +126 -0
  41. data/vendor/EigenRand/EigenRand/arch/SSE/MorePacketMath.h +501 -0
  42. data/vendor/EigenRand/EigenRand/arch/SSE/PacketFilter.h +133 -0
  43. data/vendor/EigenRand/EigenRand/arch/SSE/RandUtils.h +120 -0
  44. data/vendor/EigenRand/EigenRand/doc.h +24 -12
  45. data/vendor/EigenRand/README.md +57 -4
  46. data/vendor/eigen/COPYING.APACHE +203 -0
  47. data/vendor/eigen/COPYING.BSD +1 -1
  48. data/vendor/eigen/COPYING.MINPACK +51 -52
  49. data/vendor/eigen/Eigen/Cholesky +0 -1
  50. data/vendor/eigen/Eigen/Core +112 -265
  51. data/vendor/eigen/Eigen/Eigenvalues +2 -3
  52. data/vendor/eigen/Eigen/Geometry +5 -8
  53. data/vendor/eigen/Eigen/Householder +0 -1
  54. data/vendor/eigen/Eigen/Jacobi +0 -1
  55. data/vendor/eigen/Eigen/KLUSupport +41 -0
  56. data/vendor/eigen/Eigen/LU +2 -5
  57. data/vendor/eigen/Eigen/OrderingMethods +0 -3
  58. data/vendor/eigen/Eigen/PaStiXSupport +1 -0
  59. data/vendor/eigen/Eigen/PardisoSupport +0 -0
  60. data/vendor/eigen/Eigen/QR +2 -3
  61. data/vendor/eigen/Eigen/QtAlignedMalloc +0 -1
  62. data/vendor/eigen/Eigen/SVD +0 -1
  63. data/vendor/eigen/Eigen/Sparse +0 -2
  64. data/vendor/eigen/Eigen/SparseCholesky +0 -8
  65. data/vendor/eigen/Eigen/SparseLU +4 -0
  66. data/vendor/eigen/Eigen/SparseQR +0 -1
  67. data/vendor/eigen/Eigen/src/Cholesky/LDLT.h +42 -27
  68. data/vendor/eigen/Eigen/src/Cholesky/LLT.h +39 -23
  69. data/vendor/eigen/Eigen/src/CholmodSupport/CholmodSupport.h +90 -47
  70. data/vendor/eigen/Eigen/src/Core/ArithmeticSequence.h +413 -0
  71. data/vendor/eigen/Eigen/src/Core/Array.h +99 -11
  72. data/vendor/eigen/Eigen/src/Core/ArrayBase.h +3 -3
  73. data/vendor/eigen/Eigen/src/Core/ArrayWrapper.h +21 -21
  74. data/vendor/eigen/Eigen/src/Core/Assign.h +1 -1
  75. data/vendor/eigen/Eigen/src/Core/AssignEvaluator.h +125 -50
  76. data/vendor/eigen/Eigen/src/Core/Assign_MKL.h +10 -10
  77. data/vendor/eigen/Eigen/src/Core/BandMatrix.h +16 -16
  78. data/vendor/eigen/Eigen/src/Core/Block.h +56 -60
  79. data/vendor/eigen/Eigen/src/Core/BooleanRedux.h +29 -31
  80. data/vendor/eigen/Eigen/src/Core/CommaInitializer.h +7 -3
  81. data/vendor/eigen/Eigen/src/Core/CoreEvaluators.h +325 -272
  82. data/vendor/eigen/Eigen/src/Core/CoreIterators.h +5 -0
  83. data/vendor/eigen/Eigen/src/Core/CwiseBinaryOp.h +21 -22
  84. data/vendor/eigen/Eigen/src/Core/CwiseNullaryOp.h +153 -18
  85. data/vendor/eigen/Eigen/src/Core/CwiseUnaryOp.h +6 -6
  86. data/vendor/eigen/Eigen/src/Core/CwiseUnaryView.h +14 -10
  87. data/vendor/eigen/Eigen/src/Core/DenseBase.h +132 -42
  88. data/vendor/eigen/Eigen/src/Core/DenseCoeffsBase.h +25 -21
  89. data/vendor/eigen/Eigen/src/Core/DenseStorage.h +153 -71
  90. data/vendor/eigen/Eigen/src/Core/Diagonal.h +21 -23
  91. data/vendor/eigen/Eigen/src/Core/DiagonalMatrix.h +50 -2
  92. data/vendor/eigen/Eigen/src/Core/DiagonalProduct.h +1 -1
  93. data/vendor/eigen/Eigen/src/Core/Dot.h +10 -10
  94. data/vendor/eigen/Eigen/src/Core/EigenBase.h +10 -9
  95. data/vendor/eigen/Eigen/src/Core/ForceAlignedAccess.h +8 -4
  96. data/vendor/eigen/Eigen/src/Core/Fuzzy.h +3 -3
  97. data/vendor/eigen/Eigen/src/Core/GeneralProduct.h +20 -10
  98. data/vendor/eigen/Eigen/src/Core/GenericPacketMath.h +599 -152
  99. data/vendor/eigen/Eigen/src/Core/GlobalFunctions.h +40 -33
  100. data/vendor/eigen/Eigen/src/Core/IO.h +40 -7
  101. data/vendor/eigen/Eigen/src/Core/IndexedView.h +237 -0
  102. data/vendor/eigen/Eigen/src/Core/Inverse.h +9 -10
  103. data/vendor/eigen/Eigen/src/Core/Map.h +7 -7
  104. data/vendor/eigen/Eigen/src/Core/MapBase.h +10 -3
  105. data/vendor/eigen/Eigen/src/Core/MathFunctions.h +767 -125
  106. data/vendor/eigen/Eigen/src/Core/MathFunctionsImpl.h +118 -19
  107. data/vendor/eigen/Eigen/src/Core/Matrix.h +131 -25
  108. data/vendor/eigen/Eigen/src/Core/MatrixBase.h +21 -3
  109. data/vendor/eigen/Eigen/src/Core/NestByValue.h +25 -50
  110. data/vendor/eigen/Eigen/src/Core/NoAlias.h +4 -3
  111. data/vendor/eigen/Eigen/src/Core/NumTraits.h +107 -20
  112. data/vendor/eigen/Eigen/src/Core/PartialReduxEvaluator.h +232 -0
  113. data/vendor/eigen/Eigen/src/Core/PermutationMatrix.h +3 -31
  114. data/vendor/eigen/Eigen/src/Core/PlainObjectBase.h +152 -59
  115. data/vendor/eigen/Eigen/src/Core/Product.h +30 -25
  116. data/vendor/eigen/Eigen/src/Core/ProductEvaluators.h +192 -125
  117. data/vendor/eigen/Eigen/src/Core/Random.h +37 -1
  118. data/vendor/eigen/Eigen/src/Core/Redux.h +180 -170
  119. data/vendor/eigen/Eigen/src/Core/Ref.h +121 -23
  120. data/vendor/eigen/Eigen/src/Core/Replicate.h +8 -8
  121. data/vendor/eigen/Eigen/src/Core/Reshaped.h +454 -0
  122. data/vendor/eigen/Eigen/src/Core/ReturnByValue.h +7 -5
  123. data/vendor/eigen/Eigen/src/Core/Reverse.h +18 -12
  124. data/vendor/eigen/Eigen/src/Core/Select.h +8 -6
  125. data/vendor/eigen/Eigen/src/Core/SelfAdjointView.h +33 -20
  126. data/vendor/eigen/Eigen/src/Core/Solve.h +14 -14
  127. data/vendor/eigen/Eigen/src/Core/SolveTriangular.h +16 -16
  128. data/vendor/eigen/Eigen/src/Core/SolverBase.h +41 -3
  129. data/vendor/eigen/Eigen/src/Core/StableNorm.h +100 -70
  130. data/vendor/eigen/Eigen/src/Core/StlIterators.h +463 -0
  131. data/vendor/eigen/Eigen/src/Core/Stride.h +9 -4
  132. data/vendor/eigen/Eigen/src/Core/Swap.h +5 -4
  133. data/vendor/eigen/Eigen/src/Core/Transpose.h +88 -27
  134. data/vendor/eigen/Eigen/src/Core/Transpositions.h +26 -47
  135. data/vendor/eigen/Eigen/src/Core/TriangularMatrix.h +93 -75
  136. data/vendor/eigen/Eigen/src/Core/VectorBlock.h +5 -5
  137. data/vendor/eigen/Eigen/src/Core/VectorwiseOp.h +159 -70
  138. data/vendor/eigen/Eigen/src/Core/Visitor.h +137 -29
  139. data/vendor/eigen/Eigen/src/Core/arch/AVX/Complex.h +50 -129
  140. data/vendor/eigen/Eigen/src/Core/arch/AVX/MathFunctions.h +126 -337
  141. data/vendor/eigen/Eigen/src/Core/arch/AVX/PacketMath.h +1092 -155
  142. data/vendor/eigen/Eigen/src/Core/arch/AVX/TypeCasting.h +65 -1
  143. data/vendor/eigen/Eigen/src/Core/arch/AVX512/Complex.h +422 -0
  144. data/vendor/eigen/Eigen/src/Core/arch/AVX512/MathFunctions.h +207 -236
  145. data/vendor/eigen/Eigen/src/Core/arch/AVX512/PacketMath.h +1482 -495
  146. data/vendor/eigen/Eigen/src/Core/arch/AVX512/TypeCasting.h +89 -0
  147. data/vendor/eigen/Eigen/src/Core/arch/AltiVec/Complex.h +152 -165
  148. data/vendor/eigen/Eigen/src/Core/arch/AltiVec/MathFunctions.h +19 -251
  149. data/vendor/eigen/Eigen/src/Core/arch/AltiVec/MatrixProduct.h +2937 -0
  150. data/vendor/eigen/Eigen/src/Core/arch/AltiVec/MatrixProductCommon.h +221 -0
  151. data/vendor/eigen/Eigen/src/Core/arch/AltiVec/MatrixProductMMA.h +629 -0
  152. data/vendor/eigen/Eigen/src/Core/arch/AltiVec/PacketMath.h +2042 -392
  153. data/vendor/eigen/Eigen/src/Core/arch/CUDA/Complex.h +235 -80
  154. data/vendor/eigen/Eigen/src/Core/arch/Default/BFloat16.h +700 -0
  155. data/vendor/eigen/Eigen/src/Core/arch/Default/ConjHelper.h +102 -14
  156. data/vendor/eigen/Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h +1649 -0
  157. data/vendor/eigen/Eigen/src/Core/arch/Default/GenericPacketMathFunctionsFwd.h +110 -0
  158. data/vendor/eigen/Eigen/src/Core/arch/Default/Half.h +942 -0
  159. data/vendor/eigen/Eigen/src/Core/arch/Default/Settings.h +1 -1
  160. data/vendor/eigen/Eigen/src/Core/arch/Default/TypeCasting.h +120 -0
  161. data/vendor/eigen/Eigen/src/Core/arch/{CUDA → GPU}/MathFunctions.h +16 -4
  162. data/vendor/eigen/Eigen/src/Core/arch/GPU/PacketMath.h +1685 -0
  163. data/vendor/eigen/Eigen/src/Core/arch/GPU/TypeCasting.h +80 -0
  164. data/vendor/eigen/Eigen/src/Core/arch/HIP/hcc/math_constants.h +23 -0
  165. data/vendor/eigen/Eigen/src/Core/arch/MSA/Complex.h +648 -0
  166. data/vendor/eigen/Eigen/src/Core/arch/MSA/MathFunctions.h +387 -0
  167. data/vendor/eigen/Eigen/src/Core/arch/MSA/PacketMath.h +1233 -0
  168. data/vendor/eigen/Eigen/src/Core/arch/NEON/Complex.h +313 -219
  169. data/vendor/eigen/Eigen/src/Core/arch/NEON/GeneralBlockPanelKernel.h +183 -0
  170. data/vendor/eigen/Eigen/src/Core/arch/NEON/MathFunctions.h +54 -70
  171. data/vendor/eigen/Eigen/src/Core/arch/NEON/PacketMath.h +4376 -549
  172. data/vendor/eigen/Eigen/src/Core/arch/NEON/TypeCasting.h +1419 -0
  173. data/vendor/eigen/Eigen/src/Core/arch/SSE/Complex.h +59 -179
  174. data/vendor/eigen/Eigen/src/Core/arch/SSE/MathFunctions.h +65 -428
  175. data/vendor/eigen/Eigen/src/Core/arch/SSE/PacketMath.h +893 -283
  176. data/vendor/eigen/Eigen/src/Core/arch/SSE/TypeCasting.h +65 -0
  177. data/vendor/eigen/Eigen/src/Core/arch/SVE/MathFunctions.h +44 -0
  178. data/vendor/eigen/Eigen/src/Core/arch/SVE/PacketMath.h +752 -0
  179. data/vendor/eigen/Eigen/src/Core/arch/SVE/TypeCasting.h +49 -0
  180. data/vendor/eigen/Eigen/src/Core/arch/SYCL/InteropHeaders.h +232 -0
  181. data/vendor/eigen/Eigen/src/Core/arch/SYCL/MathFunctions.h +301 -0
  182. data/vendor/eigen/Eigen/src/Core/arch/SYCL/PacketMath.h +670 -0
  183. data/vendor/eigen/Eigen/src/Core/arch/SYCL/SyclMemoryModel.h +694 -0
  184. data/vendor/eigen/Eigen/src/Core/arch/SYCL/TypeCasting.h +85 -0
  185. data/vendor/eigen/Eigen/src/Core/arch/ZVector/Complex.h +212 -183
  186. data/vendor/eigen/Eigen/src/Core/arch/ZVector/MathFunctions.h +101 -5
  187. data/vendor/eigen/Eigen/src/Core/arch/ZVector/PacketMath.h +510 -395
  188. data/vendor/eigen/Eigen/src/Core/functors/AssignmentFunctors.h +11 -2
  189. data/vendor/eigen/Eigen/src/Core/functors/BinaryFunctors.h +112 -46
  190. data/vendor/eigen/Eigen/src/Core/functors/NullaryFunctors.h +31 -30
  191. data/vendor/eigen/Eigen/src/Core/functors/StlFunctors.h +32 -2
  192. data/vendor/eigen/Eigen/src/Core/functors/UnaryFunctors.h +355 -16
  193. data/vendor/eigen/Eigen/src/Core/products/GeneralBlockPanelKernel.h +1075 -586
  194. data/vendor/eigen/Eigen/src/Core/products/GeneralMatrixMatrix.h +49 -24
  195. data/vendor/eigen/Eigen/src/Core/products/GeneralMatrixMatrixTriangular.h +41 -35
  196. data/vendor/eigen/Eigen/src/Core/products/GeneralMatrixMatrixTriangular_BLAS.h +6 -6
  197. data/vendor/eigen/Eigen/src/Core/products/GeneralMatrixMatrix_BLAS.h +4 -2
  198. data/vendor/eigen/Eigen/src/Core/products/GeneralMatrixVector.h +382 -483
  199. data/vendor/eigen/Eigen/src/Core/products/Parallelizer.h +22 -5
  200. data/vendor/eigen/Eigen/src/Core/products/SelfadjointMatrixMatrix.h +53 -30
  201. data/vendor/eigen/Eigen/src/Core/products/SelfadjointMatrixMatrix_BLAS.h +16 -8
  202. data/vendor/eigen/Eigen/src/Core/products/SelfadjointMatrixVector.h +8 -6
  203. data/vendor/eigen/Eigen/src/Core/products/SelfadjointProduct.h +4 -4
  204. data/vendor/eigen/Eigen/src/Core/products/SelfadjointRank2Update.h +5 -4
  205. data/vendor/eigen/Eigen/src/Core/products/TriangularMatrixMatrix.h +33 -27
  206. data/vendor/eigen/Eigen/src/Core/products/TriangularMatrixMatrix_BLAS.h +14 -12
  207. data/vendor/eigen/Eigen/src/Core/products/TriangularSolverMatrix.h +36 -34
  208. data/vendor/eigen/Eigen/src/Core/products/TriangularSolverMatrix_BLAS.h +8 -4
  209. data/vendor/eigen/Eigen/src/Core/products/TriangularSolverVector.h +13 -10
  210. data/vendor/eigen/Eigen/src/Core/util/BlasUtil.h +304 -119
  211. data/vendor/eigen/Eigen/src/Core/util/ConfigureVectorization.h +512 -0
  212. data/vendor/eigen/Eigen/src/Core/util/Constants.h +25 -9
  213. data/vendor/eigen/Eigen/src/Core/util/DisableStupidWarnings.h +26 -3
  214. data/vendor/eigen/Eigen/src/Core/util/ForwardDeclarations.h +29 -9
  215. data/vendor/eigen/Eigen/src/Core/util/IndexedViewHelper.h +186 -0
  216. data/vendor/eigen/Eigen/src/Core/util/IntegralConstant.h +272 -0
  217. data/vendor/eigen/Eigen/src/Core/util/MKL_support.h +8 -1
  218. data/vendor/eigen/Eigen/src/Core/util/Macros.h +709 -246
  219. data/vendor/eigen/Eigen/src/Core/util/Memory.h +222 -52
  220. data/vendor/eigen/Eigen/src/Core/util/Meta.h +355 -77
  221. data/vendor/eigen/Eigen/src/Core/util/ReenableStupidWarnings.h +5 -1
  222. data/vendor/eigen/Eigen/src/Core/util/ReshapedHelper.h +51 -0
  223. data/vendor/eigen/Eigen/src/Core/util/StaticAssert.h +8 -5
  224. data/vendor/eigen/Eigen/src/Core/util/SymbolicIndex.h +293 -0
  225. data/vendor/eigen/Eigen/src/Core/util/XprHelper.h +65 -30
  226. data/vendor/eigen/Eigen/src/Eigenvalues/ComplexEigenSolver.h +1 -1
  227. data/vendor/eigen/Eigen/src/Eigenvalues/ComplexSchur.h +7 -4
  228. data/vendor/eigen/Eigen/src/Eigenvalues/EigenSolver.h +2 -2
  229. data/vendor/eigen/Eigen/src/Eigenvalues/GeneralizedSelfAdjointEigenSolver.h +1 -1
  230. data/vendor/eigen/Eigen/src/Eigenvalues/HessenbergDecomposition.h +2 -2
  231. data/vendor/eigen/Eigen/src/Eigenvalues/MatrixBaseEigenvalues.h +2 -2
  232. data/vendor/eigen/Eigen/src/Eigenvalues/RealQZ.h +9 -6
  233. data/vendor/eigen/Eigen/src/Eigenvalues/RealSchur.h +21 -9
  234. data/vendor/eigen/Eigen/src/Eigenvalues/SelfAdjointEigenSolver.h +77 -43
  235. data/vendor/eigen/Eigen/src/Eigenvalues/Tridiagonalization.h +20 -15
  236. data/vendor/eigen/Eigen/src/Geometry/AlignedBox.h +99 -5
  237. data/vendor/eigen/Eigen/src/Geometry/AngleAxis.h +4 -4
  238. data/vendor/eigen/Eigen/src/Geometry/EulerAngles.h +3 -3
  239. data/vendor/eigen/Eigen/src/Geometry/Homogeneous.h +15 -11
  240. data/vendor/eigen/Eigen/src/Geometry/Hyperplane.h +1 -1
  241. data/vendor/eigen/Eigen/src/Geometry/OrthoMethods.h +3 -2
  242. data/vendor/eigen/Eigen/src/Geometry/ParametrizedLine.h +39 -2
  243. data/vendor/eigen/Eigen/src/Geometry/Quaternion.h +70 -14
  244. data/vendor/eigen/Eigen/src/Geometry/Rotation2D.h +3 -3
  245. data/vendor/eigen/Eigen/src/Geometry/Scaling.h +23 -5
  246. data/vendor/eigen/Eigen/src/Geometry/Transform.h +88 -67
  247. data/vendor/eigen/Eigen/src/Geometry/Translation.h +6 -12
  248. data/vendor/eigen/Eigen/src/Geometry/Umeyama.h +1 -1
  249. data/vendor/eigen/Eigen/src/Geometry/arch/Geometry_SIMD.h +168 -0
  250. data/vendor/eigen/Eigen/src/Householder/BlockHouseholder.h +9 -2
  251. data/vendor/eigen/Eigen/src/Householder/Householder.h +8 -4
  252. data/vendor/eigen/Eigen/src/Householder/HouseholderSequence.h +123 -48
  253. data/vendor/eigen/Eigen/src/IterativeLinearSolvers/BasicPreconditioners.h +15 -15
  254. data/vendor/eigen/Eigen/src/IterativeLinearSolvers/BiCGSTAB.h +7 -23
  255. data/vendor/eigen/Eigen/src/IterativeLinearSolvers/ConjugateGradient.h +5 -22
  256. data/vendor/eigen/Eigen/src/IterativeLinearSolvers/IncompleteCholesky.h +41 -47
  257. data/vendor/eigen/Eigen/src/IterativeLinearSolvers/IncompleteLUT.h +51 -60
  258. data/vendor/eigen/Eigen/src/IterativeLinearSolvers/IterativeSolverBase.h +70 -20
  259. data/vendor/eigen/Eigen/src/IterativeLinearSolvers/LeastSquareConjugateGradient.h +2 -20
  260. data/vendor/eigen/Eigen/src/IterativeLinearSolvers/SolveWithGuess.h +11 -9
  261. data/vendor/eigen/Eigen/src/Jacobi/Jacobi.h +31 -10
  262. data/vendor/eigen/Eigen/src/KLUSupport/KLUSupport.h +358 -0
  263. data/vendor/eigen/Eigen/src/LU/Determinant.h +35 -19
  264. data/vendor/eigen/Eigen/src/LU/FullPivLU.h +29 -43
  265. data/vendor/eigen/Eigen/src/LU/InverseImpl.h +25 -8
  266. data/vendor/eigen/Eigen/src/LU/PartialPivLU.h +71 -58
  267. data/vendor/eigen/Eigen/src/LU/arch/InverseSize4.h +351 -0
  268. data/vendor/eigen/Eigen/src/OrderingMethods/Amd.h +7 -17
  269. data/vendor/eigen/Eigen/src/OrderingMethods/Eigen_Colamd.h +297 -277
  270. data/vendor/eigen/Eigen/src/OrderingMethods/Ordering.h +6 -10
  271. data/vendor/eigen/Eigen/src/PaStiXSupport/PaStiXSupport.h +1 -1
  272. data/vendor/eigen/Eigen/src/PardisoSupport/PardisoSupport.h +9 -7
  273. data/vendor/eigen/Eigen/src/QR/ColPivHouseholderQR.h +41 -20
  274. data/vendor/eigen/Eigen/src/QR/CompleteOrthogonalDecomposition.h +100 -27
  275. data/vendor/eigen/Eigen/src/QR/FullPivHouseholderQR.h +59 -22
  276. data/vendor/eigen/Eigen/src/QR/HouseholderQR.h +48 -23
  277. data/vendor/eigen/Eigen/src/SPQRSupport/SuiteSparseQRSupport.h +25 -3
  278. data/vendor/eigen/Eigen/src/SVD/BDCSVD.h +183 -63
  279. data/vendor/eigen/Eigen/src/SVD/JacobiSVD.h +22 -14
  280. data/vendor/eigen/Eigen/src/SVD/SVDBase.h +83 -22
  281. data/vendor/eigen/Eigen/src/SVD/UpperBidiagonalization.h +3 -3
  282. data/vendor/eigen/Eigen/src/SparseCholesky/SimplicialCholesky.h +17 -9
  283. data/vendor/eigen/Eigen/src/SparseCholesky/SimplicialCholesky_impl.h +12 -37
  284. data/vendor/eigen/Eigen/src/SparseCore/AmbiVector.h +3 -2
  285. data/vendor/eigen/Eigen/src/SparseCore/CompressedStorage.h +16 -0
  286. data/vendor/eigen/Eigen/src/SparseCore/ConservativeSparseSparseProduct.h +6 -6
  287. data/vendor/eigen/Eigen/src/SparseCore/SparseAssign.h +81 -27
  288. data/vendor/eigen/Eigen/src/SparseCore/SparseBlock.h +25 -57
  289. data/vendor/eigen/Eigen/src/SparseCore/SparseCompressedBase.h +40 -11
  290. data/vendor/eigen/Eigen/src/SparseCore/SparseCwiseBinaryOp.h +11 -15
  291. data/vendor/eigen/Eigen/src/SparseCore/SparseCwiseUnaryOp.h +4 -2
  292. data/vendor/eigen/Eigen/src/SparseCore/SparseDenseProduct.h +30 -8
  293. data/vendor/eigen/Eigen/src/SparseCore/SparseMatrix.h +126 -11
  294. data/vendor/eigen/Eigen/src/SparseCore/SparseMatrixBase.h +5 -12
  295. data/vendor/eigen/Eigen/src/SparseCore/SparseProduct.h +13 -1
  296. data/vendor/eigen/Eigen/src/SparseCore/SparseRef.h +7 -7
  297. data/vendor/eigen/Eigen/src/SparseCore/SparseSelfAdjointView.h +5 -2
  298. data/vendor/eigen/Eigen/src/SparseCore/SparseUtil.h +8 -0
  299. data/vendor/eigen/Eigen/src/SparseCore/SparseVector.h +1 -1
  300. data/vendor/eigen/Eigen/src/SparseCore/SparseView.h +1 -0
  301. data/vendor/eigen/Eigen/src/SparseLU/SparseLU.h +162 -12
  302. data/vendor/eigen/Eigen/src/SparseLU/SparseLU_Memory.h +1 -1
  303. data/vendor/eigen/Eigen/src/SparseLU/SparseLU_SupernodalMatrix.h +76 -2
  304. data/vendor/eigen/Eigen/src/SparseLU/SparseLU_column_dfs.h +2 -2
  305. data/vendor/eigen/Eigen/src/SparseLU/SparseLU_gemm_kernel.h +1 -1
  306. data/vendor/eigen/Eigen/src/SparseLU/SparseLU_panel_bmod.h +1 -1
  307. data/vendor/eigen/Eigen/src/SparseQR/SparseQR.h +19 -6
  308. data/vendor/eigen/Eigen/src/StlSupport/StdDeque.h +2 -12
  309. data/vendor/eigen/Eigen/src/StlSupport/StdList.h +2 -2
  310. data/vendor/eigen/Eigen/src/StlSupport/StdVector.h +2 -2
  311. data/vendor/eigen/Eigen/src/SuperLUSupport/SuperLUSupport.h +6 -8
  312. data/vendor/eigen/Eigen/src/UmfPackSupport/UmfPackSupport.h +175 -39
  313. data/vendor/eigen/Eigen/src/misc/lapacke.h +5 -4
  314. data/vendor/eigen/Eigen/src/plugins/ArrayCwiseBinaryOps.h +28 -2
  315. data/vendor/eigen/Eigen/src/plugins/ArrayCwiseUnaryOps.h +155 -11
  316. data/vendor/eigen/Eigen/src/plugins/BlockMethods.h +626 -242
  317. data/vendor/eigen/Eigen/src/plugins/CommonCwiseUnaryOps.h +14 -0
  318. data/vendor/eigen/Eigen/src/plugins/IndexedViewMethods.h +262 -0
  319. data/vendor/eigen/Eigen/src/plugins/MatrixCwiseBinaryOps.h +4 -4
  320. data/vendor/eigen/Eigen/src/plugins/MatrixCwiseUnaryOps.h +10 -0
  321. data/vendor/eigen/Eigen/src/plugins/ReshapedMethods.h +149 -0
  322. data/vendor/eigen/README.md +2 -0
  323. data/vendor/eigen/bench/btl/README +1 -1
  324. data/vendor/eigen/bench/tensors/README +6 -7
  325. data/vendor/eigen/ci/README.md +56 -0
  326. data/vendor/eigen/demos/mix_eigen_and_c/README +1 -1
  327. data/vendor/eigen/unsupported/Eigen/CXX11/src/Tensor/README.md +213 -158
  328. data/vendor/eigen/unsupported/README.txt +1 -1
  329. data/vendor/tomotopy/README.kr.rst +78 -0
  330. data/vendor/tomotopy/README.rst +75 -0
  331. data/vendor/tomotopy/src/Labeling/FoRelevance.cpp +2 -2
  332. data/vendor/tomotopy/src/Labeling/Phraser.hpp +4 -4
  333. data/vendor/tomotopy/src/TopicModel/CTModel.hpp +7 -3
  334. data/vendor/tomotopy/src/TopicModel/DMRModel.hpp +7 -3
  335. data/vendor/tomotopy/src/TopicModel/DTModel.hpp +6 -3
  336. data/vendor/tomotopy/src/TopicModel/GDMRModel.hpp +2 -2
  337. data/vendor/tomotopy/src/TopicModel/HDP.h +1 -0
  338. data/vendor/tomotopy/src/TopicModel/HDPModel.hpp +57 -6
  339. data/vendor/tomotopy/src/TopicModel/HLDAModel.hpp +6 -3
  340. data/vendor/tomotopy/src/TopicModel/HPAModel.hpp +3 -2
  341. data/vendor/tomotopy/src/TopicModel/LDA.h +3 -3
  342. data/vendor/tomotopy/src/TopicModel/LDACVB0Model.hpp +5 -5
  343. data/vendor/tomotopy/src/TopicModel/LDAModel.hpp +50 -19
  344. data/vendor/tomotopy/src/TopicModel/LLDAModel.hpp +6 -2
  345. data/vendor/tomotopy/src/TopicModel/MGLDAModel.hpp +3 -2
  346. data/vendor/tomotopy/src/TopicModel/PAModel.hpp +1 -1
  347. data/vendor/tomotopy/src/TopicModel/PLDAModel.hpp +6 -2
  348. data/vendor/tomotopy/src/TopicModel/PT.h +3 -1
  349. data/vendor/tomotopy/src/TopicModel/PTModel.hpp +36 -3
  350. data/vendor/tomotopy/src/TopicModel/SLDAModel.hpp +6 -3
  351. data/vendor/tomotopy/src/TopicModel/TopicModel.hpp +55 -26
  352. data/vendor/tomotopy/src/Utils/AliasMethod.hpp +5 -4
  353. data/vendor/tomotopy/src/Utils/Dictionary.h +2 -2
  354. data/vendor/tomotopy/src/Utils/EigenAddonOps.hpp +36 -1
  355. data/vendor/tomotopy/src/Utils/MultiNormalDistribution.hpp +1 -1
  356. data/vendor/tomotopy/src/Utils/TruncMultiNormal.hpp +1 -1
  357. data/vendor/tomotopy/src/Utils/exception.h +6 -0
  358. data/vendor/tomotopy/src/Utils/math.h +2 -2
  359. data/vendor/tomotopy/src/Utils/sample.hpp +14 -12
  360. data/vendor/tomotopy/src/Utils/serializer.hpp +30 -5
  361. data/vendor/tomotopy/src/Utils/sse_gamma.h +0 -3
  362. metadata +64 -18
  363. data/vendor/eigen/Eigen/CMakeLists.txt +0 -19
  364. data/vendor/eigen/Eigen/src/Core/arch/CUDA/Half.h +0 -674
  365. data/vendor/eigen/Eigen/src/Core/arch/CUDA/PacketMath.h +0 -333
  366. data/vendor/eigen/Eigen/src/Core/arch/CUDA/PacketMathHalf.h +0 -1124
  367. data/vendor/eigen/Eigen/src/Core/arch/CUDA/TypeCasting.h +0 -212
  368. data/vendor/eigen/Eigen/src/Geometry/arch/Geometry_SSE.h +0 -161
  369. data/vendor/eigen/Eigen/src/LU/arch/Inverse_SSE.h +0 -338
@@ -1,1124 +0,0 @@
1
- // This file is part of Eigen, a lightweight C++ template library
2
- // for linear algebra.
3
- //
4
- // Copyright (C) 2016 Benoit Steiner <benoit.steiner.goog@gmail.com>
5
- //
6
- // This Source Code Form is subject to the terms of the Mozilla
7
- // Public License v. 2.0. If a copy of the MPL was not distributed
8
- // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9
-
10
- #ifndef EIGEN_PACKET_MATH_HALF_CUDA_H
11
- #define EIGEN_PACKET_MATH_HALF_CUDA_H
12
-
13
-
14
- namespace Eigen {
15
- namespace internal {
16
-
17
- // Most of the following operations require arch >= 3.0
18
- #if defined(EIGEN_HAS_CUDA_FP16) && defined(__CUDACC__) && defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 300
19
-
20
- template<> struct is_arithmetic<half2> { enum { value = true }; };
21
-
22
- template<> struct packet_traits<Eigen::half> : default_packet_traits
23
- {
24
- typedef half2 type;
25
- typedef half2 half;
26
- enum {
27
- Vectorizable = 1,
28
- AlignedOnScalar = 1,
29
- size=2,
30
- HasHalfPacket = 0,
31
- HasAdd = 1,
32
- HasMul = 1,
33
- HasDiv = 1,
34
- HasSqrt = 1,
35
- HasRsqrt = 1,
36
- HasExp = 1,
37
- HasLog = 1,
38
- HasLog1p = 1
39
- };
40
- };
41
-
42
- template<> struct unpacket_traits<half2> { typedef Eigen::half type; enum {size=2, alignment=Aligned16}; typedef half2 half; };
43
-
44
- template<> __device__ EIGEN_STRONG_INLINE half2 pset1<half2>(const Eigen::half& from) {
45
- return __half2half2(from);
46
- }
47
-
48
- template<> __device__ EIGEN_STRONG_INLINE half2 pload<half2>(const Eigen::half* from) {
49
- return *reinterpret_cast<const half2*>(from);
50
- }
51
-
52
- template<> __device__ EIGEN_STRONG_INLINE half2 ploadu<half2>(const Eigen::half* from) {
53
- return __halves2half2(from[0], from[1]);
54
- }
55
-
56
- template<> EIGEN_STRONG_INLINE half2 ploaddup<half2>(const Eigen::half* from) {
57
- return __halves2half2(from[0], from[0]);
58
- }
59
-
60
- template<> __device__ EIGEN_STRONG_INLINE void pstore<Eigen::half>(Eigen::half* to, const half2& from) {
61
- *reinterpret_cast<half2*>(to) = from;
62
- }
63
-
64
- template<> __device__ EIGEN_STRONG_INLINE void pstoreu<Eigen::half>(Eigen::half* to, const half2& from) {
65
- to[0] = __low2half(from);
66
- to[1] = __high2half(from);
67
- }
68
-
69
- template<>
70
- __device__ EIGEN_ALWAYS_INLINE half2 ploadt_ro<half2, Aligned>(const Eigen::half* from) {
71
- #if __CUDA_ARCH__ >= 350
72
- return __ldg((const half2*)from);
73
- #else
74
- return __halves2half2(*(from+0), *(from+1));
75
- #endif
76
- }
77
-
78
- template<>
79
- __device__ EIGEN_ALWAYS_INLINE half2 ploadt_ro<half2, Unaligned>(const Eigen::half* from) {
80
- #if __CUDA_ARCH__ >= 350
81
- return __halves2half2(__ldg(from+0), __ldg(from+1));
82
- #else
83
- return __halves2half2(*(from+0), *(from+1));
84
- #endif
85
- }
86
-
87
- template<> __device__ EIGEN_STRONG_INLINE half2 pgather<Eigen::half, half2>(const Eigen::half* from, Index stride) {
88
- return __halves2half2(from[0*stride], from[1*stride]);
89
- }
90
-
91
- template<> __device__ EIGEN_STRONG_INLINE void pscatter<Eigen::half, half2>(Eigen::half* to, const half2& from, Index stride) {
92
- to[stride*0] = __low2half(from);
93
- to[stride*1] = __high2half(from);
94
- }
95
-
96
- template<> __device__ EIGEN_STRONG_INLINE Eigen::half pfirst<half2>(const half2& a) {
97
- return __low2half(a);
98
- }
99
-
100
- template<> __device__ EIGEN_STRONG_INLINE half2 pabs<half2>(const half2& a) {
101
- half2 result;
102
- unsigned temp = *(reinterpret_cast<const unsigned*>(&(a)));
103
- *(reinterpret_cast<unsigned*>(&(result))) = temp & 0x7FFF7FFF;
104
- return result;
105
- }
106
-
107
-
108
- __device__ EIGEN_STRONG_INLINE void
109
- ptranspose(PacketBlock<half2,2>& kernel) {
110
- __half a1 = __low2half(kernel.packet[0]);
111
- __half a2 = __high2half(kernel.packet[0]);
112
- __half b1 = __low2half(kernel.packet[1]);
113
- __half b2 = __high2half(kernel.packet[1]);
114
- kernel.packet[0] = __halves2half2(a1, b1);
115
- kernel.packet[1] = __halves2half2(a2, b2);
116
- }
117
-
118
- template<> __device__ EIGEN_STRONG_INLINE half2 plset<half2>(const Eigen::half& a) {
119
- #if __CUDA_ARCH__ >= 530
120
- return __halves2half2(a, __hadd(a, __float2half(1.0f)));
121
- #else
122
- float f = __half2float(a) + 1.0f;
123
- return __halves2half2(a, __float2half(f));
124
- #endif
125
- }
126
-
127
- template<> __device__ EIGEN_STRONG_INLINE half2 padd<half2>(const half2& a, const half2& b) {
128
- #if __CUDA_ARCH__ >= 530
129
- return __hadd2(a, b);
130
- #else
131
- float a1 = __low2float(a);
132
- float a2 = __high2float(a);
133
- float b1 = __low2float(b);
134
- float b2 = __high2float(b);
135
- float r1 = a1 + b1;
136
- float r2 = a2 + b2;
137
- return __floats2half2_rn(r1, r2);
138
- #endif
139
- }
140
-
141
- template<> __device__ EIGEN_STRONG_INLINE half2 psub<half2>(const half2& a, const half2& b) {
142
- #if __CUDA_ARCH__ >= 530
143
- return __hsub2(a, b);
144
- #else
145
- float a1 = __low2float(a);
146
- float a2 = __high2float(a);
147
- float b1 = __low2float(b);
148
- float b2 = __high2float(b);
149
- float r1 = a1 - b1;
150
- float r2 = a2 - b2;
151
- return __floats2half2_rn(r1, r2);
152
- #endif
153
- }
154
-
155
- template<> __device__ EIGEN_STRONG_INLINE half2 pnegate(const half2& a) {
156
- #if __CUDA_ARCH__ >= 530
157
- return __hneg2(a);
158
- #else
159
- float a1 = __low2float(a);
160
- float a2 = __high2float(a);
161
- return __floats2half2_rn(-a1, -a2);
162
- #endif
163
- }
164
-
165
- template<> __device__ EIGEN_STRONG_INLINE half2 pconj(const half2& a) { return a; }
166
-
167
- template<> __device__ EIGEN_STRONG_INLINE half2 pmul<half2>(const half2& a, const half2& b) {
168
- #if __CUDA_ARCH__ >= 530
169
- return __hmul2(a, b);
170
- #else
171
- float a1 = __low2float(a);
172
- float a2 = __high2float(a);
173
- float b1 = __low2float(b);
174
- float b2 = __high2float(b);
175
- float r1 = a1 * b1;
176
- float r2 = a2 * b2;
177
- return __floats2half2_rn(r1, r2);
178
- #endif
179
- }
180
-
181
- template<> __device__ EIGEN_STRONG_INLINE half2 pmadd<half2>(const half2& a, const half2& b, const half2& c) {
182
- #if __CUDA_ARCH__ >= 530
183
- return __hfma2(a, b, c);
184
- #else
185
- float a1 = __low2float(a);
186
- float a2 = __high2float(a);
187
- float b1 = __low2float(b);
188
- float b2 = __high2float(b);
189
- float c1 = __low2float(c);
190
- float c2 = __high2float(c);
191
- float r1 = a1 * b1 + c1;
192
- float r2 = a2 * b2 + c2;
193
- return __floats2half2_rn(r1, r2);
194
- #endif
195
- }
196
-
197
- template<> __device__ EIGEN_STRONG_INLINE half2 pdiv<half2>(const half2& a, const half2& b) {
198
- float a1 = __low2float(a);
199
- float a2 = __high2float(a);
200
- float b1 = __low2float(b);
201
- float b2 = __high2float(b);
202
- float r1 = a1 / b1;
203
- float r2 = a2 / b2;
204
- return __floats2half2_rn(r1, r2);
205
- }
206
-
207
- template<> __device__ EIGEN_STRONG_INLINE half2 pmin<half2>(const half2& a, const half2& b) {
208
- float a1 = __low2float(a);
209
- float a2 = __high2float(a);
210
- float b1 = __low2float(b);
211
- float b2 = __high2float(b);
212
- __half r1 = a1 < b1 ? __low2half(a) : __low2half(b);
213
- __half r2 = a2 < b2 ? __high2half(a) : __high2half(b);
214
- return __halves2half2(r1, r2);
215
- }
216
-
217
- template<> __device__ EIGEN_STRONG_INLINE half2 pmax<half2>(const half2& a, const half2& b) {
218
- float a1 = __low2float(a);
219
- float a2 = __high2float(a);
220
- float b1 = __low2float(b);
221
- float b2 = __high2float(b);
222
- __half r1 = a1 > b1 ? __low2half(a) : __low2half(b);
223
- __half r2 = a2 > b2 ? __high2half(a) : __high2half(b);
224
- return __halves2half2(r1, r2);
225
- }
226
-
227
- template<> __device__ EIGEN_STRONG_INLINE Eigen::half predux<half2>(const half2& a) {
228
- #if __CUDA_ARCH__ >= 530
229
- return __hadd(__low2half(a), __high2half(a));
230
- #else
231
- float a1 = __low2float(a);
232
- float a2 = __high2float(a);
233
- return Eigen::half(half_impl::raw_uint16_to_half(__float2half_rn(a1 + a2)));
234
- #endif
235
- }
236
-
237
- template<> __device__ EIGEN_STRONG_INLINE Eigen::half predux_max<half2>(const half2& a) {
238
- #if __CUDA_ARCH__ >= 530
239
- __half first = __low2half(a);
240
- __half second = __high2half(a);
241
- return __hgt(first, second) ? first : second;
242
- #else
243
- float a1 = __low2float(a);
244
- float a2 = __high2float(a);
245
- return a1 > a2 ? __low2half(a) : __high2half(a);
246
- #endif
247
- }
248
-
249
- template<> __device__ EIGEN_STRONG_INLINE Eigen::half predux_min<half2>(const half2& a) {
250
- #if __CUDA_ARCH__ >= 530
251
- __half first = __low2half(a);
252
- __half second = __high2half(a);
253
- return __hlt(first, second) ? first : second;
254
- #else
255
- float a1 = __low2float(a);
256
- float a2 = __high2float(a);
257
- return a1 < a2 ? __low2half(a) : __high2half(a);
258
- #endif
259
- }
260
-
261
- template<> __device__ EIGEN_STRONG_INLINE Eigen::half predux_mul<half2>(const half2& a) {
262
- #if __CUDA_ARCH__ >= 530
263
- return __hmul(__low2half(a), __high2half(a));
264
- #else
265
- float a1 = __low2float(a);
266
- float a2 = __high2float(a);
267
- return Eigen::half(half_impl::raw_uint16_to_half(__float2half_rn(a1 * a2)));
268
- #endif
269
- }
270
-
271
- template<> __device__ EIGEN_STRONG_INLINE half2 plog1p<half2>(const half2& a) {
272
- float a1 = __low2float(a);
273
- float a2 = __high2float(a);
274
- float r1 = log1pf(a1);
275
- float r2 = log1pf(a2);
276
- return __floats2half2_rn(r1, r2);
277
- }
278
-
279
- #if EIGEN_CUDACC_VER >= 80000 && defined EIGEN_CUDA_ARCH && EIGEN_CUDA_ARCH >= 530
280
-
281
- template<> __device__ EIGEN_STRONG_INLINE
282
- half2 plog<half2>(const half2& a) {
283
- return h2log(a);
284
- }
285
-
286
- template<> __device__ EIGEN_STRONG_INLINE
287
- half2 pexp<half2>(const half2& a) {
288
- return h2exp(a);
289
- }
290
-
291
- template<> __device__ EIGEN_STRONG_INLINE
292
- half2 psqrt<half2>(const half2& a) {
293
- return h2sqrt(a);
294
- }
295
-
296
- template<> __device__ EIGEN_STRONG_INLINE
297
- half2 prsqrt<half2>(const half2& a) {
298
- return h2rsqrt(a);
299
- }
300
-
301
- #else
302
-
303
- template<> __device__ EIGEN_STRONG_INLINE half2 plog<half2>(const half2& a) {
304
- float a1 = __low2float(a);
305
- float a2 = __high2float(a);
306
- float r1 = logf(a1);
307
- float r2 = logf(a2);
308
- return __floats2half2_rn(r1, r2);
309
- }
310
-
311
- template<> __device__ EIGEN_STRONG_INLINE half2 pexp<half2>(const half2& a) {
312
- float a1 = __low2float(a);
313
- float a2 = __high2float(a);
314
- float r1 = expf(a1);
315
- float r2 = expf(a2);
316
- return __floats2half2_rn(r1, r2);
317
- }
318
-
319
- template<> __device__ EIGEN_STRONG_INLINE half2 psqrt<half2>(const half2& a) {
320
- float a1 = __low2float(a);
321
- float a2 = __high2float(a);
322
- float r1 = sqrtf(a1);
323
- float r2 = sqrtf(a2);
324
- return __floats2half2_rn(r1, r2);
325
- }
326
-
327
- template<> __device__ EIGEN_STRONG_INLINE half2 prsqrt<half2>(const half2& a) {
328
- float a1 = __low2float(a);
329
- float a2 = __high2float(a);
330
- float r1 = rsqrtf(a1);
331
- float r2 = rsqrtf(a2);
332
- return __floats2half2_rn(r1, r2);
333
- }
334
-
335
- #endif
336
-
337
- #elif defined EIGEN_VECTORIZE_AVX512
338
-
339
- typedef struct {
340
- __m256i x;
341
- } Packet16h;
342
-
343
-
344
- template<> struct is_arithmetic<Packet16h> { enum { value = true }; };
345
-
346
- template <>
347
- struct packet_traits<half> : default_packet_traits {
348
- typedef Packet16h type;
349
- // There is no half-size packet for Packet16h.
350
- typedef Packet16h half;
351
- enum {
352
- Vectorizable = 1,
353
- AlignedOnScalar = 1,
354
- size = 16,
355
- HasHalfPacket = 0,
356
- HasAdd = 0,
357
- HasSub = 0,
358
- HasMul = 0,
359
- HasNegate = 0,
360
- HasAbs = 0,
361
- HasAbs2 = 0,
362
- HasMin = 0,
363
- HasMax = 0,
364
- HasConj = 0,
365
- HasSetLinear = 0,
366
- HasDiv = 0,
367
- HasSqrt = 0,
368
- HasRsqrt = 0,
369
- HasExp = 0,
370
- HasLog = 0,
371
- HasBlend = 0
372
- };
373
- };
374
-
375
-
376
- template<> struct unpacket_traits<Packet16h> { typedef Eigen::half type; enum {size=16, alignment=Aligned32}; typedef Packet16h half; };
377
-
378
- template<> EIGEN_STRONG_INLINE Packet16h pset1<Packet16h>(const Eigen::half& from) {
379
- Packet16h result;
380
- result.x = _mm256_set1_epi16(from.x);
381
- return result;
382
- }
383
-
384
- template<> EIGEN_STRONG_INLINE Eigen::half pfirst<Packet16h>(const Packet16h& from) {
385
- return half_impl::raw_uint16_to_half(static_cast<unsigned short>(_mm256_extract_epi16(from.x, 0)));
386
- }
387
-
388
- template<> EIGEN_STRONG_INLINE Packet16h pload<Packet16h>(const Eigen::half* from) {
389
- Packet16h result;
390
- result.x = _mm256_load_si256(reinterpret_cast<const __m256i*>(from));
391
- return result;
392
- }
393
-
394
- template<> EIGEN_STRONG_INLINE Packet16h ploadu<Packet16h>(const Eigen::half* from) {
395
- Packet16h result;
396
- result.x = _mm256_loadu_si256(reinterpret_cast<const __m256i*>(from));
397
- return result;
398
- }
399
-
400
- template<> EIGEN_STRONG_INLINE void pstore<half>(Eigen::half* to, const Packet16h& from) {
401
- _mm256_store_si256((__m256i*)to, from.x);
402
- }
403
-
404
- template<> EIGEN_STRONG_INLINE void pstoreu<half>(Eigen::half* to, const Packet16h& from) {
405
- _mm256_storeu_si256((__m256i*)to, from.x);
406
- }
407
-
408
- template<> EIGEN_STRONG_INLINE Packet16h
409
- ploadquad(const Eigen::half* from) {
410
- Packet16h result;
411
- unsigned short a = from[0].x;
412
- unsigned short b = from[1].x;
413
- unsigned short c = from[2].x;
414
- unsigned short d = from[3].x;
415
- result.x = _mm256_set_epi16(d, d, d, d, c, c, c, c, b, b, b, b, a, a, a, a);
416
- return result;
417
- }
418
-
419
- EIGEN_STRONG_INLINE Packet16f half2float(const Packet16h& a) {
420
- #ifdef EIGEN_HAS_FP16_C
421
- return _mm512_cvtph_ps(a.x);
422
- #else
423
- EIGEN_ALIGN64 half aux[16];
424
- pstore(aux, a);
425
- float f0(aux[0]);
426
- float f1(aux[1]);
427
- float f2(aux[2]);
428
- float f3(aux[3]);
429
- float f4(aux[4]);
430
- float f5(aux[5]);
431
- float f6(aux[6]);
432
- float f7(aux[7]);
433
- float f8(aux[8]);
434
- float f9(aux[9]);
435
- float fa(aux[10]);
436
- float fb(aux[11]);
437
- float fc(aux[12]);
438
- float fd(aux[13]);
439
- float fe(aux[14]);
440
- float ff(aux[15]);
441
-
442
- return _mm512_set_ps(
443
- ff, fe, fd, fc, fb, fa, f9, f8, f7, f6, f5, f4, f3, f2, f1, f0);
444
- #endif
445
- }
446
-
447
- EIGEN_STRONG_INLINE Packet16h float2half(const Packet16f& a) {
448
- #ifdef EIGEN_HAS_FP16_C
449
- Packet16h result;
450
- result.x = _mm512_cvtps_ph(a, _MM_FROUND_TO_NEAREST_INT|_MM_FROUND_NO_EXC);
451
- return result;
452
- #else
453
- EIGEN_ALIGN64 float aux[16];
454
- pstore(aux, a);
455
- half h0(aux[0]);
456
- half h1(aux[1]);
457
- half h2(aux[2]);
458
- half h3(aux[3]);
459
- half h4(aux[4]);
460
- half h5(aux[5]);
461
- half h6(aux[6]);
462
- half h7(aux[7]);
463
- half h8(aux[8]);
464
- half h9(aux[9]);
465
- half ha(aux[10]);
466
- half hb(aux[11]);
467
- half hc(aux[12]);
468
- half hd(aux[13]);
469
- half he(aux[14]);
470
- half hf(aux[15]);
471
-
472
- Packet16h result;
473
- result.x = _mm256_set_epi16(
474
- hf.x, he.x, hd.x, hc.x, hb.x, ha.x, h9.x, h8.x,
475
- h7.x, h6.x, h5.x, h4.x, h3.x, h2.x, h1.x, h0.x);
476
- return result;
477
- #endif
478
- }
479
-
480
- template<> EIGEN_STRONG_INLINE Packet16h padd<Packet16h>(const Packet16h& a, const Packet16h& b) {
481
- Packet16f af = half2float(a);
482
- Packet16f bf = half2float(b);
483
- Packet16f rf = padd(af, bf);
484
- return float2half(rf);
485
- }
486
-
487
- template<> EIGEN_STRONG_INLINE Packet16h pmul<Packet16h>(const Packet16h& a, const Packet16h& b) {
488
- Packet16f af = half2float(a);
489
- Packet16f bf = half2float(b);
490
- Packet16f rf = pmul(af, bf);
491
- return float2half(rf);
492
- }
493
-
494
- template<> EIGEN_STRONG_INLINE half predux<Packet16h>(const Packet16h& from) {
495
- Packet16f from_float = half2float(from);
496
- return half(predux(from_float));
497
- }
498
-
499
- template<> EIGEN_STRONG_INLINE Packet16h pgather<Eigen::half, Packet16h>(const Eigen::half* from, Index stride)
500
- {
501
- Packet16h result;
502
- result.x = _mm256_set_epi16(
503
- from[15*stride].x, from[14*stride].x, from[13*stride].x, from[12*stride].x,
504
- from[11*stride].x, from[10*stride].x, from[9*stride].x, from[8*stride].x,
505
- from[7*stride].x, from[6*stride].x, from[5*stride].x, from[4*stride].x,
506
- from[3*stride].x, from[2*stride].x, from[1*stride].x, from[0*stride].x);
507
- return result;
508
- }
509
-
510
- template<> EIGEN_STRONG_INLINE void pscatter<half, Packet16h>(half* to, const Packet16h& from, Index stride)
511
- {
512
- EIGEN_ALIGN64 half aux[16];
513
- pstore(aux, from);
514
- to[stride*0].x = aux[0].x;
515
- to[stride*1].x = aux[1].x;
516
- to[stride*2].x = aux[2].x;
517
- to[stride*3].x = aux[3].x;
518
- to[stride*4].x = aux[4].x;
519
- to[stride*5].x = aux[5].x;
520
- to[stride*6].x = aux[6].x;
521
- to[stride*7].x = aux[7].x;
522
- to[stride*8].x = aux[8].x;
523
- to[stride*9].x = aux[9].x;
524
- to[stride*10].x = aux[10].x;
525
- to[stride*11].x = aux[11].x;
526
- to[stride*12].x = aux[12].x;
527
- to[stride*13].x = aux[13].x;
528
- to[stride*14].x = aux[14].x;
529
- to[stride*15].x = aux[15].x;
530
- }
531
-
532
- EIGEN_STRONG_INLINE void
533
- ptranspose(PacketBlock<Packet16h,16>& kernel) {
534
- __m256i a = kernel.packet[0].x;
535
- __m256i b = kernel.packet[1].x;
536
- __m256i c = kernel.packet[2].x;
537
- __m256i d = kernel.packet[3].x;
538
- __m256i e = kernel.packet[4].x;
539
- __m256i f = kernel.packet[5].x;
540
- __m256i g = kernel.packet[6].x;
541
- __m256i h = kernel.packet[7].x;
542
- __m256i i = kernel.packet[8].x;
543
- __m256i j = kernel.packet[9].x;
544
- __m256i k = kernel.packet[10].x;
545
- __m256i l = kernel.packet[11].x;
546
- __m256i m = kernel.packet[12].x;
547
- __m256i n = kernel.packet[13].x;
548
- __m256i o = kernel.packet[14].x;
549
- __m256i p = kernel.packet[15].x;
550
-
551
- __m256i ab_07 = _mm256_unpacklo_epi16(a, b);
552
- __m256i cd_07 = _mm256_unpacklo_epi16(c, d);
553
- __m256i ef_07 = _mm256_unpacklo_epi16(e, f);
554
- __m256i gh_07 = _mm256_unpacklo_epi16(g, h);
555
- __m256i ij_07 = _mm256_unpacklo_epi16(i, j);
556
- __m256i kl_07 = _mm256_unpacklo_epi16(k, l);
557
- __m256i mn_07 = _mm256_unpacklo_epi16(m, n);
558
- __m256i op_07 = _mm256_unpacklo_epi16(o, p);
559
-
560
- __m256i ab_8f = _mm256_unpackhi_epi16(a, b);
561
- __m256i cd_8f = _mm256_unpackhi_epi16(c, d);
562
- __m256i ef_8f = _mm256_unpackhi_epi16(e, f);
563
- __m256i gh_8f = _mm256_unpackhi_epi16(g, h);
564
- __m256i ij_8f = _mm256_unpackhi_epi16(i, j);
565
- __m256i kl_8f = _mm256_unpackhi_epi16(k, l);
566
- __m256i mn_8f = _mm256_unpackhi_epi16(m, n);
567
- __m256i op_8f = _mm256_unpackhi_epi16(o, p);
568
-
569
- __m256i abcd_03 = _mm256_unpacklo_epi32(ab_07, cd_07);
570
- __m256i abcd_47 = _mm256_unpackhi_epi32(ab_07, cd_07);
571
- __m256i efgh_03 = _mm256_unpacklo_epi32(ef_07, gh_07);
572
- __m256i efgh_47 = _mm256_unpackhi_epi32(ef_07, gh_07);
573
- __m256i ijkl_03 = _mm256_unpacklo_epi32(ij_07, kl_07);
574
- __m256i ijkl_47 = _mm256_unpackhi_epi32(ij_07, kl_07);
575
- __m256i mnop_03 = _mm256_unpacklo_epi32(mn_07, op_07);
576
- __m256i mnop_47 = _mm256_unpackhi_epi32(mn_07, op_07);
577
-
578
- __m256i abcd_8b = _mm256_unpacklo_epi32(ab_8f, cd_8f);
579
- __m256i abcd_cf = _mm256_unpackhi_epi32(ab_8f, cd_8f);
580
- __m256i efgh_8b = _mm256_unpacklo_epi32(ef_8f, gh_8f);
581
- __m256i efgh_cf = _mm256_unpackhi_epi32(ef_8f, gh_8f);
582
- __m256i ijkl_8b = _mm256_unpacklo_epi32(ij_8f, kl_8f);
583
- __m256i ijkl_cf = _mm256_unpackhi_epi32(ij_8f, kl_8f);
584
- __m256i mnop_8b = _mm256_unpacklo_epi32(mn_8f, op_8f);
585
- __m256i mnop_cf = _mm256_unpackhi_epi32(mn_8f, op_8f);
586
-
587
- __m256i abcdefgh_01 = _mm256_unpacklo_epi64(abcd_03, efgh_03);
588
- __m256i abcdefgh_23 = _mm256_unpackhi_epi64(abcd_03, efgh_03);
589
- __m256i ijklmnop_01 = _mm256_unpacklo_epi64(ijkl_03, mnop_03);
590
- __m256i ijklmnop_23 = _mm256_unpackhi_epi64(ijkl_03, mnop_03);
591
- __m256i abcdefgh_45 = _mm256_unpacklo_epi64(abcd_47, efgh_47);
592
- __m256i abcdefgh_67 = _mm256_unpackhi_epi64(abcd_47, efgh_47);
593
- __m256i ijklmnop_45 = _mm256_unpacklo_epi64(ijkl_47, mnop_47);
594
- __m256i ijklmnop_67 = _mm256_unpackhi_epi64(ijkl_47, mnop_47);
595
- __m256i abcdefgh_89 = _mm256_unpacklo_epi64(abcd_8b, efgh_8b);
596
- __m256i abcdefgh_ab = _mm256_unpackhi_epi64(abcd_8b, efgh_8b);
597
- __m256i ijklmnop_89 = _mm256_unpacklo_epi64(ijkl_8b, mnop_8b);
598
- __m256i ijklmnop_ab = _mm256_unpackhi_epi64(ijkl_8b, mnop_8b);
599
- __m256i abcdefgh_cd = _mm256_unpacklo_epi64(abcd_cf, efgh_cf);
600
- __m256i abcdefgh_ef = _mm256_unpackhi_epi64(abcd_cf, efgh_cf);
601
- __m256i ijklmnop_cd = _mm256_unpacklo_epi64(ijkl_cf, mnop_cf);
602
- __m256i ijklmnop_ef = _mm256_unpackhi_epi64(ijkl_cf, mnop_cf);
603
-
604
- // NOTE: no unpacklo/hi instr in this case, so using permute instr.
605
- __m256i a_p_0 = _mm256_permute2x128_si256(abcdefgh_01, ijklmnop_01, 0x20);
606
- __m256i a_p_1 = _mm256_permute2x128_si256(abcdefgh_01, ijklmnop_01, 0x31);
607
- __m256i a_p_2 = _mm256_permute2x128_si256(abcdefgh_23, ijklmnop_23, 0x20);
608
- __m256i a_p_3 = _mm256_permute2x128_si256(abcdefgh_23, ijklmnop_23, 0x31);
609
- __m256i a_p_4 = _mm256_permute2x128_si256(abcdefgh_45, ijklmnop_45, 0x20);
610
- __m256i a_p_5 = _mm256_permute2x128_si256(abcdefgh_45, ijklmnop_45, 0x31);
611
- __m256i a_p_6 = _mm256_permute2x128_si256(abcdefgh_67, ijklmnop_67, 0x20);
612
- __m256i a_p_7 = _mm256_permute2x128_si256(abcdefgh_67, ijklmnop_67, 0x31);
613
- __m256i a_p_8 = _mm256_permute2x128_si256(abcdefgh_89, ijklmnop_89, 0x20);
614
- __m256i a_p_9 = _mm256_permute2x128_si256(abcdefgh_89, ijklmnop_89, 0x31);
615
- __m256i a_p_a = _mm256_permute2x128_si256(abcdefgh_ab, ijklmnop_ab, 0x20);
616
- __m256i a_p_b = _mm256_permute2x128_si256(abcdefgh_ab, ijklmnop_ab, 0x31);
617
- __m256i a_p_c = _mm256_permute2x128_si256(abcdefgh_cd, ijklmnop_cd, 0x20);
618
- __m256i a_p_d = _mm256_permute2x128_si256(abcdefgh_cd, ijklmnop_cd, 0x31);
619
- __m256i a_p_e = _mm256_permute2x128_si256(abcdefgh_ef, ijklmnop_ef, 0x20);
620
- __m256i a_p_f = _mm256_permute2x128_si256(abcdefgh_ef, ijklmnop_ef, 0x31);
621
-
622
- kernel.packet[0].x = a_p_0;
623
- kernel.packet[1].x = a_p_1;
624
- kernel.packet[2].x = a_p_2;
625
- kernel.packet[3].x = a_p_3;
626
- kernel.packet[4].x = a_p_4;
627
- kernel.packet[5].x = a_p_5;
628
- kernel.packet[6].x = a_p_6;
629
- kernel.packet[7].x = a_p_7;
630
- kernel.packet[8].x = a_p_8;
631
- kernel.packet[9].x = a_p_9;
632
- kernel.packet[10].x = a_p_a;
633
- kernel.packet[11].x = a_p_b;
634
- kernel.packet[12].x = a_p_c;
635
- kernel.packet[13].x = a_p_d;
636
- kernel.packet[14].x = a_p_e;
637
- kernel.packet[15].x = a_p_f;
638
- }
639
-
640
- EIGEN_STRONG_INLINE void
641
- ptranspose(PacketBlock<Packet16h,8>& kernel) {
642
- EIGEN_ALIGN64 half in[8][16];
643
- pstore<half>(in[0], kernel.packet[0]);
644
- pstore<half>(in[1], kernel.packet[1]);
645
- pstore<half>(in[2], kernel.packet[2]);
646
- pstore<half>(in[3], kernel.packet[3]);
647
- pstore<half>(in[4], kernel.packet[4]);
648
- pstore<half>(in[5], kernel.packet[5]);
649
- pstore<half>(in[6], kernel.packet[6]);
650
- pstore<half>(in[7], kernel.packet[7]);
651
-
652
- EIGEN_ALIGN64 half out[8][16];
653
-
654
- for (int i = 0; i < 8; ++i) {
655
- for (int j = 0; j < 8; ++j) {
656
- out[i][j] = in[j][2*i];
657
- }
658
- for (int j = 0; j < 8; ++j) {
659
- out[i][j+8] = in[j][2*i+1];
660
- }
661
- }
662
-
663
- kernel.packet[0] = pload<Packet16h>(out[0]);
664
- kernel.packet[1] = pload<Packet16h>(out[1]);
665
- kernel.packet[2] = pload<Packet16h>(out[2]);
666
- kernel.packet[3] = pload<Packet16h>(out[3]);
667
- kernel.packet[4] = pload<Packet16h>(out[4]);
668
- kernel.packet[5] = pload<Packet16h>(out[5]);
669
- kernel.packet[6] = pload<Packet16h>(out[6]);
670
- kernel.packet[7] = pload<Packet16h>(out[7]);
671
- }
672
-
673
- EIGEN_STRONG_INLINE void
674
- ptranspose(PacketBlock<Packet16h,4>& kernel) {
675
- EIGEN_ALIGN64 half in[4][16];
676
- pstore<half>(in[0], kernel.packet[0]);
677
- pstore<half>(in[1], kernel.packet[1]);
678
- pstore<half>(in[2], kernel.packet[2]);
679
- pstore<half>(in[3], kernel.packet[3]);
680
-
681
- EIGEN_ALIGN64 half out[4][16];
682
-
683
- for (int i = 0; i < 4; ++i) {
684
- for (int j = 0; j < 4; ++j) {
685
- out[i][j] = in[j][4*i];
686
- }
687
- for (int j = 0; j < 4; ++j) {
688
- out[i][j+4] = in[j][4*i+1];
689
- }
690
- for (int j = 0; j < 4; ++j) {
691
- out[i][j+8] = in[j][4*i+2];
692
- }
693
- for (int j = 0; j < 4; ++j) {
694
- out[i][j+12] = in[j][4*i+3];
695
- }
696
- }
697
-
698
- kernel.packet[0] = pload<Packet16h>(out[0]);
699
- kernel.packet[1] = pload<Packet16h>(out[1]);
700
- kernel.packet[2] = pload<Packet16h>(out[2]);
701
- kernel.packet[3] = pload<Packet16h>(out[3]);
702
- }
703
-
704
-
705
- #elif defined EIGEN_VECTORIZE_AVX
706
-
707
- typedef struct {
708
- __m128i x;
709
- } Packet8h;
710
-
711
-
712
- template<> struct is_arithmetic<Packet8h> { enum { value = true }; };
713
-
714
- template <>
715
- struct packet_traits<Eigen::half> : default_packet_traits {
716
- typedef Packet8h type;
717
- // There is no half-size packet for Packet8h.
718
- typedef Packet8h half;
719
- enum {
720
- Vectorizable = 1,
721
- AlignedOnScalar = 1,
722
- size = 8,
723
- HasHalfPacket = 0,
724
- HasAdd = 0,
725
- HasSub = 0,
726
- HasMul = 0,
727
- HasNegate = 0,
728
- HasAbs = 0,
729
- HasAbs2 = 0,
730
- HasMin = 0,
731
- HasMax = 0,
732
- HasConj = 0,
733
- HasSetLinear = 0,
734
- HasDiv = 0,
735
- HasSqrt = 0,
736
- HasRsqrt = 0,
737
- HasExp = 0,
738
- HasLog = 0,
739
- HasBlend = 0
740
- };
741
- };
742
-
743
-
744
- template<> struct unpacket_traits<Packet8h> { typedef Eigen::half type; enum {size=8, alignment=Aligned16}; typedef Packet8h half; };
745
-
746
- template<> EIGEN_STRONG_INLINE Packet8h pset1<Packet8h>(const Eigen::half& from) {
747
- Packet8h result;
748
- result.x = _mm_set1_epi16(from.x);
749
- return result;
750
- }
751
-
752
- template<> EIGEN_STRONG_INLINE Eigen::half pfirst<Packet8h>(const Packet8h& from) {
753
- return half_impl::raw_uint16_to_half(static_cast<unsigned short>(_mm_extract_epi16(from.x, 0)));
754
- }
755
-
756
- template<> EIGEN_STRONG_INLINE Packet8h pload<Packet8h>(const Eigen::half* from) {
757
- Packet8h result;
758
- result.x = _mm_load_si128(reinterpret_cast<const __m128i*>(from));
759
- return result;
760
- }
761
-
762
- template<> EIGEN_STRONG_INLINE Packet8h ploadu<Packet8h>(const Eigen::half* from) {
763
- Packet8h result;
764
- result.x = _mm_loadu_si128(reinterpret_cast<const __m128i*>(from));
765
- return result;
766
- }
767
-
768
- template<> EIGEN_STRONG_INLINE void pstore<Eigen::half>(Eigen::half* to, const Packet8h& from) {
769
- _mm_store_si128(reinterpret_cast<__m128i*>(to), from.x);
770
- }
771
-
772
- template<> EIGEN_STRONG_INLINE void pstoreu<Eigen::half>(Eigen::half* to, const Packet8h& from) {
773
- _mm_storeu_si128(reinterpret_cast<__m128i*>(to), from.x);
774
- }
775
-
776
- template<> EIGEN_STRONG_INLINE Packet8h
777
- ploadquad<Packet8h>(const Eigen::half* from) {
778
- Packet8h result;
779
- unsigned short a = from[0].x;
780
- unsigned short b = from[1].x;
781
- result.x = _mm_set_epi16(b, b, b, b, a, a, a, a);
782
- return result;
783
- }
784
-
785
- EIGEN_STRONG_INLINE Packet8f half2float(const Packet8h& a) {
786
- #ifdef EIGEN_HAS_FP16_C
787
- return _mm256_cvtph_ps(a.x);
788
- #else
789
- EIGEN_ALIGN32 Eigen::half aux[8];
790
- pstore(aux, a);
791
- float f0(aux[0]);
792
- float f1(aux[1]);
793
- float f2(aux[2]);
794
- float f3(aux[3]);
795
- float f4(aux[4]);
796
- float f5(aux[5]);
797
- float f6(aux[6]);
798
- float f7(aux[7]);
799
-
800
- return _mm256_set_ps(f7, f6, f5, f4, f3, f2, f1, f0);
801
- #endif
802
- }
803
-
804
- EIGEN_STRONG_INLINE Packet8h float2half(const Packet8f& a) {
805
- #ifdef EIGEN_HAS_FP16_C
806
- Packet8h result;
807
- result.x = _mm256_cvtps_ph(a, _MM_FROUND_TO_NEAREST_INT|_MM_FROUND_NO_EXC);
808
- return result;
809
- #else
810
- EIGEN_ALIGN32 float aux[8];
811
- pstore(aux, a);
812
- Eigen::half h0(aux[0]);
813
- Eigen::half h1(aux[1]);
814
- Eigen::half h2(aux[2]);
815
- Eigen::half h3(aux[3]);
816
- Eigen::half h4(aux[4]);
817
- Eigen::half h5(aux[5]);
818
- Eigen::half h6(aux[6]);
819
- Eigen::half h7(aux[7]);
820
-
821
- Packet8h result;
822
- result.x = _mm_set_epi16(h7.x, h6.x, h5.x, h4.x, h3.x, h2.x, h1.x, h0.x);
823
- return result;
824
- #endif
825
- }
826
-
827
- template<> EIGEN_STRONG_INLINE Packet8h pconj(const Packet8h& a) { return a; }
828
-
829
- template<> EIGEN_STRONG_INLINE Packet8h padd<Packet8h>(const Packet8h& a, const Packet8h& b) {
830
- Packet8f af = half2float(a);
831
- Packet8f bf = half2float(b);
832
- Packet8f rf = padd(af, bf);
833
- return float2half(rf);
834
- }
835
-
836
- template<> EIGEN_STRONG_INLINE Packet8h pmul<Packet8h>(const Packet8h& a, const Packet8h& b) {
837
- Packet8f af = half2float(a);
838
- Packet8f bf = half2float(b);
839
- Packet8f rf = pmul(af, bf);
840
- return float2half(rf);
841
- }
842
-
843
- template<> EIGEN_STRONG_INLINE Packet8h pgather<Eigen::half, Packet8h>(const Eigen::half* from, Index stride)
844
- {
845
- Packet8h result;
846
- result.x = _mm_set_epi16(from[7*stride].x, from[6*stride].x, from[5*stride].x, from[4*stride].x, from[3*stride].x, from[2*stride].x, from[1*stride].x, from[0*stride].x);
847
- return result;
848
- }
849
-
850
- template<> EIGEN_STRONG_INLINE void pscatter<Eigen::half, Packet8h>(Eigen::half* to, const Packet8h& from, Index stride)
851
- {
852
- EIGEN_ALIGN32 Eigen::half aux[8];
853
- pstore(aux, from);
854
- to[stride*0].x = aux[0].x;
855
- to[stride*1].x = aux[1].x;
856
- to[stride*2].x = aux[2].x;
857
- to[stride*3].x = aux[3].x;
858
- to[stride*4].x = aux[4].x;
859
- to[stride*5].x = aux[5].x;
860
- to[stride*6].x = aux[6].x;
861
- to[stride*7].x = aux[7].x;
862
- }
863
-
864
- template<> EIGEN_STRONG_INLINE Eigen::half predux<Packet8h>(const Packet8h& a) {
865
- Packet8f af = half2float(a);
866
- float reduced = predux<Packet8f>(af);
867
- return Eigen::half(reduced);
868
- }
869
-
870
- template<> EIGEN_STRONG_INLINE Eigen::half predux_max<Packet8h>(const Packet8h& a) {
871
- Packet8f af = half2float(a);
872
- float reduced = predux_max<Packet8f>(af);
873
- return Eigen::half(reduced);
874
- }
875
-
876
- template<> EIGEN_STRONG_INLINE Eigen::half predux_min<Packet8h>(const Packet8h& a) {
877
- Packet8f af = half2float(a);
878
- float reduced = predux_min<Packet8f>(af);
879
- return Eigen::half(reduced);
880
- }
881
-
882
- template<> EIGEN_STRONG_INLINE Eigen::half predux_mul<Packet8h>(const Packet8h& a) {
883
- Packet8f af = half2float(a);
884
- float reduced = predux_mul<Packet8f>(af);
885
- return Eigen::half(reduced);
886
- }
887
-
888
- EIGEN_STRONG_INLINE void
889
- ptranspose(PacketBlock<Packet8h,8>& kernel) {
890
- __m128i a = kernel.packet[0].x;
891
- __m128i b = kernel.packet[1].x;
892
- __m128i c = kernel.packet[2].x;
893
- __m128i d = kernel.packet[3].x;
894
- __m128i e = kernel.packet[4].x;
895
- __m128i f = kernel.packet[5].x;
896
- __m128i g = kernel.packet[6].x;
897
- __m128i h = kernel.packet[7].x;
898
-
899
- __m128i a03b03 = _mm_unpacklo_epi16(a, b);
900
- __m128i c03d03 = _mm_unpacklo_epi16(c, d);
901
- __m128i e03f03 = _mm_unpacklo_epi16(e, f);
902
- __m128i g03h03 = _mm_unpacklo_epi16(g, h);
903
- __m128i a47b47 = _mm_unpackhi_epi16(a, b);
904
- __m128i c47d47 = _mm_unpackhi_epi16(c, d);
905
- __m128i e47f47 = _mm_unpackhi_epi16(e, f);
906
- __m128i g47h47 = _mm_unpackhi_epi16(g, h);
907
-
908
- __m128i a01b01c01d01 = _mm_unpacklo_epi32(a03b03, c03d03);
909
- __m128i a23b23c23d23 = _mm_unpackhi_epi32(a03b03, c03d03);
910
- __m128i e01f01g01h01 = _mm_unpacklo_epi32(e03f03, g03h03);
911
- __m128i e23f23g23h23 = _mm_unpackhi_epi32(e03f03, g03h03);
912
- __m128i a45b45c45d45 = _mm_unpacklo_epi32(a47b47, c47d47);
913
- __m128i a67b67c67d67 = _mm_unpackhi_epi32(a47b47, c47d47);
914
- __m128i e45f45g45h45 = _mm_unpacklo_epi32(e47f47, g47h47);
915
- __m128i e67f67g67h67 = _mm_unpackhi_epi32(e47f47, g47h47);
916
-
917
- __m128i a0b0c0d0e0f0g0h0 = _mm_unpacklo_epi64(a01b01c01d01, e01f01g01h01);
918
- __m128i a1b1c1d1e1f1g1h1 = _mm_unpackhi_epi64(a01b01c01d01, e01f01g01h01);
919
- __m128i a2b2c2d2e2f2g2h2 = _mm_unpacklo_epi64(a23b23c23d23, e23f23g23h23);
920
- __m128i a3b3c3d3e3f3g3h3 = _mm_unpackhi_epi64(a23b23c23d23, e23f23g23h23);
921
- __m128i a4b4c4d4e4f4g4h4 = _mm_unpacklo_epi64(a45b45c45d45, e45f45g45h45);
922
- __m128i a5b5c5d5e5f5g5h5 = _mm_unpackhi_epi64(a45b45c45d45, e45f45g45h45);
923
- __m128i a6b6c6d6e6f6g6h6 = _mm_unpacklo_epi64(a67b67c67d67, e67f67g67h67);
924
- __m128i a7b7c7d7e7f7g7h7 = _mm_unpackhi_epi64(a67b67c67d67, e67f67g67h67);
925
-
926
- kernel.packet[0].x = a0b0c0d0e0f0g0h0;
927
- kernel.packet[1].x = a1b1c1d1e1f1g1h1;
928
- kernel.packet[2].x = a2b2c2d2e2f2g2h2;
929
- kernel.packet[3].x = a3b3c3d3e3f3g3h3;
930
- kernel.packet[4].x = a4b4c4d4e4f4g4h4;
931
- kernel.packet[5].x = a5b5c5d5e5f5g5h5;
932
- kernel.packet[6].x = a6b6c6d6e6f6g6h6;
933
- kernel.packet[7].x = a7b7c7d7e7f7g7h7;
934
- }
935
-
936
- EIGEN_STRONG_INLINE void
937
- ptranspose(PacketBlock<Packet8h,4>& kernel) {
938
- EIGEN_ALIGN32 Eigen::half in[4][8];
939
- pstore<Eigen::half>(in[0], kernel.packet[0]);
940
- pstore<Eigen::half>(in[1], kernel.packet[1]);
941
- pstore<Eigen::half>(in[2], kernel.packet[2]);
942
- pstore<Eigen::half>(in[3], kernel.packet[3]);
943
-
944
- EIGEN_ALIGN32 Eigen::half out[4][8];
945
-
946
- for (int i = 0; i < 4; ++i) {
947
- for (int j = 0; j < 4; ++j) {
948
- out[i][j] = in[j][2*i];
949
- }
950
- for (int j = 0; j < 4; ++j) {
951
- out[i][j+4] = in[j][2*i+1];
952
- }
953
- }
954
-
955
- kernel.packet[0] = pload<Packet8h>(out[0]);
956
- kernel.packet[1] = pload<Packet8h>(out[1]);
957
- kernel.packet[2] = pload<Packet8h>(out[2]);
958
- kernel.packet[3] = pload<Packet8h>(out[3]);
959
- }
960
-
961
-
962
- // Disable the following code since it's broken on too many platforms / compilers.
963
- //#elif defined(EIGEN_VECTORIZE_SSE) && (!EIGEN_ARCH_x86_64) && (!EIGEN_COMP_MSVC)
964
- #elif 0
965
-
966
- typedef struct {
967
- __m64 x;
968
- } Packet4h;
969
-
970
-
971
- template<> struct is_arithmetic<Packet4h> { enum { value = true }; };
972
-
973
- template <>
974
- struct packet_traits<Eigen::half> : default_packet_traits {
975
- typedef Packet4h type;
976
- // There is no half-size packet for Packet4h.
977
- typedef Packet4h half;
978
- enum {
979
- Vectorizable = 1,
980
- AlignedOnScalar = 1,
981
- size = 4,
982
- HasHalfPacket = 0,
983
- HasAdd = 0,
984
- HasSub = 0,
985
- HasMul = 0,
986
- HasNegate = 0,
987
- HasAbs = 0,
988
- HasAbs2 = 0,
989
- HasMin = 0,
990
- HasMax = 0,
991
- HasConj = 0,
992
- HasSetLinear = 0,
993
- HasDiv = 0,
994
- HasSqrt = 0,
995
- HasRsqrt = 0,
996
- HasExp = 0,
997
- HasLog = 0,
998
- HasBlend = 0
999
- };
1000
- };
1001
-
1002
-
1003
- template<> struct unpacket_traits<Packet4h> { typedef Eigen::half type; enum {size=4, alignment=Aligned16}; typedef Packet4h half; };
1004
-
1005
- template<> EIGEN_STRONG_INLINE Packet4h pset1<Packet4h>(const Eigen::half& from) {
1006
- Packet4h result;
1007
- result.x = _mm_set1_pi16(from.x);
1008
- return result;
1009
- }
1010
-
1011
- template<> EIGEN_STRONG_INLINE Eigen::half pfirst<Packet4h>(const Packet4h& from) {
1012
- return half_impl::raw_uint16_to_half(static_cast<unsigned short>(_mm_cvtsi64_si32(from.x)));
1013
- }
1014
-
1015
- template<> EIGEN_STRONG_INLINE Packet4h pconj(const Packet4h& a) { return a; }
1016
-
1017
- template<> EIGEN_STRONG_INLINE Packet4h padd<Packet4h>(const Packet4h& a, const Packet4h& b) {
1018
- __int64_t a64 = _mm_cvtm64_si64(a.x);
1019
- __int64_t b64 = _mm_cvtm64_si64(b.x);
1020
-
1021
- Eigen::half h[4];
1022
-
1023
- Eigen::half ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64));
1024
- Eigen::half hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64));
1025
- h[0] = ha + hb;
1026
- ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64 >> 16));
1027
- hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64 >> 16));
1028
- h[1] = ha + hb;
1029
- ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64 >> 32));
1030
- hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64 >> 32));
1031
- h[2] = ha + hb;
1032
- ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64 >> 48));
1033
- hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64 >> 48));
1034
- h[3] = ha + hb;
1035
- Packet4h result;
1036
- result.x = _mm_set_pi16(h[3].x, h[2].x, h[1].x, h[0].x);
1037
- return result;
1038
- }
1039
-
1040
- template<> EIGEN_STRONG_INLINE Packet4h pmul<Packet4h>(const Packet4h& a, const Packet4h& b) {
1041
- __int64_t a64 = _mm_cvtm64_si64(a.x);
1042
- __int64_t b64 = _mm_cvtm64_si64(b.x);
1043
-
1044
- Eigen::half h[4];
1045
-
1046
- Eigen::half ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64));
1047
- Eigen::half hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64));
1048
- h[0] = ha * hb;
1049
- ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64 >> 16));
1050
- hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64 >> 16));
1051
- h[1] = ha * hb;
1052
- ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64 >> 32));
1053
- hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64 >> 32));
1054
- h[2] = ha * hb;
1055
- ha = half_impl::raw_uint16_to_half(static_cast<unsigned short>(a64 >> 48));
1056
- hb = half_impl::raw_uint16_to_half(static_cast<unsigned short>(b64 >> 48));
1057
- h[3] = ha * hb;
1058
- Packet4h result;
1059
- result.x = _mm_set_pi16(h[3].x, h[2].x, h[1].x, h[0].x);
1060
- return result;
1061
- }
1062
-
1063
- template<> EIGEN_STRONG_INLINE Packet4h pload<Packet4h>(const Eigen::half* from) {
1064
- Packet4h result;
1065
- result.x = _mm_cvtsi64_m64(*reinterpret_cast<const __int64_t*>(from));
1066
- return result;
1067
- }
1068
-
1069
- template<> EIGEN_STRONG_INLINE Packet4h ploadu<Packet4h>(const Eigen::half* from) {
1070
- Packet4h result;
1071
- result.x = _mm_cvtsi64_m64(*reinterpret_cast<const __int64_t*>(from));
1072
- return result;
1073
- }
1074
-
1075
- template<> EIGEN_STRONG_INLINE void pstore<Eigen::half>(Eigen::half* to, const Packet4h& from) {
1076
- __int64_t r = _mm_cvtm64_si64(from.x);
1077
- *(reinterpret_cast<__int64_t*>(to)) = r;
1078
- }
1079
-
1080
- template<> EIGEN_STRONG_INLINE void pstoreu<Eigen::half>(Eigen::half* to, const Packet4h& from) {
1081
- __int64_t r = _mm_cvtm64_si64(from.x);
1082
- *(reinterpret_cast<__int64_t*>(to)) = r;
1083
- }
1084
-
1085
- template<> EIGEN_STRONG_INLINE Packet4h
1086
- ploadquad<Packet4h>(const Eigen::half* from) {
1087
- return pset1<Packet4h>(*from);
1088
- }
1089
-
1090
- template<> EIGEN_STRONG_INLINE Packet4h pgather<Eigen::half, Packet4h>(const Eigen::half* from, Index stride)
1091
- {
1092
- Packet4h result;
1093
- result.x = _mm_set_pi16(from[3*stride].x, from[2*stride].x, from[1*stride].x, from[0*stride].x);
1094
- return result;
1095
- }
1096
-
1097
- template<> EIGEN_STRONG_INLINE void pscatter<Eigen::half, Packet4h>(Eigen::half* to, const Packet4h& from, Index stride)
1098
- {
1099
- __int64_t a = _mm_cvtm64_si64(from.x);
1100
- to[stride*0].x = static_cast<unsigned short>(a);
1101
- to[stride*1].x = static_cast<unsigned short>(a >> 16);
1102
- to[stride*2].x = static_cast<unsigned short>(a >> 32);
1103
- to[stride*3].x = static_cast<unsigned short>(a >> 48);
1104
- }
1105
-
1106
- EIGEN_STRONG_INLINE void
1107
- ptranspose(PacketBlock<Packet4h,4>& kernel) {
1108
- __m64 T0 = _mm_unpacklo_pi16(kernel.packet[0].x, kernel.packet[1].x);
1109
- __m64 T1 = _mm_unpacklo_pi16(kernel.packet[2].x, kernel.packet[3].x);
1110
- __m64 T2 = _mm_unpackhi_pi16(kernel.packet[0].x, kernel.packet[1].x);
1111
- __m64 T3 = _mm_unpackhi_pi16(kernel.packet[2].x, kernel.packet[3].x);
1112
-
1113
- kernel.packet[0].x = _mm_unpacklo_pi32(T0, T1);
1114
- kernel.packet[1].x = _mm_unpackhi_pi32(T0, T1);
1115
- kernel.packet[2].x = _mm_unpacklo_pi32(T2, T3);
1116
- kernel.packet[3].x = _mm_unpackhi_pi32(T2, T3);
1117
- }
1118
-
1119
- #endif
1120
-
1121
- }
1122
- }
1123
-
1124
- #endif // EIGEN_PACKET_MATH_HALF_CUDA_H