Add a rule for use_reentrant with checkpoint

[kit1980]

See https://dev-discuss.pytorch.org/t/bc-breaking-update-to-torch-utils-checkpoint-not-passing-in-use-reentrant-flag-will-raise-an-error/1745 for details.

[malfet]

Looks good to me, though I wonder if there is a way to write abstract CheckDefaultArg class and re-use it for few of the checks

[kit1980]

Looks good to me, though I wonder if there is a way to write abstract CheckDefaultArg class and re-use it for few of the checks

Thanks, I'll try to refactor later.