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.

swap.c 4.3 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123
  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. #ifdef FUNCTION_PROFILE
  41. #include "functable.h"
  42. #endif
  43. #if defined(THUNDERX2T99) || defined(VULCAN)
  44. // Multithreaded swap gives performance benefits in ThunderX2T99
  45. #else
  46. // Disable multi-threading as it does not show any performance
  47. // benefits. Keep the multi-threading code for the record.
  48. #undef SMP
  49. #endif
  50. #ifndef CBLAS
  51. void NAME(blasint *N, FLOAT *x, blasint *INCX, FLOAT *y, blasint *INCY){
  52. blasint n = *N;
  53. blasint incx = *INCX;
  54. blasint incy = *INCY;
  55. #else
  56. void CNAME(blasint n, FLOAT *x, blasint incx, FLOAT *y, blasint incy){
  57. #endif
  58. #ifdef SMP
  59. int mode, nthreads;
  60. FLOAT dummyalpha[2] = {ZERO, ZERO};
  61. #endif
  62. #ifndef CBLAS
  63. PRINT_DEBUG_NAME;
  64. #else
  65. PRINT_DEBUG_CNAME;
  66. #endif
  67. if (n <= 0) return;
  68. IDEBUG_START;
  69. FUNCTION_PROFILE_START();
  70. if (incx < 0) x -= (n - 1) * incx;
  71. if (incy < 0) y -= (n - 1) * incy;
  72. #ifdef SMP
  73. //disable multi-thread when incx==0 or incy==0
  74. //In that case, the threads would be dependent.
  75. if (incx == 0 || incy == 0 || n < 2097152 * GEMM_MULTITHREAD_THRESHOLD / sizeof(FLOAT))
  76. nthreads = 1;
  77. else
  78. nthreads = num_cpu_avail(1);
  79. if (nthreads == 1) {
  80. #endif
  81. SWAP_K(n, 0, 0, ZERO, x, incx, y, incy, NULL, 0);
  82. #ifdef SMP
  83. } else {
  84. #ifdef XDOUBLE
  85. mode = BLAS_XDOUBLE | BLAS_REAL;
  86. #elif defined(DOUBLE)
  87. mode = BLAS_DOUBLE | BLAS_REAL;
  88. #else
  89. mode = BLAS_SINGLE | BLAS_REAL;
  90. #endif
  91. blas_level1_thread(mode, n, 0, 0, dummyalpha,
  92. x, incx, y, incy, NULL, 0, (void *)SWAP_K, nthreads);
  93. }
  94. #endif
  95. FUNCTION_PROFILE_END(1, 2 * n, 0);
  96. IDEBUG_END;
  97. return;
  98. }