ltfat_blas.c

Matlab时频分析工具箱,希望能对大家有所帮助啊

#include "config.h"
#include "cblas.h"

#ifdef HAVE_BLAS
#include "f77-fcn.h"

#ifdef __cplusplus
extern "C" {
#endif

  F77_RET_T
  F77_FUNC (zgemm, ZGEMM) (F77_CONST_CHAR_ARG_DECL TransA,
			   F77_CONST_CHAR_ARG_DECL TransB,
			   const int *M, const int *N,
			   const int *K, 
			   const double *alpha,
			   const double *a, const int *lda,
			   double *b, const int *ldb, 
			   const double *beta, double *c, const int *ldc
			   F77_CHAR_ARG_LEN_DECL
			   F77_CHAR_ARG_LEN_DECL);
#ifdef __cplusplus
}
#endif

#endif /* end of HAVE_BLAS */


void ltfat_zgemm(const enum CBLAS_TRANSPOSE TransA,
                 const enum CBLAS_TRANSPOSE TransB,
		 const int M, const int N, const int K,
		 const ltfat_complex *alpha,
		 const ltfat_complex *A, const int lda,
		 const ltfat_complex *B, const int ldb,
                 const ltfat_complex *beta,
		 ltfat_complex *C, const int ldc)
#ifdef HAVE_CBLAS
{

  cblas_zgemm(CblasColMajor, TransA, TransB, M, N, K,
	      (double*)alpha, (double*)A, lda, (double*)B, ldb,
	      (double*)beta, (double*)C, ldc);

}
#endif
#ifdef HAVE_BLAS
{
  char ca, cb;

  if (TransA == CblasNoTrans)   ca='N';
  if (TransA == CblasConjTrans) ca='C';

  if (TransB == CblasNoTrans)   cb='N';
  if (TransB == CblasConjTrans) cb='C';

  F77_FUNC (zgemm, ZGEMM) (F77_CONST_CHAR_ARG2 (&ca, 1),
			   F77_CONST_CHAR_ARG2 (&cb, 1),
			   &M, &N, &K,
			   (double*)alpha,
			   (double*)A, &lda,
			   (double*)B, &ldb,
			   (double*)beta, (double*)C, &ldc
			   F77_CHAR_ARG_LEN (1)
			   F77_CHAR_ARG_LEN (1)
			   );

  
}
#endif