Do more. Code less. Free software for GPU computing.
<scroll to top>
Namespaces | Functions

math.h File Reference

#include <cuComplex.h>
#include "defines.h"

Go to the source code of this file.

Namespaces

namespace  af

Functions

array diff1 (const array &in, int dim=-1)
 difference between elements along a dimension
array diff2 (const array &in, int dim=-1)
 difference between every other element along a dimension
void grad (array &dx, array &dy, const array &in)
 Gradient of a matrix (central differencing).
afError af_besselh_S (float *d_out, unsigned k, unsigned n, const float *d_in, float nu)
afError af_besselh_C (float *d_out, unsigned k, unsigned n, const float *d_in, float nu)
afError af_besselh_D (double *d_out, unsigned k, unsigned n, const double *d_in, double nu)
afError af_besselh_Z (double *d_out, unsigned k, unsigned n, const double *d_in, double nu)
afError af_besseli_S (float *d_out, unsigned n, const float *d_in, float nu)
afError af_besseli_C (float *d_out, unsigned n, const float *d_in, float nu)
afError af_besseli_D (double *d_out, unsigned n, const double *d_in, double nu)
afError af_besseli_Z (double *d_out, unsigned n, const double *d_in, double nu)
afError af_besselj_S (float *d_out, unsigned n, const float *d_in, float nu)
afError af_besselj_C (cuComplex *d_out, unsigned n, const cuComplex *d_in, float nu)
afError af_besselj_D (double *d_out, unsigned n, const double *d_in, double nu)
afError af_besselj_Z (cuDoubleComplex *d_out, unsigned n, const cuDoubleComplex *d_in, double nu)
afError af_besselk_S (float *d_out, unsigned n, const float *d_in, float nu)
afError af_besselk_C (float *d_out, unsigned n, const float *d_in, float nu)
afError af_besselk_D (double *d_out, unsigned n, const double *d_in, double nu)
afError af_besselk_Z (double *d_out, unsigned n, const double *d_in, double nu)
afError af_bessely_S (float *d_out, unsigned n, const float *d_in, float nu)
afError af_bessely_C (float *d_out, unsigned n, const float *d_in, float nu)
afError af_bessely_D (double *d_out, unsigned n, const double *d_in, double nu)
afError af_bessely_Z (double *d_out, unsigned n, const double *d_in, double nu)
Device pointer interface: Difference along a dimension
Parameters:
[in]ndims_outnumber of output dimensions
[in]dims_outoutput dimensions
[out]d_outThe output of the function
[in]ndims_innumber of input dimensions
[in]dims_ininput dimensions
[in]d_inThe input to the function
[in]orderThe order of the difference
[in]dimThe dimension along which the operation is performed
afError af_diff_S (float *d_out, unsigned ndims_in, const unsigned *dims_in, const float *d_in, int order, int dim)
 Single precision, difference function.
afError af_diff_D (double *d_out, unsigned ndims_in, const unsigned *dims_in, const double *d_in, int order, int dim)
 Double precision difference function.
Device pointer interface: Gradient along a dimension
Parameters:
[out]d_dxOutput gradient along the X dimension
[out]d_dyOutput gradient along the Y dimension
[in]nxNumber of rows in the input
[in]nyNumber of columns in the input
[in]d_inThe input the the gradient function
[in]batchThe number of input tiles being handled
afError af_grad_B (bool *d_dx, bool *d_dy, unsigned nx, unsigned ny, const bool *d_in, unsigned batch)
 Boolean gradient function.
afError af_grad_U (unsigned *d_dx, unsigned *d_dy, unsigned nx, unsigned ny, const unsigned *d_in, unsigned batch)
 Unsigne integer gradient function.
afError af_grad_I (int *d_dx, int *d_dy, unsigned nx, unsigned ny, const int *d_in, unsigned batch)
 Integer gradient function.
afError af_grad_S (float *d_dx, float *d_dy, unsigned nx, unsigned ny, const float *d_in, unsigned batch)
 Single precision, floating point gradient funciton.
afError af_grad_D (double *d_dx, double *d_dy, unsigned nx, unsigned ny, const double *d_in, unsigned batch)
 Double precision, floating point gradient funciton.
Device pointer interface : Cross Product
Parameters:
d_COutput Matrix
numelNumber of elements in input matrices A,B
offsetStride required to access next element of vectors
d_ALeft Input Matrix
batch_ANumber of tiles in matrix A
batch_BNumber of tiles in matrix B
afError af_crossProd_SS (float *d_C, unsigned numel, unsigned offset, const float *d_A, unsigned batch_A, const float *d_B, unsigned batch_B)
 Single Precision Cross Product - Real x Real.
afError af_crossProd_SC (cuComplex *d_C, unsigned numel, unsigned offset, const float *d_A, unsigned batch_A, const cuComplex *d_B, unsigned batch_B)
 Single Precision Cross Product - Real x Complex.
afError af_crossProd_CS (cuComplex *d_C, unsigned numel, unsigned offset, const cuComplex *d_A, unsigned batch_A, const float *d_B, unsigned batch_B)
 Single Precision Cross Product - Complex x Real.
afError af_crossProd_CC (cuComplex *d_C, unsigned numel, unsigned offset, const cuComplex *d_A, unsigned batch_A, const cuComplex *d_B, unsigned batch_B)
 Single Precision Cross Product - Complex x Complex.
afError af_crossProd_SD (float *d_C, unsigned numel, unsigned offset, const float *d_A, unsigned batch_A, const double *d_B, unsigned batch_B)
 Cross Product - Single-precision Real x Double-precision Real.
afError af_crossProd_SZ (cuComplex *d_C, unsigned numel, unsigned offset, const float *d_A, unsigned batch_A, const cuDoubleComplex *d_B, unsigned batch_B)
 Cross Product - Single-precision Real x Double-precision Complex.
afError af_crossProd_CD (cuComplex *d_C, unsigned numel, unsigned offset, const cuComplex *d_A, unsigned batch_A, const double *d_B, unsigned batch_B)
 Cross Product - Single-precision Complex x Double-precision Real.
afError af_crossProd_CZ (cuComplex *d_C, unsigned numel, unsigned offset, const cuComplex *d_A, unsigned batch_A, const cuDoubleComplex *d_B, unsigned batch_B)
 Cross Product - Single-precision Complex x Double-precision Complex.
afError af_crossProd_DS (float *d_C, unsigned numel, unsigned offset, const double *d_A, unsigned batch_A, const float *d_B, unsigned batch_B)
 Cross Product - Double-precision Real x Single-precision Real.
afError af_crossProd_DC (cuComplex *d_C, unsigned numel, unsigned offset, const double *d_A, unsigned batch_A, const cuComplex *d_B, unsigned batch_B)
 Cross Product - Double-precision Real x Single-precision Complex.
afError af_crossProd_ZS (cuComplex *d_C, unsigned numel, unsigned offset, const cuDoubleComplex *d_A, unsigned batch_A, const float *d_B, unsigned batch_B)
 Cross Product - Double-precision Complex x Single-precision Real.
afError af_crossProd_ZC (cuComplex *d_C, unsigned numel, unsigned offset, const cuDoubleComplex *d_A, unsigned batch_A, const cuComplex *d_B, unsigned batch_B)
 Cross Product - Double-precision Complex x Single-precision Complex.
afError af_crossProd_DD (double *d_C, unsigned numel, unsigned offset, const double *d_A, unsigned batch_A, const double *d_B, unsigned batch_B)
 Double Precision Cross Product - Real x Real.
afError af_crossProd_DZ (cuDoubleComplex *d_C, unsigned numel, unsigned offset, const double *d_A, unsigned batch_A, const cuDoubleComplex *d_B, unsigned batch_B)
 Double Precision Cross Product - Real x Complex.
afError af_crossProd_ZD (cuDoubleComplex *d_C, unsigned numel, unsigned offset, const cuDoubleComplex *d_A, unsigned batch_A, const double *d_B, unsigned batch_B)
 Double Precision Cross Product - Complex x Real.
afError af_crossProd_ZZ (cuDoubleComplex *d_C, unsigned numel, unsigned offset, const cuDoubleComplex *d_A, unsigned batch_A, const cuDoubleComplex *d_B, unsigned batch_B)
 Double Precision Cross Product - Complex x Complex.
Device pointer interface: Kronecker product
Parameters:
d_COutput Matrix
mANumber of rows of input matrix A
nANumber of columns of input matrix A
d_ALeft Input Matrix
batch_ANumber of tiles in matrix A
mBNumber of rows of input matrix B
nBNumber of columns of input matrix B
d_BRight Input Matrix
batch_BNumber of tiles in matrix B
afError af_kronProd_SS (float *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B)
 Single Precision Kronecker Tensor Product - Real x Real.
afError af_kronProd_SC (cuComplex *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuComplex *d_B, unsigned batch_B)
 Single Precision Kronecker Tensor Product - Real x Complex.
afError af_kronProd_CS (cuComplex *d_C, unsigned mA, unsigned nA, const cuComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B)
 Single Precision Kronecker Tensor Product - Complex x Real.
afError af_kronProd_CC (cuComplex *d_C, unsigned mA, unsigned nA, const cuComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuComplex *d_B, unsigned batch_B)
 Single Precision Kronecker Tensor Product - Complex x Complex.
afError af_kronProd_SD (float *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B)
 Kronecker Tensor Product - Single-precision Real x Double-precision Real.
afError af_kronProd_SZ (cuComplex *d_C, unsigned mA, unsigned nA, const float *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuDoubleComplex *d_B, unsigned batch_B)
 Kronecker Tensor Product - Single-precision Real x Double-precision Complex.
afError af_kronProd_CD (cuComplex *d_C, unsigned mA, unsigned nA, const cuComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B)
 Kronecker Tensor Product - Single-precision Complex x Double-precision Real.
afError af_kronProd_CZ (cuComplex *d_C, unsigned mA, unsigned nA, const cuComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuDoubleComplex *d_B, unsigned batch_B)
 Kronecker Tensor Product - Single-precision Complex x Double-precision Complex.
afError af_kronProd_DS (float *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B)
 Kronecker Tensor Product - Double-precision Real x Single-precision Real.
afError af_kronProd_DC (cuComplex *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuComplex *d_B, unsigned batch_B)
 Kronecker Tensor Product - Double-precision Real x Single-precision Complex.
afError af_kronProd_ZS (cuComplex *d_C, unsigned mA, unsigned nA, const cuDoubleComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const float *d_B, unsigned batch_B)
 Kronecker Tensor Product - Double-precision Complex x Single-precision Real.
afError af_kronProd_ZC (cuComplex *d_C, unsigned mA, unsigned nA, const cuDoubleComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuComplex *d_B, unsigned batch_B)
 Kronecker Tensor Product - Double-precision Complex x Single-precision Complex.
afError af_kronProd_DD (double *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B)
 Double Precision Kronecker Tensor Product - Real x Complex.
afError af_kronProd_DZ (cuDoubleComplex *d_C, unsigned mA, unsigned nA, const double *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuDoubleComplex *d_B, unsigned batch_B)
 Double Precision Kronecker Tensor Product - Real x Complex.
afError af_kronProd_ZD (cuDoubleComplex *d_C, unsigned mA, unsigned nA, const cuDoubleComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const double *d_B, unsigned batch_B)
 Double Precision Kronecker Tensor Product - Complex x Real.
afError af_kronProd_ZZ (cuDoubleComplex *d_C, unsigned mA, unsigned nA, const cuDoubleComplex *d_A, unsigned batch_A, unsigned mB, unsigned nB, const cuDoubleComplex *d_B, unsigned batch_B)
 Double Precision Kronecker Tensor Product - Complex x Complex.

Function Documentation

afError af_besselh_S ( float *  d_out,
unsigned  k,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_besselh_C ( float *  d_out,
unsigned  k,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_besselh_D ( double *  d_out,
unsigned  k,
unsigned  n,
const double *  d_in,
double  nu 
)
afError af_besselh_Z ( double *  d_out,
unsigned  k,
unsigned  n,
const double *  d_in,
double  nu 
)
afError af_besseli_S ( float *  d_out,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_besseli_C ( float *  d_out,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_besseli_D ( double *  d_out,
unsigned  n,
const double *  d_in,
double  nu 
)
afError af_besseli_Z ( double *  d_out,
unsigned  n,
const double *  d_in,
double  nu 
)
afError af_besselj_S ( float *  d_out,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_besselj_C ( cuComplex *  d_out,
unsigned  n,
const cuComplex *  d_in,
float  nu 
)
afError af_besselj_D ( double *  d_out,
unsigned  n,
const double *  d_in,
double  nu 
)
afError af_besselj_Z ( cuDoubleComplex *  d_out,
unsigned  n,
const cuDoubleComplex *  d_in,
double  nu 
)
afError af_besselk_S ( float *  d_out,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_besselk_C ( float *  d_out,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_besselk_D ( double *  d_out,
unsigned  n,
const double *  d_in,
double  nu 
)
afError af_besselk_Z ( double *  d_out,
unsigned  n,
const double *  d_in,
double  nu 
)
afError af_bessely_S ( float *  d_out,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_bessely_C ( float *  d_out,
unsigned  n,
const float *  d_in,
float  nu 
)
afError af_bessely_D ( double *  d_out,
unsigned  n,
const double *  d_in,
double  nu 
)
afError af_bessely_Z ( double *  d_out,
unsigned  n,
const double *  d_in,
double  nu 
)
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Defines