Update AMP imports for PyTorch version compatibility#5454

Open

stefanklut wants to merge 1 commit intofacebookresearch:mainfrom

stefanklut:main

Commits on Apr 1, 2025

Replace torch.cuda.amp import with torch.amp import and device_type="cuda" partial if pytorch version > 2.4.0
stefanklut
committed