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_ltcopy_8.c 5.5 kB

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