NnmfPack
2.1
|
Header file for using the betadivergence cuda functions with GPUs. More...
Go to the source code of this file.
Functions | |
int | dbdiv_cuda (const int m, const int n, const int k, const double *A, double *W, double *H, const double beta, const int uType, const int nIter) |
int | sbdiv_cuda (const int m, const int n, const int k, const float *A, float *W, float *H, const float beta, const int uType, const int nIter) |
int | dbdivg_cuda (const int m, const int n, const int k, const double *A, double *W, double *H, const double beta, const int uType, const int nIter) |
int | sbdivg_cuda (const int m, const int n, const int k, const float *A, float *W, float *H, const float beta, const int uType, const int nIter) |
int | dbdivone_cuda (const int m, const int n, const int k, const double *A, double *W, double *H, const int uType, const int nIter) |
int | sbdivone_cuda (const int m, const int n, const int k, const float *A, float *W, float *H, const int uType, const int nIter) |
void | dkernelH_cuda (const int m, const int n, const double *L, const double *A, double *R, const double expo, cudaStream_t stream) |
void | skernelH_cuda (const int m, const int n, const float *L, const float *A, float *R, const float expo, cudaStream_t stream) |
void | dkernelW_cuda (const int m, const int n, const double *L, const double *A, double *R, const double expo, cudaStream_t stream) |
void | skernelW_cuda (const int m, const int n, const float *L, const float *A, float *R, const float expo, cudaStream_t stream) |
void | dupdate1H_cuda (const int n, const double *X, double *H, cudaStream_t stream) |
void | supdate1H_cuda (const int n, const float *X, float *H, cudaStream_t stream) |
void | dupdate2H_cuda (const int m, const int n, const double *X, const double *B, double *H, cudaStream_t stream) |
void | supdate2H_cuda (const int m, const int n, const float *X, const float *B, float *H, cudaStream_t stream) |
void | dupdate1W_cuda (const int m, const int n, const double *X, double *W, cudaStream_t stream) |
void | supdate1W_cuda (const int m, const int n, const float *X, float *W, cudaStream_t stream) |
void | dupdate2W_cuda (const int m, const int n, const double *X, const double *B, double *W, cudaStream_t stream) |
void | supdate2W_cuda (const int m, const int n, const float *X, const float *B, float *W, cudaStream_t stream) |
__global__ void | vdkernelH_cuda (const int m, const int n, const double *__restrict__ L, const double *__restrict__ A, double *R, const double expo) |
__global__ void | vskernelH_cuda (const int m, const int n, const float *__restrict__ L, const float *__restrict__ A, float *R, const float expo) |
__global__ void | vdkernelW_cuda (const int m, const int n, const double *__restrict__ L, const double *__restrict__ A, double *R, const double expo) |
__global__ void | vskernelW_cuda (const int m, const int n, const float *__restrict__ L, const float *__restrict__ A, float *R, const float expo) |
__global__ void | vdupdate1H_cuda (const int n, const double *__restrict__ X, double *H) |
__global__ void | vsupdate1H_cuda (const int n, const float *__restrict__ X, float *H) |
__global__ void | vdupdate1W_cuda (const int m, const int n, const double *__restrict__ X, double *W) |
__global__ void | vsupdate1W_cuda (const int m, const int n, const float *__restrict__ X, float *W) |
__global__ void | vdupdate2H_cuda (const int m, const int n, const double *__restrict__ X, const double *__restrict__ B, double *H) |
__global__ void | vsupdate2H_cuda (const int m, const int n, const float *__restrict__ X, const float *__restrict__ B, float *H) |
__global__ void | vdupdate2W_cuda (const int m, const int k, const double *__restrict__ X, const double *__restrict__ B, double *W) |
__global__ void | vsupdate2W_cuda (const int m, const int k, const float *__restrict__ X, const float *__restrict__ B, float *W) |
Header file for using the betadivergence cuda functions with GPUs.
Definition in file bdiv_cuda.h.
int dbdiv_cuda | ( | const int | m, |
const int | n, | ||
const int | k, | ||
const double * | A, | ||
double * | W, | ||
double * | H, | ||
const double | beta, | ||
const int | uType, | ||
const int | nIter | ||
) |
int dbdivg_cuda | ( | const int | m, |
const int | n, | ||
const int | k, | ||
const double * | A, | ||
double * | W, | ||
double * | H, | ||
const double | beta, | ||
const int | uType, | ||
const int | nIter | ||
) |
int dbdivone_cuda | ( | const int | m, |
const int | n, | ||
const int | k, | ||
const double * | A, | ||
double * | W, | ||
double * | H, | ||
const int | uType, | ||
const int | nIter | ||
) |
void dkernelH_cuda | ( | const int | m, |
const int | n, | ||
const double * | L, | ||
const double * | A, | ||
double * | R, | ||
const double | expo, | ||
cudaStream_t | stream | ||
) |
void dkernelW_cuda | ( | const int | m, |
const int | n, | ||
const double * | L, | ||
const double * | A, | ||
double * | R, | ||
const double | expo, | ||
cudaStream_t | stream | ||
) |
void dupdate1H_cuda | ( | const int | n, |
const double * | X, | ||
double * | H, | ||
cudaStream_t | stream | ||
) |
void dupdate1W_cuda | ( | const int | m, |
const int | n, | ||
const double * | X, | ||
double * | W, | ||
cudaStream_t | stream | ||
) |
void dupdate2H_cuda | ( | const int | m, |
const int | n, | ||
const double * | X, | ||
const double * | B, | ||
double * | H, | ||
cudaStream_t | stream | ||
) |
void dupdate2W_cuda | ( | const int | m, |
const int | n, | ||
const double * | X, | ||
const double * | B, | ||
double * | W, | ||
cudaStream_t | stream | ||
) |
int sbdiv_cuda | ( | const int | m, |
const int | n, | ||
const int | k, | ||
const float * | A, | ||
float * | W, | ||
float * | H, | ||
const float | beta, | ||
const int | uType, | ||
const int | nIter | ||
) |
int sbdivg_cuda | ( | const int | m, |
const int | n, | ||
const int | k, | ||
const float * | A, | ||
float * | W, | ||
float * | H, | ||
const float | beta, | ||
const int | uType, | ||
const int | nIter | ||
) |
int sbdivone_cuda | ( | const int | m, |
const int | n, | ||
const int | k, | ||
const float * | A, | ||
float * | W, | ||
float * | H, | ||
const int | uType, | ||
const int | nIter | ||
) |
void skernelH_cuda | ( | const int | m, |
const int | n, | ||
const float * | L, | ||
const float * | A, | ||
float * | R, | ||
const float | expo, | ||
cudaStream_t | stream | ||
) |
void skernelW_cuda | ( | const int | m, |
const int | n, | ||
const float * | L, | ||
const float * | A, | ||
float * | R, | ||
const float | expo, | ||
cudaStream_t | stream | ||
) |
void supdate1H_cuda | ( | const int | n, |
const float * | X, | ||
float * | H, | ||
cudaStream_t | stream | ||
) |
void supdate1W_cuda | ( | const int | m, |
const int | n, | ||
const float * | X, | ||
float * | W, | ||
cudaStream_t | stream | ||
) |
void supdate2H_cuda | ( | const int | m, |
const int | n, | ||
const float * | X, | ||
const float * | B, | ||
float * | H, | ||
cudaStream_t | stream | ||
) |
void supdate2W_cuda | ( | const int | m, |
const int | n, | ||
const float * | X, | ||
const float * | B, | ||
float * | W, | ||
cudaStream_t | stream | ||
) |
__global__ void vdkernelH_cuda | ( | const int | m, |
const int | n, | ||
const double *__restrict__ | L, | ||
const double *__restrict__ | A, | ||
double * | R, | ||
const double | expo | ||
) |
__global__ void vdkernelW_cuda | ( | const int | m, |
const int | n, | ||
const double *__restrict__ | L, | ||
const double *__restrict__ | A, | ||
double * | R, | ||
const double | expo | ||
) |
__global__ void vdupdate1H_cuda | ( | const int | n, |
const double *__restrict__ | X, | ||
double * | H | ||
) |
__global__ void vdupdate1W_cuda | ( | const int | m, |
const int | n, | ||
const double *__restrict__ | X, | ||
double * | W | ||
) |
__global__ void vdupdate2H_cuda | ( | const int | m, |
const int | n, | ||
const double *__restrict__ | X, | ||
const double *__restrict__ | B, | ||
double * | H | ||
) |
__global__ void vdupdate2W_cuda | ( | const int | m, |
const int | k, | ||
const double *__restrict__ | X, | ||
const double *__restrict__ | B, | ||
double * | W | ||
) |
__global__ void vskernelH_cuda | ( | const int | m, |
const int | n, | ||
const float *__restrict__ | L, | ||
const float *__restrict__ | A, | ||
float * | R, | ||
const float | expo | ||
) |
__global__ void vskernelW_cuda | ( | const int | m, |
const int | n, | ||
const float *__restrict__ | L, | ||
const float *__restrict__ | A, | ||
float * | R, | ||
const float | expo | ||
) |
__global__ void vsupdate1H_cuda | ( | const int | n, |
const float *__restrict__ | X, | ||
float * | H | ||
) |
__global__ void vsupdate1W_cuda | ( | const int | m, |
const int | n, | ||
const float *__restrict__ | X, | ||
float * | W | ||
) |
__global__ void vsupdate2H_cuda | ( | const int | m, |
const int | n, | ||
const float *__restrict__ | X, | ||
const float *__restrict__ | B, | ||
float * | H | ||
) |
__global__ void vsupdate2W_cuda | ( | const int | m, |
const int | k, | ||
const float *__restrict__ | X, | ||
const float *__restrict__ | B, | ||
float * | W | ||
) |