Closed denera closed 3 weeks ago
PR #916 prematurely updated PyTorch autocast API to v 2.4.0-RC. This PR reverts it back to the API in v2.3.1 and older.
Description
PR #916 prematurely updated PyTorch autocast API to v 2.4.0-RC. This PR reverts it back to the API in v2.3.1 and older.
Type of change
Changes
Checklist: