rewrite grad_vol computation without for loop