Package | Description |
---|---|
org.bytedeco.cuda.global |
Modifier and Type | Method and Description |
---|---|
static int |
cublas.cublasLtCreate(cublasLtContext lightHandle) |
static int |
cublas.cublasLtDestroy(cublasLtContext lightHandle) |
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.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.cublasLtMatmulAlgoGetHeuristic(cublasLtContext lightHandle,
cublasLtMatmulDescOpaque_t operationDesc,
cublasLtMatrixLayoutOpaque_t Adesc,
cublasLtMatrixLayoutOpaque_t Bdesc,
cublasLtMatrixLayoutOpaque_t Cdesc,
cublasLtMatrixLayoutOpaque_t Ddesc,
cublasLtMatmulPreferenceOpaque_t preference,
int requestedAlgoCount,
cublasLtMatmulHeuristicResult_t heuristicResultsArray,
int[] returnAlgoCount) |
static int |
cublas.cublasLtMatmulAlgoGetHeuristic(cublasLtContext lightHandle,
cublasLtMatmulDescOpaque_t operationDesc,
cublasLtMatrixLayoutOpaque_t Adesc,
cublasLtMatrixLayoutOpaque_t Bdesc,
cublasLtMatrixLayoutOpaque_t Cdesc,
cublasLtMatrixLayoutOpaque_t Ddesc,
cublasLtMatmulPreferenceOpaque_t preference,
int requestedAlgoCount,
cublasLtMatmulHeuristicResult_t heuristicResultsArray,
IntBuffer returnAlgoCount) |
static int |
cublas.cublasLtMatmulAlgoGetHeuristic(cublasLtContext lightHandle,
cublasLtMatmulDescOpaque_t operationDesc,
cublasLtMatrixLayoutOpaque_t Adesc,
cublasLtMatrixLayoutOpaque_t Bdesc,
cublasLtMatrixLayoutOpaque_t Cdesc,
cublasLtMatrixLayoutOpaque_t Ddesc,
cublasLtMatmulPreferenceOpaque_t preference,
int requestedAlgoCount,
cublasLtMatmulHeuristicResult_t heuristicResultsArray,
IntPointer returnAlgoCount)
Query cublasLt heuristic for algorithm appropriate for given use case.
|
static int |
cublas.cublasLtMatmulAlgoGetIds(cublasLtContext lightHandle,
int computeType,
int scaleType,
int Atype,
int Btype,
int Ctype,
int Dtype,
int requestedAlgoCount,
int[] algoIdsArray,
int[] returnAlgoCount) |
static int |
cublas.cublasLtMatmulAlgoGetIds(cublasLtContext lightHandle,
int computeType,
int scaleType,
int Atype,
int Btype,
int Ctype,
int Dtype,
int requestedAlgoCount,
IntBuffer algoIdsArray,
IntBuffer returnAlgoCount) |
static int |
cublas.cublasLtMatmulAlgoGetIds(cublasLtContext lightHandle,
int computeType,
int scaleType,
int Atype,
int Btype,
int Ctype,
int Dtype,
int requestedAlgoCount,
IntPointer algoIdsArray,
IntPointer returnAlgoCount)
Routine to get all algo IDs that can potentially run
|
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
|
static int |
cublas.cublasLtMatrixTransform(cublasLtContext lightHandle,
cublasLtMatrixTransformDescOpaque_t transformDesc,
Pointer alpha,
Pointer A,
cublasLtMatrixLayoutOpaque_t Adesc,
Pointer beta,
Pointer B,
cublasLtMatrixLayoutOpaque_t Bdesc,
Pointer C,
cublasLtMatrixLayoutOpaque_t Cdesc,
CUstream_st stream)
Matrix layout conversion helper (C = alpha * op(A) + beta * op(B))
Can be used to change memory order of data or to scale and shift the values.
|
Copyright © 2024. All rights reserved.