nvmath.bindings.cublasLt.get_matmul_preference_attribute_dtype¶

nvmath.bindings.cublasLt.get_matmul_preference_attribute_dtype(int attr)[source]¶

Get the Python data type of the corresponding MatmulPreferenceAttribute attribute.

Parameters:: attr (MatmulPreferenceAttribute) – 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_preference_get_attribute(), matmul_preference_set_attribute().