Address eellison's CR comments on AOTAutograd (#85370)
For some reason, this change didn't actually make it to master.
Signed-off-by: Edward Z. Yang <ezyangfb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85370
Approved by: https://github.com/eellison
