blas.h 7.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. /*
  2. * StarPU
  3. * Copyright (C) INRIA 2008-2009 (see AUTHORS file)
  4. *
  5. * This program is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU Lesser General Public License as published by
  7. * the Free Software Foundation; either version 2.1 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.
  13. *
  14. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. */
  16. #ifndef __BLAS_H__
  17. #define __BLAS_H__
  18. #include <starpu.h>
  19. #ifdef ATLAS
  20. #include <cblas.h>
  21. #endif
  22. void SGEMM(char *transa, char *transb, int M, int N, int K, float alpha, float *A, int lda,
  23. float *B, int ldb, float beta, float *C, int ldc);
  24. void DGEMM(char *transa, char *transb, int M, int N, int K, double alpha, double *A, int lda,
  25. double *B, int ldb, double beta, double *C, int ldc);
  26. float SASUM(int N, float *X, int incX);
  27. double DASUM(int N, double *X, int incX);
  28. void SSCAL(int N, float alpha, float *X, int incX);
  29. void DSCAL(int N, double alpha, double *X, int incX);
  30. void STRSM (const char *side, const char *uplo, const char *transa,
  31. const char *diag, const int m, const int n,
  32. const float alpha, const float *A, const int lda,
  33. float *B, const int ldb);
  34. void DGEMM(char *transa, char *transb, int M, int N, int K,
  35. double alpha, double *A, int lda, double *B, int ldb,
  36. double beta, double *C, int ldc);
  37. void SSYR (const char *uplo, const int n, const float alpha,
  38. const float *x, const int incx, float *A, const int lda);
  39. void SSYRK (const char *uplo, const char *trans, const int n,
  40. const int k, const float alpha, const float *A,
  41. const int lda, const float beta, float *C,
  42. const int ldc);
  43. void SGER (const int m, const int n, const float alpha,
  44. const float *x, const int incx, const float *y,
  45. const int incy, float *A, const int lda);
  46. void DGER(const int m, const int n, const double alpha,
  47. const double *x, const int incx, const double *y,
  48. const int incy, double *A, const int lda);
  49. void STRSV (const char *uplo, const char *trans, const char *diag,
  50. const int n, const float *A, const int lda, float *x,
  51. const int incx);
  52. void STRMM(const char *side, const char *uplo, const char *transA,
  53. const char *diag, const int m, const int n,
  54. const float alpha, const float *A, const int lda,
  55. float *B, const int ldb);
  56. void DTRMM(const char *side, const char *uplo, const char *transA,
  57. const char *diag, const int m, const int n,
  58. const double alpha, const double *A, const int lda,
  59. double *B, const int ldb);
  60. void STRMV(const char *uplo, const char *transA, const char *diag,
  61. const int n, const float *A, const int lda, float *X,
  62. const int incX);
  63. void SAXPY(const int n, const float alpha, float *X, const int incX, float *Y, const int incy);
  64. void DAXPY(const int n, const double alpha, double *X, const int incX, double *Y, const int incY);
  65. int ISAMAX (const int n, float *X, const int incX);
  66. int IDAMAX (const int n, double *X, const int incX);
  67. float SDOT(const int n, const float *x, const int incx, const float *y, const int incy);
  68. void SSWAP(const int n, float *x, const int incx, float *y, const int incy);
  69. void DSWAP(const int n, double *x, const int incx, double *y, const int incy);
  70. #if defined(GOTO) || defined(SYSTEM_BLAS)
  71. extern void sgemm_ (const char *transa, const char *transb, const int *m,
  72. const int *n, const int *k, const float *alpha,
  73. const float *A, const int *lda, const float *B,
  74. const int *ldb, const float *beta, float *C,
  75. const int *ldc);
  76. extern void dgemm_ (const char *transa, const char *transb, const int *m,
  77. const int *n, const int *k, const double *alpha,
  78. const double *A, const int *lda, const double *B,
  79. const int *ldb, const double *beta, double *C,
  80. const int *ldc);
  81. extern void ssyr_ (const char *uplo, const int *n, const float *alpha,
  82. const float *x, const int *incx, float *A, const int *lda);
  83. extern void ssyrk_ (const char *uplo, const char *trans, const int *n,
  84. const int *k, const float *alpha, const float *A,
  85. const int *lda, const float *beta, float *C,
  86. const int *ldc);
  87. extern void strsm_ (const char *side, const char *uplo, const char *transa,
  88. const char *diag, const int *m, const int *n,
  89. const float *alpha, const float *A, const int *lda,
  90. float *B, const int *ldb);
  91. extern void dtrsm_ (const char *side, const char *uplo, const char *transa,
  92. const char *diag, const int *m, const int *n,
  93. const double *alpha, const double *A, const int *lda,
  94. double *B, const int *ldb);
  95. extern double sasum_ (const int *n, const float *x, const int *incx);
  96. extern double dasum_ (const int *n, const double *x, const int *incx);
  97. extern void sscal_ (const int *n, const float *alpha, float *x,
  98. const int *incx);
  99. extern void dscal_ (const int *n, const double *alpha, double *x,
  100. const int *incx);
  101. extern void sger_(const int *m, const int *n, const float *alpha,
  102. const float *x, const int *incx, const float *y,
  103. const int *incy, float *A, const int *lda);
  104. extern void dger_(const int *m, const int *n, const double *alpha,
  105. const double *x, const int *incx, const double *y,
  106. const int *incy, double *A, const int *lda);
  107. extern void strsv_ (const char *uplo, const char *trans, const char *diag,
  108. const int *n, const float *A, const int *lda, float *x,
  109. const int *incx);
  110. extern void strmm_(const char *side, const char *uplo, const char *transA,
  111. const char *diag, const int *m, const int *n,
  112. const float *alpha, const float *A, const int *lda,
  113. float *B, const int *ldb);
  114. extern void dtrmm_(const char *side, const char *uplo, const char *transA,
  115. const char *diag, const int *m, const int *n,
  116. const double *alpha, const double *A, const int *lda,
  117. double *B, const int *ldb);
  118. extern void strmv_(const char *uplo, const char *transA, const char *diag,
  119. const int *n, const float *A, const int *lda, float *X,
  120. const int *incX);
  121. extern void saxpy_(const int *n, const float *alpha, float *X, const int *incX,
  122. float *Y, const int *incy);
  123. extern void daxpy_(const int *n, const double *alpha, double *X, const int *incX,
  124. double *Y, const int *incy);
  125. extern int isamax_(const int *n, float *X, const int *incX);
  126. extern int idamax_(const int *n, double *X, const int *incX);
  127. /* for some reason, FLOATRET is not a float but a double in GOTOBLAS */
  128. extern double sdot_(const int *n, const float *x, const int *incx, const float *y, const int *incy);
  129. extern void sswap_(const int *n, float *x, const int *incx, float *y, const int *incy);
  130. extern void dswap_(const int *n, double *x, const int *incx, double *y, const int *incy);
  131. #endif
  132. #endif // __BLAS_H__