You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The warning is seen when I tried to reproduce the Mamba-2 usage example with pytorch 2.4.0+cu121:
mamba_ssm/ops/selective_scan_interface.py:164: FutureWarning: torch.cuda.amp.custom_fwd(args...) is deprecated. Please use torch.amp.custom_fwd(args..., device_type='cuda') instead.
I'm new to github and not sure whether (and how urgently) this needs to be fixed.
The text was updated successfully, but these errors were encountered:
Hi,
The warning is seen when I tried to reproduce the Mamba-2 usage example with pytorch 2.4.0+cu121:
mamba_ssm/ops/selective_scan_interface.py:164: FutureWarning:
torch.cuda.amp.custom_fwd(args...)
is deprecated. Please usetorch.amp.custom_fwd(args..., device_type='cuda')
instead.I'm new to github and not sure whether (and how urgently) this needs to be fixed.
The text was updated successfully, but these errors were encountered: