Fixes not using the right elements when diag='U'.
Propagates the changes to the doubles kernel.