Adding missing diagonal-update functionality to DistMatrix specializations which only supported setting/getting diagonals. Which distributions support such functionality is still inconsistent, but hopefully not for long.