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.cublasLtMatmulPreferenceDestroy(cublasLtMatmulPreferenceOpaque_t pref)
Destroy matmul heuristic search preference descriptor.
|
static int |
cublas.cublasLtMatmulPreferenceGetAttribute(cublasLtMatmulPreferenceOpaque_t pref,
int attr,
Pointer buf,
long sizeInBytes,
SizeTPointer sizeWritten)
Get matmul heuristic search preference descriptor attribute.
|
static int |
cublas.cublasLtMatmulPreferenceInit_internal(cublasLtMatmulPreferenceOpaque_t pref,
long size)
Internal.
|
static int |
cublas.cublasLtMatmulPreferenceInit(cublasLtMatmulPreferenceOpaque_t pref)
Initialize matmul heuristic search preference descriptor in pre-allocated space.
|
static int |
cublas.cublasLtMatmulPreferenceSetAttribute(cublasLtMatmulPreferenceOpaque_t pref,
int attr,
Pointer buf,
long sizeInBytes)
Set matmul heuristic search preference descriptor attribute.
|