why cuda.amp was removed from version 3.35.0

[thbupt]

System Info

transformers==3.35.0 and transformers==3.34.0

Who can help?

No response

Information

• The official example scripts
• My own modified scripts

Tasks

• An officially supported task in the examples folder (such as GLUE/SQuAD, ...)
• My own task or dataset (give details below)

Reproduction

In transformers==3.35.0:

In transformers==3.34.0

Expected behavior

It seems that you can only use Mixed precision training with cpu_amp or apex, while torch.cuda.amp was removed.

[amyeroberts]

Hi @thbupt, you can find the relevant PR here: #25702

[github-actions]

This issue has been automatically marked as stale because it has not had recent activity. If you think this still needs to be addressed please comment on this thread.

Please note that issues that do not follow the contributing guidelines are likely to be ignored.