added update_gradints_diag to the add and base kernels
bugfix