gsl_cblas__source_trsm_r.h 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276
  1. /* blas/source_trsm_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. INDEX n1, n2;
  22. const int nonunit = (Diag == CblasNonUnit);
  23. int side, uplo, trans;
  24. if (Order == CblasRowMajor) {
  25. n1 = M;
  26. n2 = N;
  27. side = Side;
  28. uplo = Uplo;
  29. trans = (TransA == CblasConjTrans) ? CblasTrans : TransA;
  30. } else {
  31. n1 = N;
  32. n2 = M;
  33. side = (Side == CblasLeft) ? CblasRight : CblasLeft;
  34. uplo = (Uplo == CblasUpper) ? CblasLower : CblasUpper;
  35. trans = (TransA == CblasConjTrans) ? CblasTrans : TransA;
  36. }
  37. if (side == CblasLeft && uplo == CblasUpper && trans == CblasNoTrans) {
  38. /* form B := alpha * inv(TriU(A)) *B */
  39. if (alpha != 1.0) {
  40. for (i = 0; i < n1; i++) {
  41. for (j = 0; j < n2; j++) {
  42. B[ldb * i + j] *= alpha;
  43. }
  44. }
  45. }
  46. for (i = n1; i > 0 && i--;) {
  47. if (nonunit) {
  48. BASE Aii = A[lda * i + i];
  49. for (j = 0; j < n2; j++) {
  50. B[ldb * i + j] /= Aii;
  51. }
  52. }
  53. for (k = 0; k < i; k++) {
  54. const BASE Aki = A[k * lda + i];
  55. for (j = 0; j < n2; j++) {
  56. B[ldb * k + j] -= Aki * B[ldb * i + j];
  57. }
  58. }
  59. }
  60. } else if (side == CblasLeft && uplo == CblasUpper && trans == CblasTrans) {
  61. /* form B := alpha * inv(TriU(A))' *B */
  62. if (alpha != 1.0) {
  63. for (i = 0; i < n1; i++) {
  64. for (j = 0; j < n2; j++) {
  65. B[ldb * i + j] *= alpha;
  66. }
  67. }
  68. }
  69. for (i = 0; i < n1; i++) {
  70. if (nonunit) {
  71. BASE Aii = A[lda * i + i];
  72. for (j = 0; j < n2; j++) {
  73. B[ldb * i + j] /= Aii;
  74. }
  75. }
  76. for (k = i + 1; k < n1; k++) {
  77. const BASE Aik = A[i * lda + k];
  78. for (j = 0; j < n2; j++) {
  79. B[ldb * k + j] -= Aik * B[ldb * i + j];
  80. }
  81. }
  82. }
  83. } else if (side == CblasLeft && uplo == CblasLower && trans == CblasNoTrans) {
  84. /* form B := alpha * inv(TriL(A))*B */
  85. if (alpha != 1.0) {
  86. for (i = 0; i < n1; i++) {
  87. for (j = 0; j < n2; j++) {
  88. B[ldb * i + j] *= alpha;
  89. }
  90. }
  91. }
  92. for (i = 0; i < n1; i++) {
  93. if (nonunit) {
  94. BASE Aii = A[lda * i + i];
  95. for (j = 0; j < n2; j++) {
  96. B[ldb * i + j] /= Aii;
  97. }
  98. }
  99. for (k = i + 1; k < n1; k++) {
  100. const BASE Aki = A[k * lda + i];
  101. for (j = 0; j < n2; j++) {
  102. B[ldb * k + j] -= Aki * B[ldb * i + j];
  103. }
  104. }
  105. }
  106. } else if (side == CblasLeft && uplo == CblasLower && trans == CblasTrans) {
  107. /* form B := alpha * TriL(A)' *B */
  108. if (alpha != 1.0) {
  109. for (i = 0; i < n1; i++) {
  110. for (j = 0; j < n2; j++) {
  111. B[ldb * i + j] *= alpha;
  112. }
  113. }
  114. }
  115. for (i = n1; i > 0 && i--;) {
  116. if (nonunit) {
  117. BASE Aii = A[lda * i + i];
  118. for (j = 0; j < n2; j++) {
  119. B[ldb * i + j] /= Aii;
  120. }
  121. }
  122. for (k = 0; k < i; k++) {
  123. const BASE Aik = A[i * lda + k];
  124. for (j = 0; j < n2; j++) {
  125. B[ldb * k + j] -= Aik * B[ldb * i + j];
  126. }
  127. }
  128. }
  129. } else if (side == CblasRight && uplo == CblasUpper && trans == CblasNoTrans) {
  130. /* form B := alpha * B * inv(TriU(A)) */
  131. if (alpha != 1.0) {
  132. for (i = 0; i < n1; i++) {
  133. for (j = 0; j < n2; j++) {
  134. B[ldb * i + j] *= alpha;
  135. }
  136. }
  137. }
  138. for (i = 0; i < n1; i++) {
  139. for (j = 0; j < n2; j++) {
  140. if (nonunit) {
  141. BASE Ajj = A[lda * j + j];
  142. B[ldb * i + j] /= Ajj;
  143. }
  144. {
  145. BASE Bij = B[ldb * i + j];
  146. for (k = j + 1; k < n2; k++) {
  147. B[ldb * i + k] -= A[j * lda + k] * Bij;
  148. }
  149. }
  150. }
  151. }
  152. } else if (side == CblasRight && uplo == CblasUpper && trans == CblasTrans) {
  153. /* form B := alpha * B * inv(TriU(A))' */
  154. if (alpha != 1.0) {
  155. for (i = 0; i < n1; i++) {
  156. for (j = 0; j < n2; j++) {
  157. B[ldb * i + j] *= alpha;
  158. }
  159. }
  160. }
  161. for (i = 0; i < n1; i++) {
  162. for (j = n2; j > 0 && j--;) {
  163. if (nonunit) {
  164. BASE Ajj = A[lda * j + j];
  165. B[ldb * i + j] /= Ajj;
  166. }
  167. {
  168. BASE Bij = B[ldb * i + j];
  169. for (k = 0; k < j; k++) {
  170. B[ldb * i + k] -= A[k * lda + j] * Bij;
  171. }
  172. }
  173. }
  174. }
  175. } else if (side == CblasRight && uplo == CblasLower && trans == CblasNoTrans) {
  176. /* form B := alpha * B * inv(TriL(A)) */
  177. if (alpha != 1.0) {
  178. for (i = 0; i < n1; i++) {
  179. for (j = 0; j < n2; j++) {
  180. B[ldb * i + j] *= alpha;
  181. }
  182. }
  183. }
  184. for (i = 0; i < n1; i++) {
  185. for (j = n2; j > 0 && j--;) {
  186. if (nonunit) {
  187. BASE Ajj = A[lda * j + j];
  188. B[ldb * i + j] /= Ajj;
  189. }
  190. {
  191. BASE Bij = B[ldb * i + j];
  192. for (k = 0; k < j; k++) {
  193. B[ldb * i + k] -= A[j * lda + k] * Bij;
  194. }
  195. }
  196. }
  197. }
  198. } else if (side == CblasRight && uplo == CblasLower && trans == CblasTrans) {
  199. /* form B := alpha * B * inv(TriL(A))' */
  200. if (alpha != 1.0) {
  201. for (i = 0; i < n1; i++) {
  202. for (j = 0; j < n2; j++) {
  203. B[ldb * i + j] *= alpha;
  204. }
  205. }
  206. }
  207. for (i = 0; i < n1; i++) {
  208. for (j = 0; j < n2; j++) {
  209. if (nonunit) {
  210. BASE Ajj = A[lda * j + j];
  211. B[ldb * i + j] /= Ajj;
  212. }
  213. {
  214. BASE Bij = B[ldb * i + j];
  215. for (k = j + 1; k < n2; k++) {
  216. B[ldb * i + k] -= A[k * lda + j] * Bij;
  217. }
  218. }
  219. }
  220. }
  221. } else {
  222. BLAS_ERROR("unrecognized operation");
  223. }
  224. }