nvmath.bindings.cublasLt.get_matmul_desc_attribute_dtype¶
- nvmath.bindings.cublasLt.get_matmul_desc_attribute_dtype(int attr)[source]¶
Get the Python data type of the corresponding MatmulDescAttribute attribute.
- Parameters:
attr (MatmulDescAttribute) – 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_desc_get_attribute(),matmul_desc_set_attribute().