torch.amp.custom_fwd Error

[praina-dandy]

PR #163 had a mistake which breaks LightGlue on recent PyTorch versions.
hasattr(torch, "amp") cannot be used as the criterion to choose between torch.amp.custom_fwd and torch.cuda.amp.custom_fwd, because torch.amp was never fully removed. Only torch.amp.custom_fwd was removed.

[Phil26AT]

Thank you for reporting this error. The problem was indeed a missing check whether custom_fwd is in torch.amp. Can you confirm its working now?

[praina-dandy]

Yes, the change on the PR works. Thanks!