Package | Description |
---|---|
org.bytedeco.cuda.cublas | |
org.bytedeco.cuda.global |
Modifier and Type | Method and Description |
---|---|
cublasLtMatmulAlgo_t |
cublasLtMatmulHeuristicResult_t.algo()
Matmul algorithm descriptor.
|
cublasLtMatmulAlgo_t |
cublasLtMatmulAlgo_t.data(int i,
long setter) |
cublasLtMatmulAlgo_t |
cublasLtMatmulAlgo_t.getPointer(long i) |
cublasLtMatmulAlgo_t |
cublasLtMatmulAlgo_t.position(long position) |
Modifier and Type | Method and Description |
---|---|
cublasLtMatmulHeuristicResult_t |
cublasLtMatmulHeuristicResult_t.algo(cublasLtMatmulAlgo_t setter) |
Modifier and Type | Method and Description |
---|---|
static int |
cublas.cublasLtMatmul(cublasLtContext lightHandle,
cublasLtMatmulDescOpaque_t computeDesc,
Pointer alpha,
Pointer A,
cublasLtMatrixLayoutOpaque_t Adesc,
Pointer B,
cublasLtMatrixLayoutOpaque_t Bdesc,
Pointer beta,
Pointer C,
cublasLtMatrixLayoutOpaque_t Cdesc,
Pointer D,
cublasLtMatrixLayoutOpaque_t Ddesc,
cublasLtMatmulAlgo_t algo,
Pointer workspace,
long workspaceSizeInBytes,
CUstream_st stream)
Execute matrix multiplication (D = alpha * op(A) * op(B) + beta * C).
|
static int |
cublas.cublasLtMatmulAlgoCapGetAttribute(cublasLtMatmulAlgo_t algo,
int attr,
Pointer buf,
long sizeInBytes,
SizeTPointer sizeWritten)
Get algo capability attribute.
|
static int |
cublas.cublasLtMatmulAlgoCheck(cublasLtContext lightHandle,
cublasLtMatmulDescOpaque_t operationDesc,
cublasLtMatrixLayoutOpaque_t Adesc,
cublasLtMatrixLayoutOpaque_t Bdesc,
cublasLtMatrixLayoutOpaque_t Cdesc,
cublasLtMatrixLayoutOpaque_t Ddesc,
cublasLtMatmulAlgo_t algo,
cublasLtMatmulHeuristicResult_t result)
Check configured algo descriptor for correctness and support on current device.
|
static int |
cublas.cublasLtMatmulAlgoConfigGetAttribute(cublasLtMatmulAlgo_t algo,
int attr,
Pointer buf,
long sizeInBytes,
SizeTPointer sizeWritten)
Get algo configuration attribute.
|
static int |
cublas.cublasLtMatmulAlgoConfigSetAttribute(cublasLtMatmulAlgo_t algo,
int attr,
Pointer buf,
long sizeInBytes)
Set algo configuration attribute.
|
static int |
cublas.cublasLtMatmulAlgoInit(cublasLtContext lightHandle,
int computeType,
int scaleType,
int Atype,
int Btype,
int Ctype,
int Dtype,
int algoId,
cublasLtMatmulAlgo_t algo)
Initialize algo structure
\retval CUBLAS_STATUS_INVALID_VALUE if algo is NULL or algoId is outside of recognized range
\retval CUBLAS_STATUS_NOT_SUPPORTED if algoId is not supported for given combination of data types
\retval CUBLAS_STATUS_SUCCESS if the structure was successfully initialized
|
Copyright © 2024. All rights reserved.