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_out number of output dimensions [in] dims_out output dimensions [out] d_out The output of the function [in] ndims_in number of input dimensions [in] dims_in input dimensions [in] d_in The input to the function [in] order The order of the difference [in] dim The 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_dx Output gradient along the X dimension [out] d_dy Output gradient along the Y dimension [in] nx Number of rows in the input [in] ny Number of columns in the input [in] d_in The input the the gradient function [in] batch The 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_C Output Matrix numel Number of elements in input matrices A,B offset Stride required to access next element of vectors d_A Left Input Matrix batch_A Number of tiles in matrix A batch_B Number 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_C Output Matrix mA Number of rows of input matrix A nA Number of columns of input matrix A d_A Left Input Matrix batch_A Number of tiles in matrix A mB Number of rows of input matrix B nB Number of columns of input matrix B d_B Right Input Matrix batch_B Number 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
	)