blas.h File Reference

(2013-12-19 20:31:57 -0500, (build a6fe8a5))

#include "defines.h"
#include "constants.h"

Go to the source code of this file.

Namespaces

 af
 

Macros

#define ZERO   =0
 

Functions

template<typename ty >
ty norm (const array &in, float p=af::NaN)
 Matrix or vector norm. More...
 
template<typename ty >
ty dot (const array &x, const array &y)
 Scalar dot product between two vectors. More...
 
array matmul (const array &, const array &, af_matprop_t=af_none, af_matprop_t=af_none)
 Multiply two arrays together. More...
 
array matmul (const array &, const array &, const array &)
 Multiply three arrays together. More...
 
array matmul (const array &, const array &, const array &, const array &)
 Multiply four arrays together. More...
 
afError af_gemm_S (char TRANSA, char TRANSB, int M, int N, int K, float ALPHA, const float *d_A, int LDA, const float *d_B, int LDB, float BETA, float *d_C, int LDC, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_gemm_D (char TRANSA, char TRANSB, int M, int N, int K, double ALPHA, const double *d_A, int LDA, const double *d_B, int LDB, double BETA, double *d_C, int LDC, unsigned batch_A=0, unsigned batch_B=0)
 Double precision, Real matrix multiply. More...
 
afError af_gemm_C (char TRANSA, char TRANSB, int M, int N, int K, cfloat ALPHA, const cfloat *d_A, int LDA, const cfloat *d_B, int LDB, cfloat BETA, cfloat *d_C, int LDC, unsigned batch_A=0, unsigned batch_B=0)
 Single precision, Complex matrix multiply. More...
 
afError af_gemm_Z (char TRANSA, char TRANSB, int M, int N, int K, cdouble ALPHA, const cdouble *d_A, int LDA, const cdouble *d_B, int LDB, cdouble BETA, cdouble *d_C, int LDC, unsigned batch_A=0, unsigned batch_B=0)
 Double precision, Complex matrix multiply. More...
 
afError af_matmul_SS (const float *d_A, const float *d_B, float *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 Mixed precision, mixed complexity matrix multiply. More...
 
afError af_matmul_DS (const double *d_A, const float *d_B, float *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_SD (const float *d_A, const double *d_B, float *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_DD (const double *d_A, const double *d_B, double *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_CS (const cfloat *d_A, const float *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_ZS (const cdouble *d_A, const float *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_CD (const cfloat *d_A, const double *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_ZD (const cdouble *d_A, const double *d_B, cdouble *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_SC (const float *d_A, const cfloat *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_DC (const double *d_A, const cfloat *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_SZ (const float *d_A, const cdouble *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_DZ (const double *d_A, const cdouble *d_B, cdouble *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_CC (const cfloat *d_A, const cfloat *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_ZC (const cdouble *d_A, const cfloat *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_CZ (const cfloat *d_A, const cdouble *d_B, cfloat *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_matmul_ZZ (const cdouble *d_A, const cdouble *d_B, cdouble *d_C, int M, int N, int K, unsigned batch_A=0, unsigned batch_B=0)
 
afError af_norm_vector_S (float *h_dst, unsigned numel, const float *d_src)
 
afError af_norm_vector_D (double *h_dst, unsigned numel, const double *d_src)
 norm The value of a double precision vector More...
 
afError af_norm_S (float *d_dst, unsigned ndims, const unsigned *dims, const float *d_src, int dim)
 norm the value along a dimension in single precision matrix More...
 
afError af_norm_D (double *d_dst, unsigned ndims, const unsigned *dims, const double *d_src, int dim)
 norm the value along a dimension in double precision matrix More...
 
afError af_dot_S (float *h_res, unsigned len, const float *d_A, const float *d_B)
 
afError af_dot_D (double *h_res, unsigned len, const double *d_A, const double *d_B)
 
afError af_dot_C (cfloat *h_res, unsigned len, const cfloat *d_A, const cfloat *d_B)
 
afError af_dot_Z (cdouble *h_res, unsigned len, const cdouble *d_A, const cdouble *d_B)
 

Macro Definition Documentation

#define ZERO   =0