Skip to content
New issue

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

Sign up for GitHub

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

Already on GitHub? Sign in to your account

Jump to bottom

[trainer] add tf32-mode control #14606

Merged
merged 11 commits into from
Dec 3, 2021
Merged

[trainer] add tf32-mode control #14606

merged 11 commits into from
Dec 3, 2021

Commits on Dec 3, 2021

  1. [trainer] add --tf32 support

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    c8edc51 View commit details
    Browse the repository at this point in the history

  2. it's pt>=.17

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    20ac698 View commit details
    Browse the repository at this point in the history

  3. it's pt>=.17

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    db1f98f View commit details
    Browse the repository at this point in the history

  4. flip the default to True

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    9e8a7e6 View commit details
    Browse the repository at this point in the history

  5. add experimental note

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    0f630cb View commit details
    Browse the repository at this point in the history

  6. simplify logic

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    52ced77 View commit details
    Browse the repository at this point in the history

  7. style

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    b707c81 View commit details
    Browse the repository at this point in the history

  8. switch to 3-state logic

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    7851ee8 View commit details
    Browse the repository at this point in the history

  9. doc

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    887b116 View commit details
    Browse the repository at this point in the history

  10. Apply suggestions from code review 

    Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
    @stas00 @sgugger
    stas00 and sgugger committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    6f0a359 View commit details
    Browse the repository at this point in the history

  11. re-style code

    @stas00
    stas00 committed Dec 3, 2021
    Configuration menu
    Copy the full SHA
    f85a51e View commit details
    Browse the repository at this point in the history