warp.sparse.bsr\_set\_diag ========================== .. currentmodule:: warp.sparse .. autofunction:: bsr_set_diag