updated agrad/fwd/round