umappp 0.1.2

Sign up to get free protection for your applications and to get access to all the features.
Files changed (395) hide show
  1. checksums.yaml +7 -0
  2. data/LICENSE.txt +25 -0
  3. data/README.md +110 -0
  4. data/ext/umappp/extconf.rb +25 -0
  5. data/ext/umappp/numo.hpp +867 -0
  6. data/ext/umappp/umappp.cpp +225 -0
  7. data/lib/umappp/version.rb +5 -0
  8. data/lib/umappp.rb +41 -0
  9. data/vendor/Eigen/Cholesky +45 -0
  10. data/vendor/Eigen/CholmodSupport +48 -0
  11. data/vendor/Eigen/Core +384 -0
  12. data/vendor/Eigen/Dense +7 -0
  13. data/vendor/Eigen/Eigen +2 -0
  14. data/vendor/Eigen/Eigenvalues +60 -0
  15. data/vendor/Eigen/Geometry +59 -0
  16. data/vendor/Eigen/Householder +29 -0
  17. data/vendor/Eigen/IterativeLinearSolvers +48 -0
  18. data/vendor/Eigen/Jacobi +32 -0
  19. data/vendor/Eigen/KLUSupport +41 -0
  20. data/vendor/Eigen/LU +47 -0
  21. data/vendor/Eigen/MetisSupport +35 -0
  22. data/vendor/Eigen/OrderingMethods +70 -0
  23. data/vendor/Eigen/PaStiXSupport +49 -0
  24. data/vendor/Eigen/PardisoSupport +35 -0
  25. data/vendor/Eigen/QR +50 -0
  26. data/vendor/Eigen/QtAlignedMalloc +39 -0
  27. data/vendor/Eigen/SPQRSupport +34 -0
  28. data/vendor/Eigen/SVD +50 -0
  29. data/vendor/Eigen/Sparse +34 -0
  30. data/vendor/Eigen/SparseCholesky +37 -0
  31. data/vendor/Eigen/SparseCore +69 -0
  32. data/vendor/Eigen/SparseLU +50 -0
  33. data/vendor/Eigen/SparseQR +36 -0
  34. data/vendor/Eigen/StdDeque +27 -0
  35. data/vendor/Eigen/StdList +26 -0
  36. data/vendor/Eigen/StdVector +27 -0
  37. data/vendor/Eigen/SuperLUSupport +64 -0
  38. data/vendor/Eigen/UmfPackSupport +40 -0
  39. data/vendor/Eigen/src/Cholesky/LDLT.h +688 -0
  40. data/vendor/Eigen/src/Cholesky/LLT.h +558 -0
  41. data/vendor/Eigen/src/Cholesky/LLT_LAPACKE.h +99 -0
  42. data/vendor/Eigen/src/CholmodSupport/CholmodSupport.h +682 -0
  43. data/vendor/Eigen/src/Core/ArithmeticSequence.h +413 -0
  44. data/vendor/Eigen/src/Core/Array.h +417 -0
  45. data/vendor/Eigen/src/Core/ArrayBase.h +226 -0
  46. data/vendor/Eigen/src/Core/ArrayWrapper.h +209 -0
  47. data/vendor/Eigen/src/Core/Assign.h +90 -0
  48. data/vendor/Eigen/src/Core/AssignEvaluator.h +1010 -0
  49. data/vendor/Eigen/src/Core/Assign_MKL.h +178 -0
  50. data/vendor/Eigen/src/Core/BandMatrix.h +353 -0
  51. data/vendor/Eigen/src/Core/Block.h +448 -0
  52. data/vendor/Eigen/src/Core/BooleanRedux.h +162 -0
  53. data/vendor/Eigen/src/Core/CommaInitializer.h +164 -0
  54. data/vendor/Eigen/src/Core/ConditionEstimator.h +175 -0
  55. data/vendor/Eigen/src/Core/CoreEvaluators.h +1741 -0
  56. data/vendor/Eigen/src/Core/CoreIterators.h +132 -0
  57. data/vendor/Eigen/src/Core/CwiseBinaryOp.h +183 -0
  58. data/vendor/Eigen/src/Core/CwiseNullaryOp.h +1001 -0
  59. data/vendor/Eigen/src/Core/CwiseTernaryOp.h +197 -0
  60. data/vendor/Eigen/src/Core/CwiseUnaryOp.h +103 -0
  61. data/vendor/Eigen/src/Core/CwiseUnaryView.h +132 -0
  62. data/vendor/Eigen/src/Core/DenseBase.h +701 -0
  63. data/vendor/Eigen/src/Core/DenseCoeffsBase.h +685 -0
  64. data/vendor/Eigen/src/Core/DenseStorage.h +652 -0
  65. data/vendor/Eigen/src/Core/Diagonal.h +258 -0
  66. data/vendor/Eigen/src/Core/DiagonalMatrix.h +391 -0
  67. data/vendor/Eigen/src/Core/DiagonalProduct.h +28 -0
  68. data/vendor/Eigen/src/Core/Dot.h +318 -0
  69. data/vendor/Eigen/src/Core/EigenBase.h +160 -0
  70. data/vendor/Eigen/src/Core/ForceAlignedAccess.h +150 -0
  71. data/vendor/Eigen/src/Core/Fuzzy.h +155 -0
  72. data/vendor/Eigen/src/Core/GeneralProduct.h +465 -0
  73. data/vendor/Eigen/src/Core/GenericPacketMath.h +1040 -0
  74. data/vendor/Eigen/src/Core/GlobalFunctions.h +194 -0
  75. data/vendor/Eigen/src/Core/IO.h +258 -0
  76. data/vendor/Eigen/src/Core/IndexedView.h +237 -0
  77. data/vendor/Eigen/src/Core/Inverse.h +117 -0
  78. data/vendor/Eigen/src/Core/Map.h +171 -0
  79. data/vendor/Eigen/src/Core/MapBase.h +310 -0
  80. data/vendor/Eigen/src/Core/MathFunctions.h +2057 -0
  81. data/vendor/Eigen/src/Core/MathFunctionsImpl.h +200 -0
  82. data/vendor/Eigen/src/Core/Matrix.h +565 -0
  83. data/vendor/Eigen/src/Core/MatrixBase.h +547 -0
  84. data/vendor/Eigen/src/Core/NestByValue.h +85 -0
  85. data/vendor/Eigen/src/Core/NoAlias.h +109 -0
  86. data/vendor/Eigen/src/Core/NumTraits.h +335 -0
  87. data/vendor/Eigen/src/Core/PartialReduxEvaluator.h +232 -0
  88. data/vendor/Eigen/src/Core/PermutationMatrix.h +605 -0
  89. data/vendor/Eigen/src/Core/PlainObjectBase.h +1128 -0
  90. data/vendor/Eigen/src/Core/Product.h +191 -0
  91. data/vendor/Eigen/src/Core/ProductEvaluators.h +1179 -0
  92. data/vendor/Eigen/src/Core/Random.h +218 -0
  93. data/vendor/Eigen/src/Core/Redux.h +515 -0
  94. data/vendor/Eigen/src/Core/Ref.h +381 -0
  95. data/vendor/Eigen/src/Core/Replicate.h +142 -0
  96. data/vendor/Eigen/src/Core/Reshaped.h +454 -0
  97. data/vendor/Eigen/src/Core/ReturnByValue.h +119 -0
  98. data/vendor/Eigen/src/Core/Reverse.h +217 -0
  99. data/vendor/Eigen/src/Core/Select.h +164 -0
  100. data/vendor/Eigen/src/Core/SelfAdjointView.h +365 -0
  101. data/vendor/Eigen/src/Core/SelfCwiseBinaryOp.h +47 -0
  102. data/vendor/Eigen/src/Core/Solve.h +188 -0
  103. data/vendor/Eigen/src/Core/SolveTriangular.h +235 -0
  104. data/vendor/Eigen/src/Core/SolverBase.h +168 -0
  105. data/vendor/Eigen/src/Core/StableNorm.h +251 -0
  106. data/vendor/Eigen/src/Core/StlIterators.h +463 -0
  107. data/vendor/Eigen/src/Core/Stride.h +116 -0
  108. data/vendor/Eigen/src/Core/Swap.h +68 -0
  109. data/vendor/Eigen/src/Core/Transpose.h +464 -0
  110. data/vendor/Eigen/src/Core/Transpositions.h +386 -0
  111. data/vendor/Eigen/src/Core/TriangularMatrix.h +1001 -0
  112. data/vendor/Eigen/src/Core/VectorBlock.h +96 -0
  113. data/vendor/Eigen/src/Core/VectorwiseOp.h +784 -0
  114. data/vendor/Eigen/src/Core/Visitor.h +381 -0
  115. data/vendor/Eigen/src/Core/arch/AVX/Complex.h +372 -0
  116. data/vendor/Eigen/src/Core/arch/AVX/MathFunctions.h +228 -0
  117. data/vendor/Eigen/src/Core/arch/AVX/PacketMath.h +1574 -0
  118. data/vendor/Eigen/src/Core/arch/AVX/TypeCasting.h +115 -0
  119. data/vendor/Eigen/src/Core/arch/AVX512/Complex.h +422 -0
  120. data/vendor/Eigen/src/Core/arch/AVX512/MathFunctions.h +362 -0
  121. data/vendor/Eigen/src/Core/arch/AVX512/PacketMath.h +2303 -0
  122. data/vendor/Eigen/src/Core/arch/AVX512/TypeCasting.h +89 -0
  123. data/vendor/Eigen/src/Core/arch/AltiVec/Complex.h +417 -0
  124. data/vendor/Eigen/src/Core/arch/AltiVec/MathFunctions.h +90 -0
  125. data/vendor/Eigen/src/Core/arch/AltiVec/MatrixProduct.h +2937 -0
  126. data/vendor/Eigen/src/Core/arch/AltiVec/MatrixProductCommon.h +221 -0
  127. data/vendor/Eigen/src/Core/arch/AltiVec/MatrixProductMMA.h +629 -0
  128. data/vendor/Eigen/src/Core/arch/AltiVec/PacketMath.h +2711 -0
  129. data/vendor/Eigen/src/Core/arch/CUDA/Complex.h +258 -0
  130. data/vendor/Eigen/src/Core/arch/Default/BFloat16.h +700 -0
  131. data/vendor/Eigen/src/Core/arch/Default/ConjHelper.h +117 -0
  132. data/vendor/Eigen/src/Core/arch/Default/GenericPacketMathFunctions.h +1649 -0
  133. data/vendor/Eigen/src/Core/arch/Default/GenericPacketMathFunctionsFwd.h +110 -0
  134. data/vendor/Eigen/src/Core/arch/Default/Half.h +942 -0
  135. data/vendor/Eigen/src/Core/arch/Default/Settings.h +49 -0
  136. data/vendor/Eigen/src/Core/arch/Default/TypeCasting.h +120 -0
  137. data/vendor/Eigen/src/Core/arch/GPU/MathFunctions.h +103 -0
  138. data/vendor/Eigen/src/Core/arch/GPU/PacketMath.h +1685 -0
  139. data/vendor/Eigen/src/Core/arch/GPU/TypeCasting.h +80 -0
  140. data/vendor/Eigen/src/Core/arch/HIP/hcc/math_constants.h +23 -0
  141. data/vendor/Eigen/src/Core/arch/MSA/Complex.h +648 -0
  142. data/vendor/Eigen/src/Core/arch/MSA/MathFunctions.h +387 -0
  143. data/vendor/Eigen/src/Core/arch/MSA/PacketMath.h +1233 -0
  144. data/vendor/Eigen/src/Core/arch/NEON/Complex.h +584 -0
  145. data/vendor/Eigen/src/Core/arch/NEON/GeneralBlockPanelKernel.h +183 -0
  146. data/vendor/Eigen/src/Core/arch/NEON/MathFunctions.h +75 -0
  147. data/vendor/Eigen/src/Core/arch/NEON/PacketMath.h +4587 -0
  148. data/vendor/Eigen/src/Core/arch/NEON/TypeCasting.h +1419 -0
  149. data/vendor/Eigen/src/Core/arch/SSE/Complex.h +351 -0
  150. data/vendor/Eigen/src/Core/arch/SSE/MathFunctions.h +199 -0
  151. data/vendor/Eigen/src/Core/arch/SSE/PacketMath.h +1505 -0
  152. data/vendor/Eigen/src/Core/arch/SSE/TypeCasting.h +142 -0
  153. data/vendor/Eigen/src/Core/arch/SVE/MathFunctions.h +44 -0
  154. data/vendor/Eigen/src/Core/arch/SVE/PacketMath.h +752 -0
  155. data/vendor/Eigen/src/Core/arch/SVE/TypeCasting.h +49 -0
  156. data/vendor/Eigen/src/Core/arch/SYCL/InteropHeaders.h +232 -0
  157. data/vendor/Eigen/src/Core/arch/SYCL/MathFunctions.h +301 -0
  158. data/vendor/Eigen/src/Core/arch/SYCL/PacketMath.h +670 -0
  159. data/vendor/Eigen/src/Core/arch/SYCL/SyclMemoryModel.h +694 -0
  160. data/vendor/Eigen/src/Core/arch/SYCL/TypeCasting.h +85 -0
  161. data/vendor/Eigen/src/Core/arch/ZVector/Complex.h +426 -0
  162. data/vendor/Eigen/src/Core/arch/ZVector/MathFunctions.h +233 -0
  163. data/vendor/Eigen/src/Core/arch/ZVector/PacketMath.h +1060 -0
  164. data/vendor/Eigen/src/Core/functors/AssignmentFunctors.h +177 -0
  165. data/vendor/Eigen/src/Core/functors/BinaryFunctors.h +541 -0
  166. data/vendor/Eigen/src/Core/functors/NullaryFunctors.h +189 -0
  167. data/vendor/Eigen/src/Core/functors/StlFunctors.h +166 -0
  168. data/vendor/Eigen/src/Core/functors/TernaryFunctors.h +25 -0
  169. data/vendor/Eigen/src/Core/functors/UnaryFunctors.h +1131 -0
  170. data/vendor/Eigen/src/Core/products/GeneralBlockPanelKernel.h +2645 -0
  171. data/vendor/Eigen/src/Core/products/GeneralMatrixMatrix.h +517 -0
  172. data/vendor/Eigen/src/Core/products/GeneralMatrixMatrixTriangular.h +317 -0
  173. data/vendor/Eigen/src/Core/products/GeneralMatrixMatrixTriangular_BLAS.h +145 -0
  174. data/vendor/Eigen/src/Core/products/GeneralMatrixMatrix_BLAS.h +124 -0
  175. data/vendor/Eigen/src/Core/products/GeneralMatrixVector.h +518 -0
  176. data/vendor/Eigen/src/Core/products/GeneralMatrixVector_BLAS.h +136 -0
  177. data/vendor/Eigen/src/Core/products/Parallelizer.h +180 -0
  178. data/vendor/Eigen/src/Core/products/SelfadjointMatrixMatrix.h +544 -0
  179. data/vendor/Eigen/src/Core/products/SelfadjointMatrixMatrix_BLAS.h +295 -0
  180. data/vendor/Eigen/src/Core/products/SelfadjointMatrixVector.h +262 -0
  181. data/vendor/Eigen/src/Core/products/SelfadjointMatrixVector_BLAS.h +118 -0
  182. data/vendor/Eigen/src/Core/products/SelfadjointProduct.h +133 -0
  183. data/vendor/Eigen/src/Core/products/SelfadjointRank2Update.h +94 -0
  184. data/vendor/Eigen/src/Core/products/TriangularMatrixMatrix.h +472 -0
  185. data/vendor/Eigen/src/Core/products/TriangularMatrixMatrix_BLAS.h +317 -0
  186. data/vendor/Eigen/src/Core/products/TriangularMatrixVector.h +350 -0
  187. data/vendor/Eigen/src/Core/products/TriangularMatrixVector_BLAS.h +255 -0
  188. data/vendor/Eigen/src/Core/products/TriangularSolverMatrix.h +337 -0
  189. data/vendor/Eigen/src/Core/products/TriangularSolverMatrix_BLAS.h +167 -0
  190. data/vendor/Eigen/src/Core/products/TriangularSolverVector.h +148 -0
  191. data/vendor/Eigen/src/Core/util/BlasUtil.h +583 -0
  192. data/vendor/Eigen/src/Core/util/ConfigureVectorization.h +512 -0
  193. data/vendor/Eigen/src/Core/util/Constants.h +563 -0
  194. data/vendor/Eigen/src/Core/util/DisableStupidWarnings.h +106 -0
  195. data/vendor/Eigen/src/Core/util/ForwardDeclarations.h +322 -0
  196. data/vendor/Eigen/src/Core/util/IndexedViewHelper.h +186 -0
  197. data/vendor/Eigen/src/Core/util/IntegralConstant.h +272 -0
  198. data/vendor/Eigen/src/Core/util/MKL_support.h +137 -0
  199. data/vendor/Eigen/src/Core/util/Macros.h +1464 -0
  200. data/vendor/Eigen/src/Core/util/Memory.h +1163 -0
  201. data/vendor/Eigen/src/Core/util/Meta.h +812 -0
  202. data/vendor/Eigen/src/Core/util/NonMPL2.h +3 -0
  203. data/vendor/Eigen/src/Core/util/ReenableStupidWarnings.h +31 -0
  204. data/vendor/Eigen/src/Core/util/ReshapedHelper.h +51 -0
  205. data/vendor/Eigen/src/Core/util/StaticAssert.h +221 -0
  206. data/vendor/Eigen/src/Core/util/SymbolicIndex.h +293 -0
  207. data/vendor/Eigen/src/Core/util/XprHelper.h +856 -0
  208. data/vendor/Eigen/src/Eigenvalues/ComplexEigenSolver.h +346 -0
  209. data/vendor/Eigen/src/Eigenvalues/ComplexSchur.h +462 -0
  210. data/vendor/Eigen/src/Eigenvalues/ComplexSchur_LAPACKE.h +91 -0
  211. data/vendor/Eigen/src/Eigenvalues/EigenSolver.h +622 -0
  212. data/vendor/Eigen/src/Eigenvalues/GeneralizedEigenSolver.h +418 -0
  213. data/vendor/Eigen/src/Eigenvalues/GeneralizedSelfAdjointEigenSolver.h +226 -0
  214. data/vendor/Eigen/src/Eigenvalues/HessenbergDecomposition.h +374 -0
  215. data/vendor/Eigen/src/Eigenvalues/MatrixBaseEigenvalues.h +158 -0
  216. data/vendor/Eigen/src/Eigenvalues/RealQZ.h +657 -0
  217. data/vendor/Eigen/src/Eigenvalues/RealSchur.h +558 -0
  218. data/vendor/Eigen/src/Eigenvalues/RealSchur_LAPACKE.h +77 -0
  219. data/vendor/Eigen/src/Eigenvalues/SelfAdjointEigenSolver.h +904 -0
  220. data/vendor/Eigen/src/Eigenvalues/SelfAdjointEigenSolver_LAPACKE.h +87 -0
  221. data/vendor/Eigen/src/Eigenvalues/Tridiagonalization.h +561 -0
  222. data/vendor/Eigen/src/Geometry/AlignedBox.h +486 -0
  223. data/vendor/Eigen/src/Geometry/AngleAxis.h +247 -0
  224. data/vendor/Eigen/src/Geometry/EulerAngles.h +114 -0
  225. data/vendor/Eigen/src/Geometry/Homogeneous.h +501 -0
  226. data/vendor/Eigen/src/Geometry/Hyperplane.h +282 -0
  227. data/vendor/Eigen/src/Geometry/OrthoMethods.h +235 -0
  228. data/vendor/Eigen/src/Geometry/ParametrizedLine.h +232 -0
  229. data/vendor/Eigen/src/Geometry/Quaternion.h +870 -0
  230. data/vendor/Eigen/src/Geometry/Rotation2D.h +199 -0
  231. data/vendor/Eigen/src/Geometry/RotationBase.h +206 -0
  232. data/vendor/Eigen/src/Geometry/Scaling.h +188 -0
  233. data/vendor/Eigen/src/Geometry/Transform.h +1563 -0
  234. data/vendor/Eigen/src/Geometry/Translation.h +202 -0
  235. data/vendor/Eigen/src/Geometry/Umeyama.h +166 -0
  236. data/vendor/Eigen/src/Geometry/arch/Geometry_SIMD.h +168 -0
  237. data/vendor/Eigen/src/Householder/BlockHouseholder.h +110 -0
  238. data/vendor/Eigen/src/Householder/Householder.h +176 -0
  239. data/vendor/Eigen/src/Householder/HouseholderSequence.h +545 -0
  240. data/vendor/Eigen/src/IterativeLinearSolvers/BasicPreconditioners.h +226 -0
  241. data/vendor/Eigen/src/IterativeLinearSolvers/BiCGSTAB.h +212 -0
  242. data/vendor/Eigen/src/IterativeLinearSolvers/ConjugateGradient.h +229 -0
  243. data/vendor/Eigen/src/IterativeLinearSolvers/IncompleteCholesky.h +394 -0
  244. data/vendor/Eigen/src/IterativeLinearSolvers/IncompleteLUT.h +453 -0
  245. data/vendor/Eigen/src/IterativeLinearSolvers/IterativeSolverBase.h +444 -0
  246. data/vendor/Eigen/src/IterativeLinearSolvers/LeastSquareConjugateGradient.h +198 -0
  247. data/vendor/Eigen/src/IterativeLinearSolvers/SolveWithGuess.h +117 -0
  248. data/vendor/Eigen/src/Jacobi/Jacobi.h +483 -0
  249. data/vendor/Eigen/src/KLUSupport/KLUSupport.h +358 -0
  250. data/vendor/Eigen/src/LU/Determinant.h +117 -0
  251. data/vendor/Eigen/src/LU/FullPivLU.h +877 -0
  252. data/vendor/Eigen/src/LU/InverseImpl.h +432 -0
  253. data/vendor/Eigen/src/LU/PartialPivLU.h +624 -0
  254. data/vendor/Eigen/src/LU/PartialPivLU_LAPACKE.h +83 -0
  255. data/vendor/Eigen/src/LU/arch/InverseSize4.h +351 -0
  256. data/vendor/Eigen/src/MetisSupport/MetisSupport.h +137 -0
  257. data/vendor/Eigen/src/OrderingMethods/Amd.h +435 -0
  258. data/vendor/Eigen/src/OrderingMethods/Eigen_Colamd.h +1863 -0
  259. data/vendor/Eigen/src/OrderingMethods/Ordering.h +153 -0
  260. data/vendor/Eigen/src/PaStiXSupport/PaStiXSupport.h +678 -0
  261. data/vendor/Eigen/src/PardisoSupport/PardisoSupport.h +545 -0
  262. data/vendor/Eigen/src/QR/ColPivHouseholderQR.h +674 -0
  263. data/vendor/Eigen/src/QR/ColPivHouseholderQR_LAPACKE.h +97 -0
  264. data/vendor/Eigen/src/QR/CompleteOrthogonalDecomposition.h +635 -0
  265. data/vendor/Eigen/src/QR/FullPivHouseholderQR.h +713 -0
  266. data/vendor/Eigen/src/QR/HouseholderQR.h +434 -0
  267. data/vendor/Eigen/src/QR/HouseholderQR_LAPACKE.h +68 -0
  268. data/vendor/Eigen/src/SPQRSupport/SuiteSparseQRSupport.h +335 -0
  269. data/vendor/Eigen/src/SVD/BDCSVD.h +1366 -0
  270. data/vendor/Eigen/src/SVD/JacobiSVD.h +812 -0
  271. data/vendor/Eigen/src/SVD/JacobiSVD_LAPACKE.h +91 -0
  272. data/vendor/Eigen/src/SVD/SVDBase.h +376 -0
  273. data/vendor/Eigen/src/SVD/UpperBidiagonalization.h +414 -0
  274. data/vendor/Eigen/src/SparseCholesky/SimplicialCholesky.h +697 -0
  275. data/vendor/Eigen/src/SparseCholesky/SimplicialCholesky_impl.h +174 -0
  276. data/vendor/Eigen/src/SparseCore/AmbiVector.h +378 -0
  277. data/vendor/Eigen/src/SparseCore/CompressedStorage.h +274 -0
  278. data/vendor/Eigen/src/SparseCore/ConservativeSparseSparseProduct.h +352 -0
  279. data/vendor/Eigen/src/SparseCore/MappedSparseMatrix.h +67 -0
  280. data/vendor/Eigen/src/SparseCore/SparseAssign.h +270 -0
  281. data/vendor/Eigen/src/SparseCore/SparseBlock.h +571 -0
  282. data/vendor/Eigen/src/SparseCore/SparseColEtree.h +206 -0
  283. data/vendor/Eigen/src/SparseCore/SparseCompressedBase.h +370 -0
  284. data/vendor/Eigen/src/SparseCore/SparseCwiseBinaryOp.h +722 -0
  285. data/vendor/Eigen/src/SparseCore/SparseCwiseUnaryOp.h +150 -0
  286. data/vendor/Eigen/src/SparseCore/SparseDenseProduct.h +342 -0
  287. data/vendor/Eigen/src/SparseCore/SparseDiagonalProduct.h +138 -0
  288. data/vendor/Eigen/src/SparseCore/SparseDot.h +98 -0
  289. data/vendor/Eigen/src/SparseCore/SparseFuzzy.h +29 -0
  290. data/vendor/Eigen/src/SparseCore/SparseMap.h +305 -0
  291. data/vendor/Eigen/src/SparseCore/SparseMatrix.h +1518 -0
  292. data/vendor/Eigen/src/SparseCore/SparseMatrixBase.h +398 -0
  293. data/vendor/Eigen/src/SparseCore/SparsePermutation.h +178 -0
  294. data/vendor/Eigen/src/SparseCore/SparseProduct.h +181 -0
  295. data/vendor/Eigen/src/SparseCore/SparseRedux.h +49 -0
  296. data/vendor/Eigen/src/SparseCore/SparseRef.h +397 -0
  297. data/vendor/Eigen/src/SparseCore/SparseSelfAdjointView.h +659 -0
  298. data/vendor/Eigen/src/SparseCore/SparseSolverBase.h +124 -0
  299. data/vendor/Eigen/src/SparseCore/SparseSparseProductWithPruning.h +198 -0
  300. data/vendor/Eigen/src/SparseCore/SparseTranspose.h +92 -0
  301. data/vendor/Eigen/src/SparseCore/SparseTriangularView.h +189 -0
  302. data/vendor/Eigen/src/SparseCore/SparseUtil.h +186 -0
  303. data/vendor/Eigen/src/SparseCore/SparseVector.h +478 -0
  304. data/vendor/Eigen/src/SparseCore/SparseView.h +254 -0
  305. data/vendor/Eigen/src/SparseCore/TriangularSolver.h +315 -0
  306. data/vendor/Eigen/src/SparseLU/SparseLU.h +923 -0
  307. data/vendor/Eigen/src/SparseLU/SparseLUImpl.h +66 -0
  308. data/vendor/Eigen/src/SparseLU/SparseLU_Memory.h +226 -0
  309. data/vendor/Eigen/src/SparseLU/SparseLU_Structs.h +110 -0
  310. data/vendor/Eigen/src/SparseLU/SparseLU_SupernodalMatrix.h +375 -0
  311. data/vendor/Eigen/src/SparseLU/SparseLU_Utils.h +80 -0
  312. data/vendor/Eigen/src/SparseLU/SparseLU_column_bmod.h +181 -0
  313. data/vendor/Eigen/src/SparseLU/SparseLU_column_dfs.h +179 -0
  314. data/vendor/Eigen/src/SparseLU/SparseLU_copy_to_ucol.h +107 -0
  315. data/vendor/Eigen/src/SparseLU/SparseLU_gemm_kernel.h +280 -0
  316. data/vendor/Eigen/src/SparseLU/SparseLU_heap_relax_snode.h +126 -0
  317. data/vendor/Eigen/src/SparseLU/SparseLU_kernel_bmod.h +130 -0
  318. data/vendor/Eigen/src/SparseLU/SparseLU_panel_bmod.h +223 -0
  319. data/vendor/Eigen/src/SparseLU/SparseLU_panel_dfs.h +258 -0
  320. data/vendor/Eigen/src/SparseLU/SparseLU_pivotL.h +137 -0
  321. data/vendor/Eigen/src/SparseLU/SparseLU_pruneL.h +136 -0
  322. data/vendor/Eigen/src/SparseLU/SparseLU_relax_snode.h +83 -0
  323. data/vendor/Eigen/src/SparseQR/SparseQR.h +758 -0
  324. data/vendor/Eigen/src/StlSupport/StdDeque.h +116 -0
  325. data/vendor/Eigen/src/StlSupport/StdList.h +106 -0
  326. data/vendor/Eigen/src/StlSupport/StdVector.h +131 -0
  327. data/vendor/Eigen/src/StlSupport/details.h +84 -0
  328. data/vendor/Eigen/src/SuperLUSupport/SuperLUSupport.h +1025 -0
  329. data/vendor/Eigen/src/UmfPackSupport/UmfPackSupport.h +642 -0
  330. data/vendor/Eigen/src/misc/Image.h +82 -0
  331. data/vendor/Eigen/src/misc/Kernel.h +79 -0
  332. data/vendor/Eigen/src/misc/RealSvd2x2.h +55 -0
  333. data/vendor/Eigen/src/misc/blas.h +440 -0
  334. data/vendor/Eigen/src/misc/lapack.h +152 -0
  335. data/vendor/Eigen/src/misc/lapacke.h +16292 -0
  336. data/vendor/Eigen/src/misc/lapacke_mangling.h +17 -0
  337. data/vendor/Eigen/src/plugins/ArrayCwiseBinaryOps.h +358 -0
  338. data/vendor/Eigen/src/plugins/ArrayCwiseUnaryOps.h +696 -0
  339. data/vendor/Eigen/src/plugins/BlockMethods.h +1442 -0
  340. data/vendor/Eigen/src/plugins/CommonCwiseBinaryOps.h +115 -0
  341. data/vendor/Eigen/src/plugins/CommonCwiseUnaryOps.h +177 -0
  342. data/vendor/Eigen/src/plugins/IndexedViewMethods.h +262 -0
  343. data/vendor/Eigen/src/plugins/MatrixCwiseBinaryOps.h +152 -0
  344. data/vendor/Eigen/src/plugins/MatrixCwiseUnaryOps.h +95 -0
  345. data/vendor/Eigen/src/plugins/ReshapedMethods.h +149 -0
  346. data/vendor/aarand/aarand.hpp +114 -0
  347. data/vendor/annoy/annoylib.h +1495 -0
  348. data/vendor/annoy/kissrandom.h +120 -0
  349. data/vendor/annoy/mman.h +242 -0
  350. data/vendor/hnswlib/bruteforce.h +152 -0
  351. data/vendor/hnswlib/hnswalg.h +1192 -0
  352. data/vendor/hnswlib/hnswlib.h +108 -0
  353. data/vendor/hnswlib/space_ip.h +282 -0
  354. data/vendor/hnswlib/space_l2.h +281 -0
  355. data/vendor/hnswlib/visited_list_pool.h +79 -0
  356. data/vendor/irlba/irlba.hpp +575 -0
  357. data/vendor/irlba/lanczos.hpp +212 -0
  358. data/vendor/irlba/parallel.hpp +474 -0
  359. data/vendor/irlba/utils.hpp +224 -0
  360. data/vendor/irlba/wrappers.hpp +228 -0
  361. data/vendor/kmeans/Base.hpp +75 -0
  362. data/vendor/kmeans/Details.hpp +79 -0
  363. data/vendor/kmeans/HartiganWong.hpp +492 -0
  364. data/vendor/kmeans/InitializeKmeansPP.hpp +144 -0
  365. data/vendor/kmeans/InitializeNone.hpp +44 -0
  366. data/vendor/kmeans/InitializePCAPartition.hpp +309 -0
  367. data/vendor/kmeans/InitializeRandom.hpp +91 -0
  368. data/vendor/kmeans/Kmeans.hpp +161 -0
  369. data/vendor/kmeans/Lloyd.hpp +134 -0
  370. data/vendor/kmeans/MiniBatch.hpp +269 -0
  371. data/vendor/kmeans/QuickSearch.hpp +179 -0
  372. data/vendor/kmeans/compute_centroids.hpp +32 -0
  373. data/vendor/kmeans/compute_wcss.hpp +27 -0
  374. data/vendor/kmeans/is_edge_case.hpp +42 -0
  375. data/vendor/kmeans/random.hpp +55 -0
  376. data/vendor/knncolle/Annoy/Annoy.hpp +193 -0
  377. data/vendor/knncolle/BruteForce/BruteForce.hpp +120 -0
  378. data/vendor/knncolle/Hnsw/Hnsw.hpp +225 -0
  379. data/vendor/knncolle/Kmknn/Kmknn.hpp +286 -0
  380. data/vendor/knncolle/VpTree/VpTree.hpp +256 -0
  381. data/vendor/knncolle/knncolle.hpp +34 -0
  382. data/vendor/knncolle/utils/Base.hpp +100 -0
  383. data/vendor/knncolle/utils/NeighborQueue.hpp +94 -0
  384. data/vendor/knncolle/utils/distances.hpp +98 -0
  385. data/vendor/knncolle/utils/find_nearest_neighbors.hpp +112 -0
  386. data/vendor/powerit/PowerIterations.hpp +157 -0
  387. data/vendor/umappp/NeighborList.hpp +37 -0
  388. data/vendor/umappp/Umap.hpp +662 -0
  389. data/vendor/umappp/combine_neighbor_sets.hpp +95 -0
  390. data/vendor/umappp/find_ab.hpp +157 -0
  391. data/vendor/umappp/neighbor_similarities.hpp +136 -0
  392. data/vendor/umappp/optimize_layout.hpp +285 -0
  393. data/vendor/umappp/spectral_init.hpp +181 -0
  394. data/vendor/umappp/umappp.hpp +13 -0
  395. metadata +465 -0
@@ -0,0 +1,1419 @@
1
+ // This file is part of Eigen, a lightweight C++ template library
2
+ // for linear algebra.
3
+ //
4
+ // Copyright (C) 2018 Rasmus Munk Larsen <rmlarsen@google.com>
5
+ // Copyright (C) 2020 Antonio Sanchez <cantonios@google.com>
6
+ //
7
+ // This Source Code Form is subject to the terms of the Mozilla
8
+ // Public License v. 2.0. If a copy of the MPL was not distributed
9
+ // with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
10
+
11
+ #ifndef EIGEN_TYPE_CASTING_NEON_H
12
+ #define EIGEN_TYPE_CASTING_NEON_H
13
+
14
+ namespace Eigen {
15
+
16
+ namespace internal {
17
+
18
+ //==============================================================================
19
+ // pcast, SrcType = float
20
+ //==============================================================================
21
+ template <>
22
+ struct type_casting_traits<float, float> {
23
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
24
+ };
25
+ template <>
26
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet4f, Packet4f>(const Packet4f& a) {
27
+ return a;
28
+ }
29
+ template <>
30
+ EIGEN_STRONG_INLINE Packet2f pcast<Packet2f, Packet2f>(const Packet2f& a) {
31
+ return a;
32
+ }
33
+
34
+ template <>
35
+ struct type_casting_traits<float, numext::int64_t> {
36
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
37
+ };
38
+ template <>
39
+ struct type_casting_traits<float, numext::uint64_t> {
40
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
41
+ };
42
+ // If float64 exists, first convert to that to keep as much precision as possible.
43
+ #if EIGEN_ARCH_ARM64
44
+ template <>
45
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet4f, Packet2l>(const Packet4f& a) {
46
+ // Discard second half of input.
47
+ return vcvtq_s64_f64(vcvt_f64_f32(vget_low_f32(a)));
48
+ }
49
+ template <>
50
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet4f, Packet2ul>(const Packet4f& a) {
51
+ // Discard second half of input.
52
+ return vcvtq_u64_f64(vcvt_f64_f32(vget_low_f32(a)));
53
+ }
54
+ #else
55
+ template <>
56
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet4f, Packet2l>(const Packet4f& a) {
57
+ // Discard second half of input.
58
+ return vmovl_s32(vget_low_s32(vcvtq_s32_f32(a)));
59
+ }
60
+ template <>
61
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet4f, Packet2ul>(const Packet4f& a) {
62
+ // Discard second half of input.
63
+ return vmovl_u32(vget_low_u32(vcvtq_u32_f32(a)));
64
+ }
65
+ #endif // EIGEN_ARCH_ARM64
66
+
67
+ template <>
68
+ struct type_casting_traits<float, numext::int32_t> {
69
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
70
+ };
71
+ template <>
72
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet4f, Packet4i>(const Packet4f& a) {
73
+ return vcvtq_s32_f32(a);
74
+ }
75
+ template <>
76
+ EIGEN_STRONG_INLINE Packet2i pcast<Packet2f, Packet2i>(const Packet2f& a) {
77
+ return vcvt_s32_f32(a);
78
+ }
79
+
80
+ template <>
81
+ struct type_casting_traits<float, numext::uint32_t> {
82
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
83
+ };
84
+ template <>
85
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet4f, Packet4ui>(const Packet4f& a) {
86
+ return vcvtq_u32_f32(a);
87
+ }
88
+ template <>
89
+ EIGEN_STRONG_INLINE Packet2ui pcast<Packet2f, Packet2ui>(const Packet2f& a) {
90
+ return vcvt_u32_f32(a);
91
+ }
92
+
93
+ template <>
94
+ struct type_casting_traits<float, numext::int16_t> {
95
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
96
+ };
97
+ template <>
98
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet4f, Packet8s>(const Packet4f& a, const Packet4f& b) {
99
+ return vcombine_s16(vmovn_s32(vcvtq_s32_f32(a)), vmovn_s32(vcvtq_s32_f32(b)));
100
+ }
101
+ template <>
102
+ EIGEN_STRONG_INLINE Packet4s pcast<Packet2f, Packet4s>(const Packet2f& a, const Packet2f& b) {
103
+ return vmovn_s32(vcombine_s32(vcvt_s32_f32(a), vcvt_s32_f32(b)));
104
+ }
105
+
106
+ template <>
107
+ struct type_casting_traits<float, numext::uint16_t> {
108
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
109
+ };
110
+ template <>
111
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet4f, Packet8us>(const Packet4f& a, const Packet4f& b) {
112
+ return vcombine_u16(vmovn_u32(vcvtq_u32_f32(a)), vmovn_u32(vcvtq_u32_f32(b)));
113
+ }
114
+ template <>
115
+ EIGEN_STRONG_INLINE Packet4us pcast<Packet2f, Packet4us>(const Packet2f& a, const Packet2f& b) {
116
+ return vmovn_u32(vcombine_u32(vcvt_u32_f32(a), vcvt_u32_f32(b)));
117
+ }
118
+
119
+ template <>
120
+ struct type_casting_traits<float, numext::int8_t> {
121
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
122
+ };
123
+ template <>
124
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet4f, Packet16c>(const Packet4f& a, const Packet4f& b, const Packet4f& c,
125
+ const Packet4f& d) {
126
+ const int16x8_t ab_s16 = pcast<Packet4f, Packet8s>(a, b);
127
+ const int16x8_t cd_s16 = pcast<Packet4f, Packet8s>(c, d);
128
+ return vcombine_s8(vmovn_s16(ab_s16), vmovn_s16(cd_s16));
129
+ }
130
+ template <>
131
+ EIGEN_STRONG_INLINE Packet8c pcast<Packet2f, Packet8c>(const Packet2f& a, const Packet2f& b, const Packet2f& c,
132
+ const Packet2f& d) {
133
+ const int16x4_t ab_s16 = pcast<Packet2f, Packet4s>(a, b);
134
+ const int16x4_t cd_s16 = pcast<Packet2f, Packet4s>(c, d);
135
+ return vmovn_s16(vcombine_s16(ab_s16, cd_s16));
136
+ }
137
+
138
+ template <>
139
+ struct type_casting_traits<float, numext::uint8_t> {
140
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
141
+ };
142
+ template <>
143
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet4f, Packet16uc>(const Packet4f& a, const Packet4f& b, const Packet4f& c,
144
+ const Packet4f& d) {
145
+ const uint16x8_t ab_u16 = pcast<Packet4f, Packet8us>(a, b);
146
+ const uint16x8_t cd_u16 = pcast<Packet4f, Packet8us>(c, d);
147
+ return vcombine_u8(vmovn_u16(ab_u16), vmovn_u16(cd_u16));
148
+ }
149
+ template <>
150
+ EIGEN_STRONG_INLINE Packet8uc pcast<Packet2f, Packet8uc>(const Packet2f& a, const Packet2f& b, const Packet2f& c,
151
+ const Packet2f& d) {
152
+ const uint16x4_t ab_u16 = pcast<Packet2f, Packet4us>(a, b);
153
+ const uint16x4_t cd_u16 = pcast<Packet2f, Packet4us>(c, d);
154
+ return vmovn_u16(vcombine_u16(ab_u16, cd_u16));
155
+ }
156
+
157
+ //==============================================================================
158
+ // pcast, SrcType = int8_t
159
+ //==============================================================================
160
+ template <>
161
+ struct type_casting_traits<numext::int8_t, float> {
162
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
163
+ };
164
+ template <>
165
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet16c, Packet4f>(const Packet16c& a) {
166
+ // Discard all but first 4 bytes.
167
+ return vcvtq_f32_s32(vmovl_s16(vget_low_s16(vmovl_s8(vget_low_s8(a)))));
168
+ }
169
+ template <>
170
+ EIGEN_STRONG_INLINE Packet2f pcast<Packet8c, Packet2f>(const Packet8c& a) {
171
+ // Discard all but first 2 bytes.
172
+ return vcvt_f32_s32(vget_low_s32(vmovl_s16(vget_low_s16(vmovl_s8(a)))));
173
+ }
174
+
175
+ template <>
176
+ struct type_casting_traits<numext::int8_t, numext::int64_t> {
177
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 8 };
178
+ };
179
+ template <>
180
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet16c, Packet2l>(const Packet16c& a) {
181
+ // Discard all but first two bytes.
182
+ return vmovl_s32(vget_low_s32(vmovl_s16(vget_low_s16(vmovl_s8(vget_low_s8(a))))));
183
+ }
184
+
185
+ template <>
186
+ struct type_casting_traits<numext::int8_t, numext::uint64_t> {
187
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 8 };
188
+ };
189
+ template <>
190
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet16c, Packet2ul>(const Packet16c& a) {
191
+ return vreinterpretq_u64_s64(pcast<Packet16c, Packet2l>(a));
192
+ }
193
+
194
+ template <>
195
+ struct type_casting_traits<numext::int8_t, numext::int32_t> {
196
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
197
+ };
198
+ template <>
199
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet16c, Packet4i>(const Packet16c& a) {
200
+ // Discard all but first 4 bytes.
201
+ return vmovl_s16(vget_low_s16(vmovl_s8(vget_low_s8(a))));
202
+ }
203
+ template <>
204
+ EIGEN_STRONG_INLINE Packet2i pcast<Packet8c, Packet2i>(const Packet8c& a) {
205
+ // Discard all but first 2 bytes.
206
+ return vget_low_s32(vmovl_s16(vget_low_s16(vmovl_s8(a))));
207
+ }
208
+
209
+ template <>
210
+ struct type_casting_traits<numext::int8_t, numext::uint32_t> {
211
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
212
+ };
213
+ template <>
214
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet16c, Packet4ui>(const Packet16c& a) {
215
+ return vreinterpretq_u32_s32(pcast<Packet16c, Packet4i>(a));
216
+ }
217
+ template <>
218
+ EIGEN_STRONG_INLINE Packet2ui pcast<Packet8c, Packet2ui>(const Packet8c& a) {
219
+ return vreinterpret_u32_s32(pcast<Packet8c, Packet2i>(a));
220
+ }
221
+
222
+ template <>
223
+ struct type_casting_traits<numext::int8_t, numext::int16_t> {
224
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
225
+ };
226
+ template <>
227
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet16c, Packet8s>(const Packet16c& a) {
228
+ // Discard second half of input.
229
+ return vmovl_s8(vget_low_s8(a));
230
+ }
231
+ template <>
232
+ EIGEN_STRONG_INLINE Packet4s pcast<Packet8c, Packet4s>(const Packet8c& a) {
233
+ // Discard second half of input.
234
+ return vget_low_s16(vmovl_s8(a));
235
+ }
236
+
237
+ template <>
238
+ struct type_casting_traits<numext::int8_t, numext::uint16_t> {
239
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
240
+ };
241
+ template <>
242
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet16c, Packet8us>(const Packet16c& a) {
243
+ return vreinterpretq_u16_s16(pcast<Packet16c, Packet8s>(a));
244
+ }
245
+ template <>
246
+ EIGEN_STRONG_INLINE Packet4us pcast<Packet8c, Packet4us>(const Packet8c& a) {
247
+ return vreinterpret_u16_s16(pcast<Packet8c, Packet4s>(a));
248
+ }
249
+
250
+ template <>
251
+ struct type_casting_traits<numext::int8_t, numext::int8_t> {
252
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
253
+ };
254
+ template <>
255
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet16c, Packet16c>(const Packet16c& a) {
256
+ return a;
257
+ }
258
+ template <>
259
+ EIGEN_STRONG_INLINE Packet8c pcast<Packet8c, Packet8c>(const Packet8c& a) {
260
+ return a;
261
+ }
262
+ template <>
263
+ EIGEN_STRONG_INLINE Packet4c pcast<Packet4c, Packet4c>(const Packet4c& a) {
264
+ return a;
265
+ }
266
+
267
+ template <>
268
+ struct type_casting_traits<numext::int8_t, numext::uint8_t> {
269
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
270
+ };
271
+ template <>
272
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet16c, Packet16uc>(const Packet16c& a) {
273
+ return vreinterpretq_u8_s8(a);
274
+ }
275
+ template <>
276
+ EIGEN_STRONG_INLINE Packet8uc pcast<Packet8c, Packet8uc>(const Packet8c& a) {
277
+ return vreinterpret_u8_s8(a);
278
+ }
279
+ template <>
280
+ EIGEN_STRONG_INLINE Packet4uc pcast<Packet4c, Packet4uc>(const Packet4c& a) {
281
+ return static_cast<Packet4uc>(a);
282
+ }
283
+
284
+ //==============================================================================
285
+ // pcast, SrcType = uint8_t
286
+ //==============================================================================
287
+ template <>
288
+ struct type_casting_traits<numext::uint8_t, float> {
289
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
290
+ };
291
+ template <>
292
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet16uc, Packet4f>(const Packet16uc& a) {
293
+ // Discard all but first 4 bytes.
294
+ return vcvtq_f32_u32(vmovl_u16(vget_low_u16(vmovl_u8(vget_low_u8(a)))));
295
+ }
296
+ template <>
297
+ EIGEN_STRONG_INLINE Packet2f pcast<Packet8uc, Packet2f>(const Packet8uc& a) {
298
+ // Discard all but first 2 bytes.
299
+ return vcvt_f32_u32(vget_low_u32(vmovl_u16(vget_low_u16(vmovl_u8(a)))));
300
+ }
301
+
302
+ template <>
303
+ struct type_casting_traits<numext::uint8_t, numext::uint64_t> {
304
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 8 };
305
+ };
306
+ template <>
307
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet16uc, Packet2ul>(const Packet16uc& a) {
308
+ // Discard all but first two bytes.
309
+ return vmovl_u32(vget_low_u32(vmovl_u16(vget_low_u16(vmovl_u8(vget_low_u8(a))))));
310
+ }
311
+
312
+ template <>
313
+ struct type_casting_traits<numext::uint8_t, numext::int64_t> {
314
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 8 };
315
+ };
316
+ template <>
317
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet16uc, Packet2l>(const Packet16uc& a) {
318
+ return vreinterpretq_s64_u64(pcast<Packet16uc, Packet2ul>(a));
319
+ }
320
+
321
+ template <>
322
+ struct type_casting_traits<numext::uint8_t, numext::uint32_t> {
323
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
324
+ };
325
+ template <>
326
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet16uc, Packet4ui>(const Packet16uc& a) {
327
+ // Discard all but first 4 bytes.
328
+ return vmovl_u16(vget_low_u16(vmovl_u8(vget_low_u8(a))));
329
+ }
330
+ template <>
331
+ EIGEN_STRONG_INLINE Packet2ui pcast<Packet8uc, Packet2ui>(const Packet8uc& a) {
332
+ // Discard all but first 2 bytes.
333
+ return vget_low_u32(vmovl_u16(vget_low_u16(vmovl_u8(a))));
334
+ }
335
+
336
+ template <>
337
+ struct type_casting_traits<numext::uint8_t, numext::int32_t> {
338
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
339
+ };
340
+ template <>
341
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet16uc, Packet4i>(const Packet16uc& a) {
342
+ return vreinterpretq_s32_u32(pcast<Packet16uc, Packet4ui>(a));
343
+ }
344
+ template <>
345
+ EIGEN_STRONG_INLINE Packet2i pcast<Packet8uc, Packet2i>(const Packet8uc& a) {
346
+ return vreinterpret_s32_u32(pcast<Packet8uc, Packet2ui>(a));
347
+ }
348
+
349
+ template <>
350
+ struct type_casting_traits<numext::uint8_t, numext::uint16_t> {
351
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
352
+ };
353
+ template <>
354
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet16uc, Packet8us>(const Packet16uc& a) {
355
+ // Discard second half of input.
356
+ return vmovl_u8(vget_low_u8(a));
357
+ }
358
+ template <>
359
+ EIGEN_STRONG_INLINE Packet4us pcast<Packet8uc, Packet4us>(const Packet8uc& a) {
360
+ // Discard second half of input.
361
+ return vget_low_u16(vmovl_u8(a));
362
+ }
363
+
364
+ template <>
365
+ struct type_casting_traits<numext::uint8_t, numext::int16_t> {
366
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
367
+ };
368
+ template <>
369
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet16uc, Packet8s>(const Packet16uc& a) {
370
+ return vreinterpretq_s16_u16(pcast<Packet16uc, Packet8us>(a));
371
+ }
372
+ template <>
373
+ EIGEN_STRONG_INLINE Packet4s pcast<Packet8uc, Packet4s>(const Packet8uc& a) {
374
+ return vreinterpret_s16_u16(pcast<Packet8uc, Packet4us>(a));
375
+ }
376
+
377
+ template <>
378
+ struct type_casting_traits<numext::uint8_t, numext::uint8_t> {
379
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
380
+ };
381
+ template <>
382
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet16uc, Packet16uc>(const Packet16uc& a) {
383
+ return a;
384
+ }
385
+ template <>
386
+ EIGEN_STRONG_INLINE Packet8uc pcast<Packet8uc, Packet8uc>(const Packet8uc& a) {
387
+ return a;
388
+ }
389
+ template <>
390
+ EIGEN_STRONG_INLINE Packet4uc pcast<Packet4uc, Packet4uc>(const Packet4uc& a) {
391
+ return a;
392
+ }
393
+
394
+ template <>
395
+ struct type_casting_traits<numext::uint8_t, numext::int8_t> {
396
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
397
+ };
398
+ template <>
399
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet16uc, Packet16c>(const Packet16uc& a) {
400
+ return vreinterpretq_s8_u8(a);
401
+ }
402
+ template <>
403
+ EIGEN_STRONG_INLINE Packet8c pcast<Packet8uc, Packet8c>(const Packet8uc& a) {
404
+ return vreinterpret_s8_u8(a);
405
+ }
406
+ template <>
407
+ EIGEN_STRONG_INLINE Packet4c pcast<Packet4uc, Packet4c>(const Packet4uc& a) {
408
+ return static_cast<Packet4c>(a);
409
+ }
410
+
411
+ //==============================================================================
412
+ // pcast, SrcType = int16_t
413
+ //==============================================================================
414
+ template <>
415
+ struct type_casting_traits<numext::int16_t, float> {
416
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
417
+ };
418
+ template <>
419
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet8s, Packet4f>(const Packet8s& a) {
420
+ // Discard second half of input.
421
+ return vcvtq_f32_s32(vmovl_s16(vget_low_s16(a)));
422
+ }
423
+ template <>
424
+ EIGEN_STRONG_INLINE Packet2f pcast<Packet4s, Packet2f>(const Packet4s& a) {
425
+ // Discard second half of input.
426
+ return vcvt_f32_s32(vget_low_s32(vmovl_s16(a)));
427
+ }
428
+
429
+ template <>
430
+ struct type_casting_traits<numext::int16_t, numext::int64_t> {
431
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
432
+ };
433
+ template <>
434
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet8s, Packet2l>(const Packet8s& a) {
435
+ // Discard all but first two values.
436
+ return vmovl_s32(vget_low_s32(vmovl_s16(vget_low_s16(a))));
437
+ }
438
+
439
+ template <>
440
+ struct type_casting_traits<numext::int16_t, numext::uint64_t> {
441
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
442
+ };
443
+ template <>
444
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet8s, Packet2ul>(const Packet8s& a) {
445
+ return vreinterpretq_u64_s64(pcast<Packet8s, Packet2l>(a));
446
+ }
447
+
448
+ template <>
449
+ struct type_casting_traits<numext::int16_t, numext::int32_t> {
450
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
451
+ };
452
+ template <>
453
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet8s, Packet4i>(const Packet8s& a) {
454
+ // Discard second half of input.
455
+ return vmovl_s16(vget_low_s16(a));
456
+ }
457
+ template <>
458
+ EIGEN_STRONG_INLINE Packet2i pcast<Packet4s, Packet2i>(const Packet4s& a) {
459
+ // Discard second half of input.
460
+ return vget_low_s32(vmovl_s16(a));
461
+ }
462
+
463
+ template <>
464
+ struct type_casting_traits<numext::int16_t, numext::uint32_t> {
465
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
466
+ };
467
+ template <>
468
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet8s, Packet4ui>(const Packet8s& a) {
469
+ return vreinterpretq_u32_s32(pcast<Packet8s, Packet4i>(a));
470
+ }
471
+ template <>
472
+ EIGEN_STRONG_INLINE Packet2ui pcast<Packet4s, Packet2ui>(const Packet4s& a) {
473
+ return vreinterpret_u32_s32(pcast<Packet4s, Packet2i>(a));
474
+ }
475
+
476
+ template <>
477
+ struct type_casting_traits<numext::int16_t, numext::int16_t> {
478
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
479
+ };
480
+ template <>
481
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet8s, Packet8s>(const Packet8s& a) {
482
+ return a;
483
+ }
484
+ template <>
485
+ EIGEN_STRONG_INLINE Packet4s pcast<Packet4s, Packet4s>(const Packet4s& a) {
486
+ return a;
487
+ }
488
+
489
+ template <>
490
+ struct type_casting_traits<numext::int16_t, numext::uint16_t> {
491
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
492
+ };
493
+ template <>
494
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet8s, Packet8us>(const Packet8s& a) {
495
+ return vreinterpretq_u16_s16(a);
496
+ }
497
+ template <>
498
+ EIGEN_STRONG_INLINE Packet4us pcast<Packet4s, Packet4us>(const Packet4s& a) {
499
+ return vreinterpret_u16_s16(a);
500
+ }
501
+
502
+ template <>
503
+ struct type_casting_traits<numext::int16_t, numext::int8_t> {
504
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
505
+ };
506
+ template <>
507
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet8s, Packet16c>(const Packet8s& a, const Packet8s& b) {
508
+ return vcombine_s8(vmovn_s16(a), vmovn_s16(b));
509
+ }
510
+ template <>
511
+ EIGEN_STRONG_INLINE Packet8c pcast<Packet4s, Packet8c>(const Packet4s& a, const Packet4s& b) {
512
+ return vmovn_s16(vcombine_s16(a, b));
513
+ }
514
+
515
+ template <>
516
+ struct type_casting_traits<numext::int16_t, numext::uint8_t> {
517
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
518
+ };
519
+ template <>
520
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet8s, Packet16uc>(const Packet8s& a, const Packet8s& b) {
521
+ return vcombine_u8(vmovn_u16(vreinterpretq_u16_s16(a)), vmovn_u16(vreinterpretq_u16_s16(b)));
522
+ }
523
+ template <>
524
+ EIGEN_STRONG_INLINE Packet8uc pcast<Packet4s, Packet8uc>(const Packet4s& a, const Packet4s& b) {
525
+ return vmovn_u16(vcombine_u16(vreinterpret_u16_s16(a), vreinterpret_u16_s16(b)));
526
+ }
527
+
528
+ //==============================================================================
529
+ // pcast, SrcType = uint16_t
530
+ //==============================================================================
531
+ template <>
532
+ struct type_casting_traits<numext::uint16_t, float> {
533
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
534
+ };
535
+ template <>
536
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet8us, Packet4f>(const Packet8us& a) {
537
+ // Discard second half of input.
538
+ return vcvtq_f32_u32(vmovl_u16(vget_low_u16(a)));
539
+ }
540
+ template <>
541
+ EIGEN_STRONG_INLINE Packet2f pcast<Packet4us, Packet2f>(const Packet4us& a) {
542
+ // Discard second half of input.
543
+ return vcvt_f32_u32(vget_low_u32(vmovl_u16(a)));
544
+ }
545
+
546
+ template <>
547
+ struct type_casting_traits<numext::uint16_t, numext::uint64_t> {
548
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
549
+ };
550
+ template <>
551
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet8us, Packet2ul>(const Packet8us& a) {
552
+ // Discard all but first two values.
553
+ return vmovl_u32(vget_low_u32(vmovl_u16(vget_low_u16(a))));
554
+ }
555
+
556
+ template <>
557
+ struct type_casting_traits<numext::uint16_t, numext::int64_t> {
558
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
559
+ };
560
+ template <>
561
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet8us, Packet2l>(const Packet8us& a) {
562
+ return vreinterpretq_s64_u64(pcast<Packet8us, Packet2ul>(a));
563
+ }
564
+
565
+ template <>
566
+ struct type_casting_traits<numext::uint16_t, numext::uint32_t> {
567
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
568
+ };
569
+ template <>
570
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet8us, Packet4ui>(const Packet8us& a) {
571
+ // Discard second half of input.
572
+ return vmovl_u16(vget_low_u16(a));
573
+ }
574
+ template <>
575
+ EIGEN_STRONG_INLINE Packet2ui pcast<Packet4us, Packet2ui>(const Packet4us& a) {
576
+ // Discard second half of input.
577
+ return vget_low_u32(vmovl_u16(a));
578
+ }
579
+
580
+ template <>
581
+ struct type_casting_traits<numext::uint16_t, numext::int32_t> {
582
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
583
+ };
584
+ template <>
585
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet8us, Packet4i>(const Packet8us& a) {
586
+ return vreinterpretq_s32_u32(pcast<Packet8us, Packet4ui>(a));
587
+ }
588
+ template <>
589
+ EIGEN_STRONG_INLINE Packet2i pcast<Packet4us, Packet2i>(const Packet4us& a) {
590
+ return vreinterpret_s32_u32(pcast<Packet4us, Packet2ui>(a));
591
+ }
592
+
593
+ template <>
594
+ struct type_casting_traits<numext::uint16_t, numext::uint16_t> {
595
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
596
+ };
597
+ template <>
598
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet8us, Packet8us>(const Packet8us& a) {
599
+ return a;
600
+ }
601
+ template <>
602
+ EIGEN_STRONG_INLINE Packet4us pcast<Packet4us, Packet4us>(const Packet4us& a) {
603
+ return a;
604
+ }
605
+
606
+ template <>
607
+ struct type_casting_traits<numext::uint16_t, numext::int16_t> {
608
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
609
+ };
610
+ template <>
611
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet8us, Packet8s>(const Packet8us& a) {
612
+ return vreinterpretq_s16_u16(a);
613
+ }
614
+ template <>
615
+ EIGEN_STRONG_INLINE Packet4s pcast<Packet4us, Packet4s>(const Packet4us& a) {
616
+ return vreinterpret_s16_u16(a);
617
+ }
618
+
619
+ template <>
620
+ struct type_casting_traits<numext::uint16_t, numext::uint8_t> {
621
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
622
+ };
623
+ template <>
624
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet8us, Packet16uc>(const Packet8us& a, const Packet8us& b) {
625
+ return vcombine_u8(vmovn_u16(a), vmovn_u16(b));
626
+ }
627
+ template <>
628
+ EIGEN_STRONG_INLINE Packet8uc pcast<Packet4us, Packet8uc>(const Packet4us& a, const Packet4us& b) {
629
+ return vmovn_u16(vcombine_u16(a, b));
630
+ }
631
+
632
+ template <>
633
+ struct type_casting_traits<numext::uint16_t, numext::int8_t> {
634
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
635
+ };
636
+ template <>
637
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet8us, Packet16c>(const Packet8us& a, const Packet8us& b) {
638
+ return vreinterpretq_s8_u8(pcast<Packet8us, Packet16uc>(a, b));
639
+ }
640
+ template <>
641
+ EIGEN_STRONG_INLINE Packet8c pcast<Packet4us, Packet8c>(const Packet4us& a, const Packet4us& b) {
642
+ return vreinterpret_s8_u8(pcast<Packet4us, Packet8uc>(a, b));
643
+ }
644
+
645
+ //==============================================================================
646
+ // pcast, SrcType = int32_t
647
+ //==============================================================================
648
+ template <>
649
+ struct type_casting_traits<numext::int32_t, float> {
650
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
651
+ };
652
+ template <>
653
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet4i, Packet4f>(const Packet4i& a) {
654
+ return vcvtq_f32_s32(a);
655
+ }
656
+ template <>
657
+ EIGEN_STRONG_INLINE Packet2f pcast<Packet2i, Packet2f>(const Packet2i& a) {
658
+ return vcvt_f32_s32(a);
659
+ }
660
+
661
+ template <>
662
+ struct type_casting_traits<numext::int32_t, numext::int64_t> {
663
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
664
+ };
665
+ template <>
666
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet4i, Packet2l>(const Packet4i& a) {
667
+ // Discard second half of input.
668
+ return vmovl_s32(vget_low_s32(a));
669
+ }
670
+
671
+ template <>
672
+ struct type_casting_traits<numext::int32_t, numext::uint64_t> {
673
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
674
+ };
675
+ template <>
676
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet4i, Packet2ul>(const Packet4i& a) {
677
+ return vreinterpretq_u64_s64(pcast<Packet4i, Packet2l>(a));
678
+ }
679
+
680
+ template <>
681
+ struct type_casting_traits<numext::int32_t, numext::int32_t> {
682
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
683
+ };
684
+ template <>
685
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet4i, Packet4i>(const Packet4i& a) {
686
+ return a;
687
+ }
688
+ template <>
689
+ EIGEN_STRONG_INLINE Packet2i pcast<Packet2i, Packet2i>(const Packet2i& a) {
690
+ return a;
691
+ }
692
+
693
+ template <>
694
+ struct type_casting_traits<numext::int32_t, numext::uint32_t> {
695
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
696
+ };
697
+ template <>
698
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet4i, Packet4ui>(const Packet4i& a) {
699
+ return vreinterpretq_u32_s32(a);
700
+ }
701
+ template <>
702
+ EIGEN_STRONG_INLINE Packet2ui pcast<Packet2i, Packet2ui>(const Packet2i& a) {
703
+ return vreinterpret_u32_s32(a);
704
+ }
705
+
706
+ template <>
707
+ struct type_casting_traits<numext::int32_t, numext::int16_t> {
708
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
709
+ };
710
+ template <>
711
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet4i, Packet8s>(const Packet4i& a, const Packet4i& b) {
712
+ return vcombine_s16(vmovn_s32(a), vmovn_s32(b));
713
+ }
714
+ template <>
715
+ EIGEN_STRONG_INLINE Packet4s pcast<Packet2i, Packet4s>(const Packet2i& a, const Packet2i& b) {
716
+ return vmovn_s32(vcombine_s32(a, b));
717
+ }
718
+
719
+ template <>
720
+ struct type_casting_traits<numext::int32_t, numext::uint16_t> {
721
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
722
+ };
723
+ template <>
724
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet4i, Packet8us>(const Packet4i& a, const Packet4i& b) {
725
+ return vcombine_u16(vmovn_u32(vreinterpretq_u32_s32(a)), vmovn_u32(vreinterpretq_u32_s32(b)));
726
+ }
727
+ template <>
728
+ EIGEN_STRONG_INLINE Packet4us pcast<Packet2i, Packet4us>(const Packet2i& a, const Packet2i& b) {
729
+ return vmovn_u32(vreinterpretq_u32_s32(vcombine_s32(a, b)));
730
+ }
731
+
732
+ template <>
733
+ struct type_casting_traits<numext::int32_t, numext::int8_t> {
734
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
735
+ };
736
+ template <>
737
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet4i, Packet16c>(const Packet4i& a, const Packet4i& b, const Packet4i& c,
738
+ const Packet4i& d) {
739
+ const int16x8_t ab_s16 = pcast<Packet4i, Packet8s>(a, b);
740
+ const int16x8_t cd_s16 = pcast<Packet4i, Packet8s>(c, d);
741
+ return vcombine_s8(vmovn_s16(ab_s16), vmovn_s16(cd_s16));
742
+ }
743
+ template <>
744
+ EIGEN_STRONG_INLINE Packet8c pcast<Packet2i, Packet8c>(const Packet2i& a, const Packet2i& b, const Packet2i& c,
745
+ const Packet2i& d) {
746
+ const int16x4_t ab_s16 = vmovn_s32(vcombine_s32(a, b));
747
+ const int16x4_t cd_s16 = vmovn_s32(vcombine_s32(c, d));
748
+ return vmovn_s16(vcombine_s16(ab_s16, cd_s16));
749
+ }
750
+
751
+ template <>
752
+ struct type_casting_traits<numext::int32_t, numext::uint8_t> {
753
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
754
+ };
755
+ template <>
756
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet4i, Packet16uc>(const Packet4i& a, const Packet4i& b, const Packet4i& c,
757
+ const Packet4i& d) {
758
+ const uint16x8_t ab_u16 = pcast<Packet4i, Packet8us>(a, b);
759
+ const uint16x8_t cd_u16 = pcast<Packet4i, Packet8us>(c, d);
760
+ return vcombine_u8(vmovn_u16(ab_u16), vmovn_u16(cd_u16));
761
+ }
762
+ template <>
763
+ EIGEN_STRONG_INLINE Packet8uc pcast<Packet2i, Packet8uc>(const Packet2i& a, const Packet2i& b, const Packet2i& c,
764
+ const Packet2i& d) {
765
+ const uint16x4_t ab_u16 = pcast<Packet2i, Packet4us>(a, b);
766
+ const uint16x4_t cd_u16 = pcast<Packet2i, Packet4us>(c, d);
767
+ return vmovn_u16(vcombine_u16(ab_u16, cd_u16));
768
+ }
769
+
770
+ //==============================================================================
771
+ // pcast, SrcType = uint32_t
772
+ //==============================================================================
773
+ template <>
774
+ struct type_casting_traits<numext::uint32_t, float> {
775
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
776
+ };
777
+ template <>
778
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet4ui, Packet4f>(const Packet4ui& a) {
779
+ return vcvtq_f32_u32(a);
780
+ }
781
+ template <>
782
+ EIGEN_STRONG_INLINE Packet2f pcast<Packet2ui, Packet2f>(const Packet2ui& a) {
783
+ return vcvt_f32_u32(a);
784
+ }
785
+
786
+ template <>
787
+ struct type_casting_traits<numext::uint32_t, numext::uint64_t> {
788
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
789
+ };
790
+ template <>
791
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet4ui, Packet2ul>(const Packet4ui& a) {
792
+ // Discard second half of input.
793
+ return vmovl_u32(vget_low_u32(a));
794
+ }
795
+
796
+ template <>
797
+ struct type_casting_traits<numext::uint32_t, numext::int64_t> {
798
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
799
+ };
800
+ template <>
801
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet4ui, Packet2l>(const Packet4ui& a) {
802
+ return vreinterpretq_s64_u64(pcast<Packet4ui, Packet2ul>(a));
803
+ }
804
+
805
+ template <>
806
+ struct type_casting_traits<numext::uint32_t, numext::uint32_t> {
807
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
808
+ };
809
+ template <>
810
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet4ui, Packet4ui>(const Packet4ui& a) {
811
+ return a;
812
+ }
813
+ template <>
814
+ EIGEN_STRONG_INLINE Packet2ui pcast<Packet2ui, Packet2ui>(const Packet2ui& a) {
815
+ return a;
816
+ }
817
+
818
+ template <>
819
+ struct type_casting_traits<numext::uint32_t, numext::int32_t> {
820
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
821
+ };
822
+ template <>
823
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet4ui, Packet4i>(const Packet4ui& a) {
824
+ return vreinterpretq_s32_u32(a);
825
+ }
826
+ template <>
827
+ EIGEN_STRONG_INLINE Packet2i pcast<Packet2ui, Packet2i>(const Packet2ui& a) {
828
+ return vreinterpret_s32_u32(a);
829
+ }
830
+
831
+ template <>
832
+ struct type_casting_traits<numext::uint32_t, numext::uint16_t> {
833
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
834
+ };
835
+ template <>
836
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet4ui, Packet8us>(const Packet4ui& a, const Packet4ui& b) {
837
+ return vcombine_u16(vmovn_u32(a), vmovn_u32(b));
838
+ }
839
+ template <>
840
+ EIGEN_STRONG_INLINE Packet4us pcast<Packet2ui, Packet4us>(const Packet2ui& a, const Packet2ui& b) {
841
+ return vmovn_u32(vcombine_u32(a, b));
842
+ }
843
+
844
+ template <>
845
+ struct type_casting_traits<numext::uint32_t, numext::int16_t> {
846
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
847
+ };
848
+ template <>
849
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet4ui, Packet8s>(const Packet4ui& a, const Packet4ui& b) {
850
+ return vreinterpretq_s16_u16(pcast<Packet4ui, Packet8us>(a, b));
851
+ }
852
+ template <>
853
+ EIGEN_STRONG_INLINE Packet4s pcast<Packet2ui, Packet4s>(const Packet2ui& a, const Packet2ui& b) {
854
+ return vreinterpret_s16_u16(pcast<Packet2ui, Packet4us>(a, b));
855
+ }
856
+
857
+ template <>
858
+ struct type_casting_traits<numext::uint32_t, numext::uint8_t> {
859
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
860
+ };
861
+ template <>
862
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet4ui, Packet16uc>(const Packet4ui& a, const Packet4ui& b, const Packet4ui& c,
863
+ const Packet4ui& d) {
864
+ const uint16x8_t ab_u16 = vcombine_u16(vmovn_u32(a), vmovn_u32(b));
865
+ const uint16x8_t cd_u16 = vcombine_u16(vmovn_u32(c), vmovn_u32(d));
866
+ return vcombine_u8(vmovn_u16(ab_u16), vmovn_u16(cd_u16));
867
+ }
868
+ template <>
869
+ EIGEN_STRONG_INLINE Packet8uc pcast<Packet2ui, Packet8uc>(const Packet2ui& a, const Packet2ui& b, const Packet2ui& c,
870
+ const Packet2ui& d) {
871
+ const uint16x4_t ab_u16 = vmovn_u32(vcombine_u32(a, b));
872
+ const uint16x4_t cd_u16 = vmovn_u32(vcombine_u32(c, d));
873
+ return vmovn_u16(vcombine_u16(ab_u16, cd_u16));
874
+ }
875
+
876
+ template <>
877
+ struct type_casting_traits<numext::uint32_t, numext::int8_t> {
878
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
879
+ };
880
+ template <>
881
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet4ui, Packet16c>(const Packet4ui& a, const Packet4ui& b, const Packet4ui& c,
882
+ const Packet4ui& d) {
883
+ return vreinterpretq_s8_u8(pcast<Packet4ui, Packet16uc>(a, b, c, d));
884
+ }
885
+ template <>
886
+ EIGEN_STRONG_INLINE Packet8c pcast<Packet2ui, Packet8c>(const Packet2ui& a, const Packet2ui& b, const Packet2ui& c,
887
+ const Packet2ui& d) {
888
+ return vreinterpret_s8_u8(pcast<Packet2ui, Packet8uc>(a, b, c, d));
889
+ }
890
+
891
+ //==============================================================================
892
+ // pcast, SrcType = int64_t
893
+ //==============================================================================
894
+ template <>
895
+ struct type_casting_traits<numext::int64_t, float> {
896
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
897
+ };
898
+ template <>
899
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet2l, Packet4f>(const Packet2l& a, const Packet2l& b) {
900
+ return vcvtq_f32_s32(vcombine_s32(vmovn_s64(a), vmovn_s64(b)));
901
+ }
902
+
903
+ template <>
904
+ struct type_casting_traits<numext::int64_t, numext::int64_t> {
905
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
906
+ };
907
+ template <>
908
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet2l, Packet2l>(const Packet2l& a) {
909
+ return a;
910
+ }
911
+
912
+ template <>
913
+ struct type_casting_traits<numext::int64_t, numext::uint64_t> {
914
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
915
+ };
916
+ template <>
917
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet2l, Packet2ul>(const Packet2l& a) {
918
+ return vreinterpretq_u64_s64(a);
919
+ }
920
+
921
+ template <>
922
+ struct type_casting_traits<numext::int64_t, numext::int32_t> {
923
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
924
+ };
925
+ template <>
926
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet2l, Packet4i>(const Packet2l& a, const Packet2l& b) {
927
+ return vcombine_s32(vmovn_s64(a), vmovn_s64(b));
928
+ }
929
+
930
+ template <>
931
+ struct type_casting_traits<numext::int64_t, numext::uint32_t> {
932
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
933
+ };
934
+ template <>
935
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet2l, Packet4ui>(const Packet2l& a, const Packet2l& b) {
936
+ return vcombine_u32(vmovn_u64(vreinterpretq_u64_s64(a)), vmovn_u64(vreinterpretq_u64_s64(b)));
937
+ }
938
+
939
+ template <>
940
+ struct type_casting_traits<numext::int64_t, numext::int16_t> {
941
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
942
+ };
943
+ template <>
944
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet2l, Packet8s>(const Packet2l& a, const Packet2l& b, const Packet2l& c,
945
+ const Packet2l& d) {
946
+ const int32x4_t ab_s32 = pcast<Packet2l, Packet4i>(a, b);
947
+ const int32x4_t cd_s32 = pcast<Packet2l, Packet4i>(c, d);
948
+ return vcombine_s16(vmovn_s32(ab_s32), vmovn_s32(cd_s32));
949
+ }
950
+
951
+ template <>
952
+ struct type_casting_traits<numext::int64_t, numext::uint16_t> {
953
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
954
+ };
955
+ template <>
956
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet2l, Packet8us>(const Packet2l& a, const Packet2l& b, const Packet2l& c,
957
+ const Packet2l& d) {
958
+ const uint32x4_t ab_u32 = pcast<Packet2l, Packet4ui>(a, b);
959
+ const uint32x4_t cd_u32 = pcast<Packet2l, Packet4ui>(c, d);
960
+ return vcombine_u16(vmovn_u32(ab_u32), vmovn_u32(cd_u32));
961
+ }
962
+
963
+ template <>
964
+ struct type_casting_traits<numext::int64_t, numext::int8_t> {
965
+ enum { VectorizedCast = 1, SrcCoeffRatio = 8, TgtCoeffRatio = 1 };
966
+ };
967
+ template <>
968
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet2l, Packet16c>(const Packet2l& a, const Packet2l& b, const Packet2l& c,
969
+ const Packet2l& d, const Packet2l& e, const Packet2l& f,
970
+ const Packet2l& g, const Packet2l& h) {
971
+ const int16x8_t abcd_s16 = pcast<Packet2l, Packet8s>(a, b, c, d);
972
+ const int16x8_t efgh_s16 = pcast<Packet2l, Packet8s>(e, f, g, h);
973
+ return vcombine_s8(vmovn_s16(abcd_s16), vmovn_s16(efgh_s16));
974
+ }
975
+
976
+ template <>
977
+ struct type_casting_traits<numext::int64_t, numext::uint8_t> {
978
+ enum { VectorizedCast = 1, SrcCoeffRatio = 8, TgtCoeffRatio = 1 };
979
+ };
980
+ template <>
981
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet2l, Packet16uc>(const Packet2l& a, const Packet2l& b, const Packet2l& c,
982
+ const Packet2l& d, const Packet2l& e, const Packet2l& f,
983
+ const Packet2l& g, const Packet2l& h) {
984
+ const uint16x8_t abcd_u16 = pcast<Packet2l, Packet8us>(a, b, c, d);
985
+ const uint16x8_t efgh_u16 = pcast<Packet2l, Packet8us>(e, f, g, h);
986
+ return vcombine_u8(vmovn_u16(abcd_u16), vmovn_u16(efgh_u16));
987
+ }
988
+
989
+ //==============================================================================
990
+ // pcast, SrcType = uint64_t
991
+ //==============================================================================
992
+ template <>
993
+ struct type_casting_traits<numext::uint64_t, float> {
994
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
995
+ };
996
+ template <>
997
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet2ul, Packet4f>(const Packet2ul& a, const Packet2ul& b) {
998
+ return vcvtq_f32_u32(vcombine_u32(vmovn_u64(a), vmovn_u64(b)));
999
+ }
1000
+
1001
+ template <>
1002
+ struct type_casting_traits<numext::uint64_t, numext::uint64_t> {
1003
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
1004
+ };
1005
+ template <>
1006
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet2ul, Packet2ul>(const Packet2ul& a) {
1007
+ return a;
1008
+ }
1009
+
1010
+ template <>
1011
+ struct type_casting_traits<numext::uint64_t, numext::int64_t> {
1012
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
1013
+ };
1014
+ template <>
1015
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet2ul, Packet2l>(const Packet2ul& a) {
1016
+ return vreinterpretq_s64_u64(a);
1017
+ }
1018
+
1019
+ template <>
1020
+ struct type_casting_traits<numext::uint64_t, numext::uint32_t> {
1021
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
1022
+ };
1023
+ template <>
1024
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet2ul, Packet4ui>(const Packet2ul& a, const Packet2ul& b) {
1025
+ return vcombine_u32(vmovn_u64(a), vmovn_u64(b));
1026
+ }
1027
+
1028
+ template <>
1029
+ struct type_casting_traits<numext::uint64_t, numext::int32_t> {
1030
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
1031
+ };
1032
+ template <>
1033
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet2ul, Packet4i>(const Packet2ul& a, const Packet2ul& b) {
1034
+ return vreinterpretq_s32_u32(pcast<Packet2ul, Packet4ui>(a, b));
1035
+ }
1036
+
1037
+ template <>
1038
+ struct type_casting_traits<numext::uint64_t, numext::uint16_t> {
1039
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
1040
+ };
1041
+ template <>
1042
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet2ul, Packet8us>(const Packet2ul& a, const Packet2ul& b, const Packet2ul& c,
1043
+ const Packet2ul& d) {
1044
+ const uint16x4_t ab_u16 = vmovn_u32(vcombine_u32(vmovn_u64(a), vmovn_u64(b)));
1045
+ const uint16x4_t cd_u16 = vmovn_u32(vcombine_u32(vmovn_u64(c), vmovn_u64(d)));
1046
+ return vcombine_u16(ab_u16, cd_u16);
1047
+ }
1048
+
1049
+ template <>
1050
+ struct type_casting_traits<numext::uint64_t, numext::int16_t> {
1051
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
1052
+ };
1053
+ template <>
1054
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet2ul, Packet8s>(const Packet2ul& a, const Packet2ul& b, const Packet2ul& c,
1055
+ const Packet2ul& d) {
1056
+ return vreinterpretq_s16_u16(pcast<Packet2ul, Packet8us>(a, b, c, d));
1057
+ }
1058
+
1059
+ template <>
1060
+ struct type_casting_traits<numext::uint64_t, numext::uint8_t> {
1061
+ enum { VectorizedCast = 1, SrcCoeffRatio = 8, TgtCoeffRatio = 1 };
1062
+ };
1063
+ template <>
1064
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet2ul, Packet16uc>(const Packet2ul& a, const Packet2ul& b, const Packet2ul& c,
1065
+ const Packet2ul& d, const Packet2ul& e, const Packet2ul& f,
1066
+ const Packet2ul& g, const Packet2ul& h) {
1067
+ const uint16x8_t abcd_u16 = pcast<Packet2ul, Packet8us>(a, b, c, d);
1068
+ const uint16x8_t efgh_u16 = pcast<Packet2ul, Packet8us>(e, f, g, h);
1069
+ return vcombine_u8(vmovn_u16(abcd_u16), vmovn_u16(efgh_u16));
1070
+ }
1071
+
1072
+ template <>
1073
+ struct type_casting_traits<numext::uint64_t, numext::int8_t> {
1074
+ enum { VectorizedCast = 1, SrcCoeffRatio = 8, TgtCoeffRatio = 1 };
1075
+ };
1076
+ template <>
1077
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet2ul, Packet16c>(const Packet2ul& a, const Packet2ul& b, const Packet2ul& c,
1078
+ const Packet2ul& d, const Packet2ul& e, const Packet2ul& f,
1079
+ const Packet2ul& g, const Packet2ul& h) {
1080
+ return vreinterpretq_s8_u8(pcast<Packet2ul, Packet16uc>(a, b, c, d, e, f, g, h));
1081
+ }
1082
+
1083
+ //==============================================================================
1084
+ // preinterpret
1085
+ //==============================================================================
1086
+ template <>
1087
+ EIGEN_STRONG_INLINE Packet2f preinterpret<Packet2f, Packet2i>(const Packet2i& a) {
1088
+ return vreinterpret_f32_s32(a);
1089
+ }
1090
+ template <>
1091
+ EIGEN_STRONG_INLINE Packet2f preinterpret<Packet2f, Packet2ui>(const Packet2ui& a) {
1092
+ return vreinterpret_f32_u32(a);
1093
+ }
1094
+ template <>
1095
+ EIGEN_STRONG_INLINE Packet4f preinterpret<Packet4f, Packet4i>(const Packet4i& a) {
1096
+ return vreinterpretq_f32_s32(a);
1097
+ }
1098
+ template <>
1099
+ EIGEN_STRONG_INLINE Packet4f preinterpret<Packet4f, Packet4ui>(const Packet4ui& a) {
1100
+ return vreinterpretq_f32_u32(a);
1101
+ }
1102
+
1103
+ template <>
1104
+ EIGEN_STRONG_INLINE Packet4c preinterpret<Packet4c, Packet4uc>(const Packet4uc& a) {
1105
+ return static_cast<Packet4c>(a);
1106
+ }
1107
+ template <>
1108
+ EIGEN_STRONG_INLINE Packet8c preinterpret<Packet8c, Packet8uc>(const Packet8uc& a) {
1109
+ return vreinterpret_s8_u8(a);
1110
+ }
1111
+ template <>
1112
+ EIGEN_STRONG_INLINE Packet16c preinterpret<Packet16c, Packet16uc>(const Packet16uc& a) {
1113
+ return vreinterpretq_s8_u8(a);
1114
+ }
1115
+
1116
+ template <>
1117
+ EIGEN_STRONG_INLINE Packet4uc preinterpret<Packet4uc, Packet4c>(const Packet4c& a) {
1118
+ return static_cast<Packet4uc>(a);
1119
+ }
1120
+ template <>
1121
+ EIGEN_STRONG_INLINE Packet8uc preinterpret<Packet8uc, Packet8c>(const Packet8c& a) {
1122
+ return vreinterpret_u8_s8(a);
1123
+ }
1124
+ template <>
1125
+ EIGEN_STRONG_INLINE Packet16uc preinterpret<Packet16uc, Packet16c>(const Packet16c& a) {
1126
+ return vreinterpretq_u8_s8(a);
1127
+ }
1128
+
1129
+ template <>
1130
+ EIGEN_STRONG_INLINE Packet4s preinterpret<Packet4s, Packet4us>(const Packet4us& a) {
1131
+ return vreinterpret_s16_u16(a);
1132
+ }
1133
+ template <>
1134
+ EIGEN_STRONG_INLINE Packet8s preinterpret<Packet8s, Packet8us>(const Packet8us& a) {
1135
+ return vreinterpretq_s16_u16(a);
1136
+ }
1137
+
1138
+ template <>
1139
+ EIGEN_STRONG_INLINE Packet4us preinterpret<Packet4us, Packet4s>(const Packet4s& a) {
1140
+ return vreinterpret_u16_s16(a);
1141
+ }
1142
+ template <>
1143
+ EIGEN_STRONG_INLINE Packet8us preinterpret<Packet8us, Packet8s>(const Packet8s& a) {
1144
+ return vreinterpretq_u16_s16(a);
1145
+ }
1146
+
1147
+ template <>
1148
+ EIGEN_STRONG_INLINE Packet2i preinterpret<Packet2i, Packet2f>(const Packet2f& a) {
1149
+ return vreinterpret_s32_f32(a);
1150
+ }
1151
+ template <>
1152
+ EIGEN_STRONG_INLINE Packet2i preinterpret<Packet2i, Packet2ui>(const Packet2ui& a) {
1153
+ return vreinterpret_s32_u32(a);
1154
+ }
1155
+ template <>
1156
+ EIGEN_STRONG_INLINE Packet4i preinterpret<Packet4i, Packet4f>(const Packet4f& a) {
1157
+ return vreinterpretq_s32_f32(a);
1158
+ }
1159
+ template <>
1160
+ EIGEN_STRONG_INLINE Packet4i preinterpret<Packet4i, Packet4ui>(const Packet4ui& a) {
1161
+ return vreinterpretq_s32_u32(a);
1162
+ }
1163
+
1164
+ template <>
1165
+ EIGEN_STRONG_INLINE Packet2ui preinterpret<Packet2ui, Packet2f>(const Packet2f& a) {
1166
+ return vreinterpret_u32_f32(a);
1167
+ }
1168
+ template <>
1169
+ EIGEN_STRONG_INLINE Packet2ui preinterpret<Packet2ui, Packet2i>(const Packet2i& a) {
1170
+ return vreinterpret_u32_s32(a);
1171
+ }
1172
+ template <>
1173
+ EIGEN_STRONG_INLINE Packet4ui preinterpret<Packet4ui, Packet4f>(const Packet4f& a) {
1174
+ return vreinterpretq_u32_f32(a);
1175
+ }
1176
+ template <>
1177
+ EIGEN_STRONG_INLINE Packet4ui preinterpret<Packet4ui, Packet4i>(const Packet4i& a) {
1178
+ return vreinterpretq_u32_s32(a);
1179
+ }
1180
+
1181
+ template <>
1182
+ EIGEN_STRONG_INLINE Packet2l preinterpret<Packet2l, Packet2ul>(const Packet2ul& a) {
1183
+ return vreinterpretq_s64_u64(a);
1184
+ }
1185
+ template <>
1186
+ EIGEN_STRONG_INLINE Packet2ul preinterpret<Packet2ul, Packet2l>(const Packet2l& a) {
1187
+ return vreinterpretq_u64_s64(a);
1188
+ }
1189
+
1190
+ #if EIGEN_ARCH_ARM64
1191
+
1192
+ //==============================================================================
1193
+ // pcast/preinterpret, Double
1194
+ //==============================================================================
1195
+
1196
+ template <>
1197
+ struct type_casting_traits<double, double> {
1198
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
1199
+ };
1200
+ template <>
1201
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet2d, Packet2d>(const Packet2d& a) {
1202
+ return a;
1203
+ }
1204
+
1205
+ template <>
1206
+ struct type_casting_traits<double, float> {
1207
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
1208
+ };
1209
+ template <>
1210
+ EIGEN_STRONG_INLINE Packet4f pcast<Packet2d, Packet4f>(const Packet2d& a, const Packet2d& b) {
1211
+ return vcombine_f32(vcvt_f32_f64(a), vcvt_f32_f64(b));
1212
+ }
1213
+
1214
+ template <>
1215
+ struct type_casting_traits<double, numext::int64_t> {
1216
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
1217
+ };
1218
+ template <>
1219
+ EIGEN_STRONG_INLINE Packet2l pcast<Packet2d, Packet2l>(const Packet2d& a) {
1220
+ return vcvtq_s64_f64(a);
1221
+ }
1222
+
1223
+ template <>
1224
+ struct type_casting_traits<double, numext::uint64_t> {
1225
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
1226
+ };
1227
+ template <>
1228
+ EIGEN_STRONG_INLINE Packet2ul pcast<Packet2d, Packet2ul>(const Packet2d& a) {
1229
+ return vcvtq_u64_f64(a);
1230
+ }
1231
+
1232
+ template <>
1233
+ struct type_casting_traits<double, numext::int32_t> {
1234
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
1235
+ };
1236
+ template <>
1237
+ EIGEN_STRONG_INLINE Packet4i pcast<Packet2d, Packet4i>(const Packet2d& a, const Packet2d& b) {
1238
+ return vcombine_s32(vmovn_s64(vcvtq_s64_f64(a)), vmovn_s64(vcvtq_s64_f64(b)));
1239
+ }
1240
+
1241
+ template <>
1242
+ struct type_casting_traits<double, numext::uint32_t> {
1243
+ enum { VectorizedCast = 1, SrcCoeffRatio = 2, TgtCoeffRatio = 1 };
1244
+ };
1245
+ template <>
1246
+ EIGEN_STRONG_INLINE Packet4ui pcast<Packet2d, Packet4ui>(const Packet2d& a, const Packet2d& b) {
1247
+ return vcombine_u32(vmovn_u64(vcvtq_u64_f64(a)), vmovn_u64(vcvtq_u64_f64(b)));
1248
+ }
1249
+
1250
+ template <>
1251
+ struct type_casting_traits<double, numext::int16_t> {
1252
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
1253
+ };
1254
+ template <>
1255
+ EIGEN_STRONG_INLINE Packet8s pcast<Packet2d, Packet8s>(const Packet2d& a, const Packet2d& b, const Packet2d& c,
1256
+ const Packet2d& d) {
1257
+ const int32x4_t ab_s32 = pcast<Packet2d, Packet4i>(a, b);
1258
+ const int32x4_t cd_s32 = pcast<Packet2d, Packet4i>(c, d);
1259
+ return vcombine_s16(vmovn_s32(ab_s32), vmovn_s32(cd_s32));
1260
+ }
1261
+
1262
+ template <>
1263
+ struct type_casting_traits<double, numext::uint16_t> {
1264
+ enum { VectorizedCast = 1, SrcCoeffRatio = 4, TgtCoeffRatio = 1 };
1265
+ };
1266
+ template <>
1267
+ EIGEN_STRONG_INLINE Packet8us pcast<Packet2d, Packet8us>(const Packet2d& a, const Packet2d& b, const Packet2d& c,
1268
+ const Packet2d& d) {
1269
+ const uint32x4_t ab_u32 = pcast<Packet2d, Packet4ui>(a, b);
1270
+ const uint32x4_t cd_u32 = pcast<Packet2d, Packet4ui>(c, d);
1271
+ return vcombine_u16(vmovn_u32(ab_u32), vmovn_u32(cd_u32));
1272
+ }
1273
+
1274
+ template <>
1275
+ struct type_casting_traits<double, numext::int8_t> {
1276
+ enum { VectorizedCast = 1, SrcCoeffRatio = 8, TgtCoeffRatio = 1 };
1277
+ };
1278
+ template <>
1279
+ EIGEN_STRONG_INLINE Packet16c pcast<Packet2d, Packet16c>(const Packet2d& a, const Packet2d& b, const Packet2d& c,
1280
+ const Packet2d& d, const Packet2d& e, const Packet2d& f,
1281
+ const Packet2d& g, const Packet2d& h) {
1282
+ const int16x8_t abcd_s16 = pcast<Packet2d, Packet8s>(a, b, c, d);
1283
+ const int16x8_t efgh_s16 = pcast<Packet2d, Packet8s>(e, f, g, h);
1284
+ return vcombine_s8(vmovn_s16(abcd_s16), vmovn_s16(efgh_s16));
1285
+ }
1286
+
1287
+ template <>
1288
+ struct type_casting_traits<double, numext::uint8_t> {
1289
+ enum { VectorizedCast = 1, SrcCoeffRatio = 8, TgtCoeffRatio = 1 };
1290
+ };
1291
+ template <>
1292
+ EIGEN_STRONG_INLINE Packet16uc pcast<Packet2d, Packet16uc>(const Packet2d& a, const Packet2d& b, const Packet2d& c,
1293
+ const Packet2d& d, const Packet2d& e, const Packet2d& f,
1294
+ const Packet2d& g, const Packet2d& h) {
1295
+ const uint16x8_t abcd_u16 = pcast<Packet2d, Packet8us>(a, b, c, d);
1296
+ const uint16x8_t efgh_u16 = pcast<Packet2d, Packet8us>(e, f, g, h);
1297
+ return vcombine_u8(vmovn_u16(abcd_u16), vmovn_u16(efgh_u16));
1298
+ }
1299
+
1300
+ template <>
1301
+ struct type_casting_traits<float, double> {
1302
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
1303
+ };
1304
+ template <>
1305
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet4f, Packet2d>(const Packet4f& a) {
1306
+ // Discard second-half of input.
1307
+ return vcvt_f64_f32(vget_low_f32(a));
1308
+ }
1309
+
1310
+ template <>
1311
+ struct type_casting_traits<numext::int8_t, double> {
1312
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 8 };
1313
+ };
1314
+ template <>
1315
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet16c, Packet2d>(const Packet16c& a) {
1316
+ // Discard all but first two values.
1317
+ return vcvt_f64_f32(pcast<Packet8c, Packet2f>(vget_low_s8(a)));
1318
+ }
1319
+
1320
+ template <>
1321
+ struct type_casting_traits<numext::uint8_t, double> {
1322
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 8 };
1323
+ };
1324
+ template <>
1325
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet16uc, Packet2d>(const Packet16uc& a) {
1326
+ // Discard all but first two values.
1327
+ return vcvt_f64_f32(pcast<Packet8uc, Packet2f>(vget_low_u8(a)));
1328
+ }
1329
+
1330
+ template <>
1331
+ struct type_casting_traits<numext::int16_t, double> {
1332
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
1333
+ };
1334
+ template <>
1335
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet8s, Packet2d>(const Packet8s& a) {
1336
+ // Discard all but first two values.
1337
+ return vcvt_f64_f32(pcast<Packet4s, Packet2f>(vget_low_s16(a)));
1338
+ }
1339
+
1340
+ template <>
1341
+ struct type_casting_traits<numext::uint16_t, double> {
1342
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 4 };
1343
+ };
1344
+ template <>
1345
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet8us, Packet2d>(const Packet8us& a) {
1346
+ // Discard all but first two values.
1347
+ return vcvt_f64_f32(pcast<Packet4us, Packet2f>(vget_low_u16(a)));
1348
+ }
1349
+
1350
+ template <>
1351
+ struct type_casting_traits<numext::int32_t, double> {
1352
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
1353
+ };
1354
+ template <>
1355
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet4i, Packet2d>(const Packet4i& a) {
1356
+ // Discard second half of input.
1357
+ return vcvtq_f64_s64(vmovl_s32(vget_low_s32(a)));
1358
+ }
1359
+
1360
+ template <>
1361
+ struct type_casting_traits<numext::uint32_t, double> {
1362
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 2 };
1363
+ };
1364
+ template <>
1365
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet4ui, Packet2d>(const Packet4ui& a) {
1366
+ // Discard second half of input.
1367
+ return vcvtq_f64_u64(vmovl_u32(vget_low_u32(a)));
1368
+ }
1369
+
1370
+ template <>
1371
+ struct type_casting_traits<numext::int64_t, double> {
1372
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
1373
+ };
1374
+ template <>
1375
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet2l, Packet2d>(const Packet2l& a) {
1376
+ return vcvtq_f64_s64(a);
1377
+ }
1378
+
1379
+ template <>
1380
+ struct type_casting_traits<numext::uint64_t, double> {
1381
+ enum { VectorizedCast = 1, SrcCoeffRatio = 1, TgtCoeffRatio = 1 };
1382
+ };
1383
+ template <>
1384
+ EIGEN_STRONG_INLINE Packet2d pcast<Packet2ul, Packet2d>(const Packet2ul& a) {
1385
+ return vcvtq_f64_u64(a);
1386
+ }
1387
+
1388
+ template <>
1389
+ EIGEN_STRONG_INLINE Packet2d preinterpret<Packet2d, Packet2l>(const Packet2l& a) {
1390
+ return vreinterpretq_f64_s64(a);
1391
+ }
1392
+ template <>
1393
+ EIGEN_STRONG_INLINE Packet2d preinterpret<Packet2d, Packet2ul>(const Packet2ul& a) {
1394
+ return vreinterpretq_f64_u64(a);
1395
+ }
1396
+ template <>
1397
+ EIGEN_STRONG_INLINE Packet2l preinterpret<Packet2l, Packet2d>(const Packet2d& a) {
1398
+ return vreinterpretq_s64_f64(a);
1399
+ }
1400
+ template <>
1401
+ EIGEN_STRONG_INLINE Packet2ul preinterpret<Packet2ul, Packet2d>(const Packet2d& a) {
1402
+ return vreinterpretq_u64_f64(a);
1403
+ }
1404
+ template <>
1405
+ EIGEN_STRONG_INLINE Packet2d preinterpret<Packet2d, Packet4i>(const Packet4i& a) {
1406
+ return vreinterpretq_f64_s32(a);
1407
+ }
1408
+ template <>
1409
+ EIGEN_STRONG_INLINE Packet4i preinterpret<Packet4i, Packet2d>(const Packet2d& a) {
1410
+ return vreinterpretq_s32_f64(a);
1411
+ }
1412
+
1413
+ #endif // EIGEN_ARCH_ARM64
1414
+
1415
+ } // end namespace internal
1416
+
1417
+ } // end namespace Eigen
1418
+
1419
+ #endif // EIGEN_TYPE_CASTING_NEON_H