New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

#

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Jump to bottom

Fix deprecation warnings #1382

Open

rongou wants to merge 1 commit into Dao-AILab:main from rongou:fix-deprecation-warnings

rongou commented Dec 12, 2024

A couple of torch calls are deprecated:

fused_dense.py:30: FutureWarning: `torch.cuda.amp.custom_fwd(args...)` is deprecated. Please use `torch.amp.custom_fwd(args..., device_type='cuda')` instead.
fused_dense.py:71: FutureWarning: `torch.cuda.amp.custom_bwd(args...)` is deprecated. Please use `torch.amp.custom_bwd(args..., device_type='cuda')` instead.


          Fix deprecation warnings

8b47cca

# for free to join this conversation on GitHub. Already have an account? # to comment

Labels

None yet