new parameter for diag matrix computation