shorten the compute gradient of UA2