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