gsl_cblas__source_syrk_c.h 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165
  1. /* blas/source_syrk_r.h
  2. *
  3. * Copyright (C) 2001, 2007 Brian Gough
  4. *
  5. * This program is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU General Public License as published by
  7. * the Free Software Foundation; either version 3 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * This program is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  13. * General Public License for more details.
  14. *
  15. * You should have received a copy of the GNU General Public License
  16. * along with this program; if not, write to the Free Software
  17. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18. */
  19. {
  20. INDEX i, j, k;
  21. int uplo, trans;
  22. const BASE alpha_real = CONST_REAL0(alpha);
  23. const BASE alpha_imag = CONST_IMAG0(alpha);
  24. const BASE beta_real = CONST_REAL0(beta);
  25. const BASE beta_imag = CONST_IMAG0(beta);
  26. if ((alpha_real == 0.0 && alpha_imag == 0.0)
  27. && (beta_real == 1.0 && beta_imag == 0.0))
  28. return;
  29. if (Order == CblasRowMajor) {
  30. uplo = Uplo;
  31. /* FIXME: original blas does not make distinction between Trans and ConjTrans?? */
  32. trans = (Trans == CblasNoTrans) ? CblasNoTrans : CblasTrans;
  33. } else {
  34. uplo = (Uplo == CblasUpper) ? CblasLower : CblasUpper;
  35. trans = (Trans == CblasNoTrans) ? CblasTrans : CblasNoTrans;
  36. }
  37. /* form y := beta*y */
  38. if (beta_real == 0.0 && beta_imag == 0.0) {
  39. if (uplo == CblasUpper) {
  40. for (i = 0; i < N; i++) {
  41. for (j = i; j < N; j++) {
  42. REAL(C, ldc * i + j) = 0.0;
  43. IMAG(C, ldc * i + j) = 0.0;
  44. }
  45. }
  46. } else {
  47. for (i = 0; i < N; i++) {
  48. for (j = 0; j <= i; j++) {
  49. REAL(C, ldc * i + j) = 0.0;
  50. IMAG(C, ldc * i + j) = 0.0;
  51. }
  52. }
  53. }
  54. } else if (!(beta_real == 1.0 && beta_imag == 0.0)) {
  55. if (uplo == CblasUpper) {
  56. for (i = 0; i < N; i++) {
  57. for (j = i; j < N; j++) {
  58. const BASE Cij_real = REAL(C, ldc * i + j);
  59. const BASE Cij_imag = IMAG(C, ldc * i + j);
  60. REAL(C, ldc * i + j) = beta_real * Cij_real - beta_imag * Cij_imag;
  61. IMAG(C, ldc * i + j) = beta_real * Cij_imag + beta_imag * Cij_real;
  62. }
  63. }
  64. } else {
  65. for (i = 0; i < N; i++) {
  66. for (j = 0; j <= i; j++) {
  67. const BASE Cij_real = REAL(C, ldc * i + j);
  68. const BASE Cij_imag = IMAG(C, ldc * i + j);
  69. REAL(C, ldc * i + j) = beta_real * Cij_real - beta_imag * Cij_imag;
  70. IMAG(C, ldc * i + j) = beta_real * Cij_imag + beta_imag * Cij_real;
  71. }
  72. }
  73. }
  74. }
  75. if (alpha_real == 0.0 && alpha_imag == 0.0)
  76. return;
  77. if (uplo == CblasUpper && trans == CblasNoTrans) {
  78. for (i = 0; i < N; i++) {
  79. for (j = i; j < N; j++) {
  80. BASE temp_real = 0.0;
  81. BASE temp_imag = 0.0;
  82. for (k = 0; k < K; k++) {
  83. const BASE Aik_real = CONST_REAL(A, i * lda + k);
  84. const BASE Aik_imag = CONST_IMAG(A, i * lda + k);
  85. const BASE Ajk_real = CONST_REAL(A, j * lda + k);
  86. const BASE Ajk_imag = CONST_IMAG(A, j * lda + k);
  87. temp_real += Aik_real * Ajk_real - Aik_imag * Ajk_imag;
  88. temp_imag += Aik_real * Ajk_imag + Aik_imag * Ajk_real;
  89. }
  90. REAL(C, i * ldc + j) += alpha_real * temp_real - alpha_imag * temp_imag;
  91. IMAG(C, i * ldc + j) += alpha_real * temp_imag + alpha_imag * temp_real;
  92. }
  93. }
  94. } else if (uplo == CblasUpper && trans == CblasTrans) {
  95. for (i = 0; i < N; i++) {
  96. for (j = i; j < N; j++) {
  97. BASE temp_real = 0.0;
  98. BASE temp_imag = 0.0;
  99. for (k = 0; k < K; k++) {
  100. const BASE Aki_real = CONST_REAL(A, k * lda + i);
  101. const BASE Aki_imag = CONST_IMAG(A, k * lda + i);
  102. const BASE Akj_real = CONST_REAL(A, k * lda + j);
  103. const BASE Akj_imag = CONST_IMAG(A, k * lda + j);
  104. temp_real += Aki_real * Akj_real - Aki_imag * Akj_imag;
  105. temp_imag += Aki_real * Akj_imag + Aki_imag * Akj_real;
  106. }
  107. REAL(C, i * ldc + j) += alpha_real * temp_real - alpha_imag * temp_imag;
  108. IMAG(C, i * ldc + j) += alpha_real * temp_imag + alpha_imag * temp_real;
  109. }
  110. }
  111. } else if (uplo == CblasLower && trans == CblasNoTrans) {
  112. for (i = 0; i < N; i++) {
  113. for (j = 0; j <= i; j++) {
  114. BASE temp_real = 0.0;
  115. BASE temp_imag = 0.0;
  116. for (k = 0; k < K; k++) {
  117. const BASE Aik_real = CONST_REAL(A, i * lda + k);
  118. const BASE Aik_imag = CONST_IMAG(A, i * lda + k);
  119. const BASE Ajk_real = CONST_REAL(A, j * lda + k);
  120. const BASE Ajk_imag = CONST_IMAG(A, j * lda + k);
  121. temp_real += Aik_real * Ajk_real - Aik_imag * Ajk_imag;
  122. temp_imag += Aik_real * Ajk_imag + Aik_imag * Ajk_real;
  123. }
  124. REAL(C, i * ldc + j) += alpha_real * temp_real - alpha_imag * temp_imag;
  125. IMAG(C, i * ldc + j) += alpha_real * temp_imag + alpha_imag * temp_real;
  126. }
  127. }
  128. } else if (uplo == CblasLower && trans == CblasTrans) {
  129. for (i = 0; i < N; i++) {
  130. for (j = 0; j <= i; j++) {
  131. BASE temp_real = 0.0;
  132. BASE temp_imag = 0.0;
  133. for (k = 0; k < K; k++) {
  134. const BASE Aki_real = CONST_REAL(A, k * lda + i);
  135. const BASE Aki_imag = CONST_IMAG(A, k * lda + i);
  136. const BASE Akj_real = CONST_REAL(A, k * lda + j);
  137. const BASE Akj_imag = CONST_IMAG(A, k * lda + j);
  138. temp_real += Aki_real * Akj_real - Aki_imag * Akj_imag;
  139. temp_imag += Aki_real * Akj_imag + Aki_imag * Akj_real;
  140. }
  141. REAL(C, i * ldc + j) += alpha_real * temp_real - alpha_imag * temp_imag;
  142. IMAG(C, i * ldc + j) += alpha_real * temp_imag + alpha_imag * temp_real;
  143. }
  144. }
  145. } else {
  146. BLAS_ERROR("unrecognized operation");
  147. }
  148. }