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.
|