create ()
|
See cublasLtCreate . |
destroy (intptr_t light_handle)
|
See cublasLtDestroy . |
get_version ()
|
See cublasLtGetVersion . |
get_cudart_version ()
|
See cublasLtGetCudartVersion . |
get_property (int type)
|
See cublasLtGetProperty . |
matmul (intptr_t light_handle, ...)
|
See cublasLtMatmul . |
matrix_transform (intptr_t light_handle, ...)
|
See cublasLtMatrixTransform . |
matrix_layout_create (int type, ...)
|
See cublasLtMatrixLayoutCreate . |
matrix_layout_destroy (intptr_t mat_layout)
|
See cublasLtMatrixLayoutDestroy . |
get_matrix_layout_attribute_dtype (int attr)
|
Get the Python data type of the corresponding MatrixLayoutAttribute attribute. |
matrix_layout_set_attribute (...)
|
See cublasLtMatrixLayoutSetAttribute . |
matrix_layout_get_attribute (...)
|
See cublasLtMatrixLayoutGetAttribute . |
matmul_desc_create (int compute_type, ...)
|
See cublasLtMatmulDescCreate . |
matmul_desc_destroy (intptr_t matmul_desc)
|
See cublasLtMatmulDescDestroy . |
get_matmul_desc_attribute_dtype (int attr)
|
Get the Python data type of the corresponding MatmulDescAttribute attribute. |
matmul_desc_set_attribute (...)
|
See cublasLtMatmulDescSetAttribute . |
matmul_desc_get_attribute (...)
|
See cublasLtMatmulDescGetAttribute . |
matrix_transform_desc_create (int scale_type)
|
See cublasLtMatrixTransformDescCreate . |
matrix_transform_desc_destroy (...)
|
See cublasLtMatrixTransformDescDestroy . |
get_matrix_transform_desc_attribute_dtype (...)
|
Get the Python data type of the corresponding MatrixTransformDescAttribute attribute. |
matrix_transform_desc_set_attribute (...)
|
See cublasLtMatrixTransformDescSetAttribute . |
matrix_transform_desc_get_attribute (...)
|
See cublasLtMatrixTransformDescGetAttribute . |
matmul_preference_create ()
|
See cublasLtMatmulPreferenceCreate . |
matmul_preference_destroy (intptr_t pref)
|
See cublasLtMatmulPreferenceDestroy . |
get_matmul_preference_attribute_dtype (int attr)
|
Get the Python data type of the corresponding MatmulPreferenceAttribute attribute. |
matmul_preference_set_attribute (...)
|
See cublasLtMatmulPreferenceSetAttribute . |
matmul_preference_get_attribute (...)
|
See cublasLtMatmulPreferenceGetAttribute . |
matmul_algo_get_heuristic (...)
|
See cublasLtMatmulAlgoGetHeuristic . |
matmul_algo_init (intptr_t light_handle, ...)
|
See cublasLtMatmulAlgoInit . |
matmul_algo_check (intptr_t light_handle, ...)
|
See cublasLtMatmulAlgoCheck . |
get_matmul_algo_cap_attribute_dtype (int attr)
|
Get the Python data type of the corresponding MatmulAlgoCapAttribute attribute. |
matmul_algo_cap_get_attribute (intptr_t algo, ...)
|
See cublasLtMatmulAlgoCapGetAttribute . |
get_matmul_algo_config_attribute_dtype (int attr)
|
Get the Python data type of the corresponding MatmulAlgoConfigAttribute attribute. |
matmul_algo_config_set_attribute (...)
|
See cublasLtMatmulAlgoConfigSetAttribute . |
matmul_algo_config_get_attribute (...)
|
See cublasLtMatmulAlgoConfigGetAttribute . |
logger_open_file (log_file)
|
See cublasLtLoggerOpenFile . |
logger_set_level (int level)
|
See cublasLtLoggerSetLevel . |
logger_set_mask (int mask)
|
See cublasLtLoggerSetMask . |
logger_force_disable ()
|
See cublasLtLoggerForceDisable . |
get_status_name (int status)
|
See cublasLtGetStatusName . |
get_status_string (int status)
|
See cublasLtGetStatusString . |
heuristics_cache_get_capacity ()
|
See cublasLtHeuristicsCacheGetCapacity . |
heuristics_cache_set_capacity (size_t capacity)
|
See cublasLtHeuristicsCacheSetCapacity . |
disable_cpu_instructions_set_mask (...)
|
See cublasLtDisableCpuInstructionsSetMask . |
matmul_algo_get_ids (intptr_t light_handle, ...)
|
|