get_matmul_alg_attribute_dtype

nvmath.bindings.cusparseLt.get_matmul_alg_attribute_dtype(int attr)

Get the Python data type of the corresponding MatmulAlgAttribute attribute.

Parameters:

attr (MatmulAlgAttribute) – 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_alg_get_attribute(), matmul_alg_set_attribute().