public class JCusolverMg extends Object
Modifier and Type | Method and Description |
---|---|
static int |
cusolverMgCreate(cusolverMgHandle handle) |
static int |
cusolverMgCreateDeviceGrid(cudaLibMgGrid grid,
int numRowDevices,
int numColDevices,
int[] deviceId,
int mapping)
Allocates resources related to the shared memory device grid..
|
static int |
cusolverMgCreateMatrixDesc(cudaLibMgMatrixDesc desc,
long numRows,
long numCols,
long rowBlockSize,
long colBlockSize,
int dataType,
cudaLibMgGrid grid)
Allocates resources related to the distributed matrix descriptor..
|
static int |
cusolverMgDestroy(cusolverMgHandle handle) |
static int |
cusolverMgDestroyGrid(cudaLibMgGrid grid)
Releases the allocated resources related to the distributed grid..
|
static int |
cusolverMgDestroyMatrixDesc(cudaLibMgMatrixDesc desc)
Releases the allocated resources related to the distributed matrix descriptor..
|
static int |
cusolverMgDeviceSelect(cusolverMgHandle handle,
int nbDevices,
int[] deviceId) |
static int |
cusolverMgGetrf_bufferSize(cusolverMgHandle handle,
int M,
int N,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
int[][] array_d_IPIV,
int computeType,
int[] lwork) |
static int |
cusolverMgGetrf(cusolverMgHandle handle,
int M,
int N,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
int[][] array_d_IPIV,
int computeType,
Pointer[] array_d_work,
long lwork,
Pointer info) |
static int |
cusolverMgGetrs_bufferSize(cusolverMgHandle handle,
int TRANS,
int N,
int NRHS,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
int[][] array_d_IPIV,
Pointer[] array_d_B,
int IB,
int JB,
cudaLibMgMatrixDesc descrB,
int computeType,
int[] lwork) |
static int |
cusolverMgGetrs(cusolverMgHandle handle,
int TRANS,
int N,
int NRHS,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
int[][] array_d_IPIV,
Pointer[] array_d_B,
int IB,
int JB,
cudaLibMgMatrixDesc descrB,
int computeType,
Pointer[] array_d_work,
long lwork,
Pointer info) |
static int |
cusolverMgPotrf_bufferSize(cusolverMgHandle handle,
int uplo,
int N,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
int computeType,
int[] lwork) |
static int |
cusolverMgPotrf(cusolverMgHandle handle,
int uplo,
int N,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
int computeType,
Pointer[] array_d_work,
long lwork,
Pointer h_info) |
static int |
cusolverMgPotri_bufferSize(cusolverMgHandle handle,
int uplo,
int N,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
int computeType,
int[] lwork) |
static int |
cusolverMgPotri(cusolverMgHandle handle,
int uplo,
int N,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
int computeType,
Pointer[] array_d_work,
long lwork,
Pointer h_info) |
static int |
cusolverMgPotrs_bufferSize(cusolverMgHandle handle,
int uplo,
int n,
int nrhs,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
Pointer[] array_d_B,
int IB,
int JB,
cudaLibMgMatrixDesc descrB,
int computeType,
int[] lwork) |
static int |
cusolverMgPotrs(cusolverMgHandle handle,
int uplo,
int n,
int nrhs,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
Pointer[] array_d_B,
int IB,
int JB,
cudaLibMgMatrixDesc descrB,
int computeType,
Pointer[] array_d_work,
long lwork,
Pointer h_info) |
static int |
cusolverMgSyevd_bufferSize(cusolverMgHandle handle,
int jobz,
int uplo,
int N,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
Pointer W,
int dataTypeW,
int computeType,
int[] lwork) |
static int |
cusolverMgSyevd(cusolverMgHandle handle,
int jobz,
int uplo,
int N,
Pointer[] array_d_A,
int IA,
int JA,
cudaLibMgMatrixDesc descrA,
Pointer W,
int dataTypeW,
int computeType,
Pointer[] array_d_work,
long lwork,
Pointer info) |
public static int cusolverMgCreate(cusolverMgHandle handle)
public static int cusolverMgDestroy(cusolverMgHandle handle)
public static int cusolverMgDeviceSelect(cusolverMgHandle handle, int nbDevices, int[] deviceId)
public static int cusolverMgCreateDeviceGrid(cudaLibMgGrid grid, int numRowDevices, int numColDevices, int[] deviceId, int mapping)
Allocates resources related to the shared memory device grid..
grid
- (out) the opaque data strcuture that holds the gridnumRowDevices
- (in) number of devices in the rownumColDevices
- (in) number of devices in the columndeviceId
- (in) This array of size height * width stores the
device-ids of the 2D grid; each entry must correspond to a valid gpu or to -1 (denoting CPU).mapping
- (in) whether the 2D grid is in row/column majorpublic static int cusolverMgDestroyGrid(cudaLibMgGrid grid)
Releases the allocated resources related to the distributed grid..
grid
- (in) the opaque data strcuture that holds the distributed gridpublic static int cusolverMgCreateMatrixDesc(cudaLibMgMatrixDesc desc, long numRows, long numCols, long rowBlockSize, long colBlockSize, int dataType, cudaLibMgGrid grid)
Allocates resources related to the distributed matrix descriptor..
desc
- (out) the opaque data strcuture that holds the descriptornumRows
- (in) number of total rowsnumCols
- (in) number of total columnsrowBlockSize
- (in) row block sizecolBlockSize
- (in) column block sizedataType
- (in) the data type of each element in cudaDataTypegrid
- (in) the opaque data structure of the distributed gridpublic static int cusolverMgDestroyMatrixDesc(cudaLibMgMatrixDesc desc)
Releases the allocated resources related to the distributed matrix descriptor..
desc
- (in) the opaque data strcuture that holds the descriptorpublic static int cusolverMgSyevd_bufferSize(cusolverMgHandle handle, int jobz, int uplo, int N, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, Pointer W, int dataTypeW, int computeType, int[] lwork)
public static int cusolverMgSyevd(cusolverMgHandle handle, int jobz, int uplo, int N, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, Pointer W, int dataTypeW, int computeType, Pointer[] array_d_work, long lwork, Pointer info)
public static int cusolverMgGetrf_bufferSize(cusolverMgHandle handle, int M, int N, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, int[][] array_d_IPIV, int computeType, int[] lwork)
public static int cusolverMgGetrf(cusolverMgHandle handle, int M, int N, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, int[][] array_d_IPIV, int computeType, Pointer[] array_d_work, long lwork, Pointer info)
public static int cusolverMgGetrs_bufferSize(cusolverMgHandle handle, int TRANS, int N, int NRHS, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, int[][] array_d_IPIV, Pointer[] array_d_B, int IB, int JB, cudaLibMgMatrixDesc descrB, int computeType, int[] lwork)
public static int cusolverMgGetrs(cusolverMgHandle handle, int TRANS, int N, int NRHS, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, int[][] array_d_IPIV, Pointer[] array_d_B, int IB, int JB, cudaLibMgMatrixDesc descrB, int computeType, Pointer[] array_d_work, long lwork, Pointer info)
public static int cusolverMgPotrf_bufferSize(cusolverMgHandle handle, int uplo, int N, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, int computeType, int[] lwork)
public static int cusolverMgPotrf(cusolverMgHandle handle, int uplo, int N, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, int computeType, Pointer[] array_d_work, long lwork, Pointer h_info)
public static int cusolverMgPotrs_bufferSize(cusolverMgHandle handle, int uplo, int n, int nrhs, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, Pointer[] array_d_B, int IB, int JB, cudaLibMgMatrixDesc descrB, int computeType, int[] lwork)
public static int cusolverMgPotrs(cusolverMgHandle handle, int uplo, int n, int nrhs, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, Pointer[] array_d_B, int IB, int JB, cudaLibMgMatrixDesc descrB, int computeType, Pointer[] array_d_work, long lwork, Pointer h_info)
public static int cusolverMgPotri_bufferSize(cusolverMgHandle handle, int uplo, int N, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, int computeType, int[] lwork)
public static int cusolverMgPotri(cusolverMgHandle handle, int uplo, int N, Pointer[] array_d_A, int IA, int JA, cudaLibMgMatrixDesc descrA, int computeType, Pointer[] array_d_work, long lwork, Pointer h_info)
Copyright © 2020. All rights reserved.