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.

ztrsm_utcopy_8.c 5.6 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209
  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. int CNAME(BLASLONG m, BLASLONG n, FLOAT *a, BLASLONG lda, BLASLONG offset, FLOAT *b){
  41. BLASLONG i, ii, j, jj, k;
  42. FLOAT *a1, data1, data2;
  43. lda *= 2;
  44. jj = offset;
  45. j = (n >> 3);
  46. while (j > 0){
  47. a1 = a;
  48. a += 16;
  49. ii = 0;
  50. for (i = 0; i < m; i++) {
  51. if ((ii >= jj ) && (ii - jj < 8)) {
  52. for (k = 0; k < ii - jj; k ++) {
  53. *(b + k * 2 + 0) = *(a1 + k * 2 + 0);
  54. *(b + k * 2 + 1) = *(a1 + k * 2 + 1);
  55. }
  56. data1 = *(a1 + (ii - jj) * 2 + 0);
  57. data2 = *(a1 + (ii - jj) * 2 + 1);
  58. compinv(b + (ii - jj) * 2, data1, data2);
  59. }
  60. if (ii - jj >= 8) {
  61. *(b + 0) = *(a1 + 0);
  62. *(b + 1) = *(a1 + 1);
  63. *(b + 2) = *(a1 + 2);
  64. *(b + 3) = *(a1 + 3);
  65. *(b + 4) = *(a1 + 4);
  66. *(b + 5) = *(a1 + 5);
  67. *(b + 6) = *(a1 + 6);
  68. *(b + 7) = *(a1 + 7);
  69. *(b + 8) = *(a1 + 8);
  70. *(b + 9) = *(a1 + 9);
  71. *(b + 10) = *(a1 + 10);
  72. *(b + 11) = *(a1 + 11);
  73. *(b + 12) = *(a1 + 12);
  74. *(b + 13) = *(a1 + 13);
  75. *(b + 14) = *(a1 + 14);
  76. *(b + 15) = *(a1 + 15);
  77. }
  78. b += 16;
  79. a1 += lda;
  80. ii ++;
  81. }
  82. jj += 8;
  83. j --;
  84. }
  85. j = (n & 4);
  86. if (j > 0) {
  87. a1 = a;
  88. a += 8;
  89. ii = 0;
  90. for (i = 0; i < m; i++) {
  91. if ((ii >= jj ) && (ii - jj < 4)) {
  92. for (k = 0; k < ii - jj; k ++) {
  93. *(b + k * 2 + 0) = *(a1 + k * 2 + 0);
  94. *(b + k * 2 + 1) = *(a1 + k * 2 + 1);
  95. }
  96. data1 = *(a1 + (ii - jj) * 2 + 0);
  97. data2 = *(a1 + (ii - jj) * 2 + 1);
  98. compinv(b + (ii - jj) * 2, data1, data2);
  99. }
  100. if (ii - jj >= 4) {
  101. *(b + 0) = *(a1 + 0);
  102. *(b + 1) = *(a1 + 1);
  103. *(b + 2) = *(a1 + 2);
  104. *(b + 3) = *(a1 + 3);
  105. *(b + 4) = *(a1 + 4);
  106. *(b + 5) = *(a1 + 5);
  107. *(b + 6) = *(a1 + 6);
  108. *(b + 7) = *(a1 + 7);
  109. }
  110. b += 8;
  111. a1 += lda;
  112. ii ++;
  113. }
  114. jj += 4;
  115. }
  116. j = (n & 2);
  117. if (j > 0) {
  118. a1 = a;
  119. a += 4;
  120. ii = 0;
  121. for (i = 0; i < m; i++) {
  122. if ((ii >= jj ) && (ii - jj < 2)) {
  123. for (k = 0; k < ii - jj; k ++) {
  124. *(b + k * 2 + 0) = *(a1 + k * 2 + 0);
  125. *(b + k * 2 + 1) = *(a1 + k * 2 + 1);
  126. }
  127. data1 = *(a1 + (ii - jj) * 2 + 0);
  128. data2 = *(a1 + (ii - jj) * 2 + 1);
  129. compinv(b + (ii - jj) * 2, data1, data2);
  130. }
  131. if (ii - jj >= 2) {
  132. *(b + 0) = *(a1 + 0);
  133. *(b + 1) = *(a1 + 1);
  134. *(b + 2) = *(a1 + 2);
  135. *(b + 3) = *(a1 + 3);
  136. }
  137. b += 4;
  138. a1 += lda;
  139. ii ++;
  140. }
  141. jj += 2;
  142. }
  143. j = (n & 1);
  144. if (j > 0) {
  145. a1 = a;
  146. ii = 0;
  147. for (i = 0; i < m; i++) {
  148. if ((ii >= jj ) && (ii - jj < 1)) {
  149. for (k = 0; k < ii - jj; k ++) {
  150. *(b + k * 2 + 0) = *(a1 + k * 2 + 0);
  151. *(b + k * 2 + 1) = *(a1 + k * 2 + 1);
  152. }
  153. data1 = *(a1 + (ii - jj) * 2 + 0);
  154. data2 = *(a1 + (ii - jj) * 2 + 1);
  155. compinv(b + (ii - jj) * 2, data1, data2);
  156. }
  157. if (ii - jj >= 1) {
  158. *(b + 0) = *(a1 + 0);
  159. *(b + 1) = *(a1 + 1);
  160. }
  161. b += 2;
  162. a1 += lda;
  163. ii ++;
  164. }
  165. }
  166. return 0;
  167. }