Due to urgent security maintenance, Fraunhofer GitLab CE and GitLab EE will be unavailable on Monday, April 7th, between 5 PM and 6:30 PM CET. Please log out during the maintenance to avoid potential data loss.
* fix amp * fix amp * make logic consistent with torch.autocast * support multiple device * fix as comment * fix as comment * avoid circle import
