You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

syr2k_k.c 3.8 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. /*********************************************************************/
  2. /* Copyright 2009, 2010 The University of Texas at Austin. */
  3. /* All rights reserved. */
  4. /* */
  5. /* Redistribution and use in source and binary forms, with or */
  6. /* without modification, are permitted provided that the following */
  7. /* conditions are met: */
  8. /* */
  9. /* 1. Redistributions of source code must retain the above */
  10. /* copyright notice, this list of conditions and the following */
  11. /* disclaimer. */
  12. /* */
  13. /* 2. Redistributions in binary form must reproduce the above */
  14. /* copyright notice, this list of conditions and the following */
  15. /* disclaimer in the documentation and/or other materials */
  16. /* provided with the distribution. */
  17. /* */
  18. /* THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF TEXAS AT */
  19. /* AUSTIN ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, */
  20. /* INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF */
  21. /* MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE */
  22. /* DISCLAIMED. IN NO EVENT SHALL THE UNIVERSITY OF TEXAS AT */
  23. /* AUSTIN OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, */
  24. /* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES */
  25. /* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE */
  26. /* GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR */
  27. /* BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF */
  28. /* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT */
  29. /* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT */
  30. /* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE */
  31. /* POSSIBILITY OF SUCH DAMAGE. */
  32. /* */
  33. /* The views and conclusions contained in the software and */
  34. /* documentation are those of the authors and should not be */
  35. /* interpreted as representing official policies, either expressed */
  36. /* or implied, of The University of Texas at Austin. */
  37. /*********************************************************************/
  38. #include <stdio.h>
  39. #include "common.h"
  40. #ifndef KERNEL_FUNC
  41. #ifndef LOWER
  42. #define KERNEL_FUNC SYR2K_KERNEL_U
  43. #else
  44. #define KERNEL_FUNC SYR2K_KERNEL_L
  45. #endif
  46. #endif
  47. static __inline int syrk_beta(BLASLONG m_from, BLASLONG m_to, BLASLONG n_from, BLASLONG n_to, FLOAT *alpha, FLOAT *c, BLASLONG ldc) {
  48. BLASLONG i;
  49. #ifndef LOWER
  50. if (m_from > n_from) n_from = m_from;
  51. if (m_to > n_to ) m_to = n_to;
  52. #else
  53. if (m_from < n_from) m_from = n_from;
  54. if (m_to < n_to ) n_to = m_to;
  55. #endif
  56. c += (m_from + n_from * ldc) * COMPSIZE;
  57. m_to -= m_from;
  58. n_to -= n_from;
  59. for (i = 0; i < n_to; i++){
  60. #ifndef LOWER
  61. SCAL_K(MIN(i + n_from - m_from + 1, m_to), 0, 0, alpha[0],
  62. #ifdef COMPLEX
  63. alpha[1],
  64. #endif
  65. c, 1, NULL, 0, NULL, 0);
  66. c += ldc * COMPSIZE;
  67. #else
  68. SCAL_K(MIN(m_to - i + m_from - n_from, m_to), 0, 0, alpha[0],
  69. #ifdef COMPLEX
  70. alpha[1],
  71. #endif
  72. c, 1, NULL, 0, NULL, 0);
  73. if (i < m_from - n_from) {
  74. c += ldc * COMPSIZE;
  75. } else {
  76. c += (1 + ldc) * COMPSIZE;
  77. }
  78. #endif
  79. }
  80. return 0;
  81. }
  82. #ifdef THREADED_LEVEL3
  83. #include "level3_syr2k_threaded.c"
  84. #else
  85. #include "level3_syr2k.c"
  86. #endif