nvmath.bindings.cusparse.get_sp_mat_attribute_dtype

nvmath.bindings.cusparse.get_sp_mat_attribute_dtype(int attr)[source]

Get the Python data type of the corresponding SpMatAttribute attribute.

Parameters:

attr (SpMatAttribute) – 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 sp_mat_get_attribute(), sp_mat_set_attribute().