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.

trsm_ltcopy_16.c 5.5 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228
  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. #ifndef UNIT
  41. #define INV(a) (ONE / (a))
  42. #else
  43. #define INV(a) (ONE)
  44. #endif
  45. int CNAME(BLASLONG m, BLASLONG n, FLOAT *a, BLASLONG lda, BLASLONG offset, FLOAT *b){
  46. BLASLONG i, ii, j, jj, k;
  47. FLOAT *a1;
  48. jj = offset;
  49. j = (n >> 4);
  50. while (j > 0){
  51. a1 = a;
  52. a += 16;
  53. ii = 0;
  54. for (i = 0; i < m; i++) {
  55. if ((ii >= jj ) && (ii - jj < 16)) {
  56. *(b + ii - jj) = INV(*(a1 + ii - jj));
  57. for (k = ii - jj + 1; k < 16; k ++) {
  58. *(b + k) = *(a1 + k);
  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 += 16;
  84. j --;
  85. }
  86. j = (n & 8);
  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 < 8)) {
  93. *(b + ii - jj) = INV(*(a1 + ii - jj));
  94. for (k = ii - jj + 1; k < 8; k ++) {
  95. *(b + k) = *(a1 + k);
  96. }
  97. }
  98. if (ii - jj < 0) {
  99. *(b + 0) = *(a1 + 0);
  100. *(b + 1) = *(a1 + 1);
  101. *(b + 2) = *(a1 + 2);
  102. *(b + 3) = *(a1 + 3);
  103. *(b + 4) = *(a1 + 4);
  104. *(b + 5) = *(a1 + 5);
  105. *(b + 6) = *(a1 + 6);
  106. *(b + 7) = *(a1 + 7);
  107. }
  108. b += 8;
  109. a1 += lda;
  110. ii ++;
  111. }
  112. jj += 8;
  113. }
  114. j = (n & 4);
  115. if (j > 0) {
  116. a1 = a;
  117. a += 4;
  118. ii = 0;
  119. for (i = 0; i < m; i++) {
  120. if ((ii >= jj ) && (ii - jj < 4)) {
  121. *(b + ii - jj) = INV(*(a1 + ii - jj));
  122. for (k = ii - jj + 1; k < 4; k ++) {
  123. *(b + k) = *(a1 + k);
  124. }
  125. }
  126. if (ii - jj < 0) {
  127. *(b + 0) = *(a1 + 0);
  128. *(b + 1) = *(a1 + 1);
  129. *(b + 2) = *(a1 + 2);
  130. *(b + 3) = *(a1 + 3);
  131. }
  132. b += 4;
  133. a1 += lda;
  134. ii ++;
  135. }
  136. jj += 4;
  137. }
  138. j = (n & 2);
  139. if (j > 0) {
  140. a1 = a;
  141. a += 2;
  142. ii = 0;
  143. for (i = 0; i < m; i++) {
  144. if ((ii >= jj ) && (ii - jj < 2)) {
  145. *(b + ii - jj) = INV(*(a1 + ii - jj));
  146. for (k = ii - jj + 1; k < 2; k ++) {
  147. *(b + k) = *(a1 + k);
  148. }
  149. }
  150. if (ii - jj < 0) {
  151. *(b + 0) = *(a1 + 0);
  152. *(b + 1) = *(a1 + 1);
  153. }
  154. b += 2;
  155. a1 += lda;
  156. ii ++;
  157. }
  158. jj += 2;
  159. }
  160. j = (n & 1);
  161. if (j > 0) {
  162. a1 = a;
  163. ii = 0;
  164. for (i = 0; i < m; i++) {
  165. if ((ii >= jj ) && (ii - jj < 1)) {
  166. *(b + ii - jj) = INV(*(a1 + ii - jj));
  167. }
  168. if (ii - jj < 0) {
  169. *(b + 0) = *(a1 + 0);
  170. }
  171. b += 1;
  172. a1 += lda;
  173. ii ++;
  174. }
  175. }
  176. return 0;
  177. }