correct rev auto-diff and add fwd auto-diff