updating unit-agrad-fwd/matrix/diag_post_multiply test