nvmath.bindings.cublasLt.get_matmul_algo_config_attribute_dtype¶
- nvmath.bindings.cublasLt.get_matmul_algo_config_attribute_dtype(int attr)[source]¶
Get the Python data type of the corresponding MatmulAlgoConfigAttribute attribute.
- Parameters:
attr (MatmulAlgoConfigAttribute) – The attribute to query.
- Returns:
The data type of the queried attribute.
Note
This API has no C counterpart and is a convenient helper for allocating memory for
matmul_algo_config_get_attribute()
,matmul_algo_config_set_attribute()
.