Use SymInt signature to compute saved variables (#84354)
This seems to have been accidentally working, but it broke
when I added support for saving optional SymInt directly
from input arguments.
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84354
Approved by: https://github.com/Krovatkin