svd3x3_avx.h 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  1. // This file is part of libigl, a simple c++ geometry processing library.
  2. //
  3. // Copyright (C) 2013 Alec Jacobson <alecjacobson@gmail.com>
  4. //
  5. // This Source Code Form is subject to the terms of the Mozilla Public License
  6. // v. 2.0. If a copy of the MPL was not distributed with this file, You can
  7. // obtain one at http://mozilla.org/MPL/2.0/.
  8. #ifndef IGL_SVD3X3_AVX_H
  9. #define IGL_SVD3X3_AVX_H
  10. #include "igl_inline.h"
  11. #include <Eigen/Dense>
  12. namespace igl
  13. {
  14. // Super fast 3x3 SVD according to
  15. // http://pages.cs.wisc.edu/~sifakis/project_pages/svd.html This is AVX
  16. // version of svd3x3 (see svd3x3.h) which works on 8 matrices at a time These
  17. // eight matrices are simply stacked in columns, the rest is the same as for
  18. // svd3x3
  19. //
  20. // Inputs:
  21. // A 12 by 3 stack of 3x3 matrices
  22. // Outputs:
  23. // U 12x3 left singular vectors stacked
  24. // S 12x1 singular values stacked
  25. // V 12x3 right singular vectors stacked
  26. //
  27. // Known bugs: this will not work correctly for double precision.
  28. template<typename T>
  29. IGL_INLINE void svd3x3_avx(
  30. const Eigen::Matrix<T, 3*8, 3>& A,
  31. Eigen::Matrix<T, 3*8, 3> &U,
  32. Eigen::Matrix<T, 3*8, 1> &S,
  33. Eigen::Matrix<T, 3*8, 3>&V);
  34. }
  35. #ifndef IGL_STATIC_LIBRARY
  36. # include "svd3x3_avx.cpp"
  37. #endif
  38. #endif