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

[GPTNeoX] Flex Attention + Refactor #34896

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Open

[GPTNeoX] Flex Attention + Refactor #34896

wants to merge 8 commits into from
+365 −247

Commits on Nov 23, 2024

  1. gpt neox flex attention + refactor

    @vasqu
    vasqu committed Nov 23, 2024
    Configuration menu
    Copy the full SHA
    65da253 View commit details
    Browse the repository at this point in the history

  2. some formatting

    @vasqu
    vasqu committed Nov 23, 2024
    Configuration menu
    Copy the full SHA
    ebfdc1f View commit details
    Browse the repository at this point in the history

  3. small fix on dropout

    @vasqu
    vasqu committed Nov 23, 2024
    Configuration menu
    Copy the full SHA
    b3c2b11 View commit details
    Browse the repository at this point in the history

  4. add assertion on flex attn test

    @vasqu
    vasqu committed Nov 23, 2024
    Configuration menu
    Copy the full SHA
    06685f8 View commit details
    Browse the repository at this point in the history

  5. flaky ci :(

    @vasqu
    vasqu committed Nov 23, 2024
    Configuration menu
    Copy the full SHA
    53f7319 View commit details
    Browse the repository at this point in the history

  6. add head mask support

    @vasqu
    vasqu committed Nov 23, 2024
    Configuration menu
    Copy the full SHA
    21941c5 View commit details
    Browse the repository at this point in the history

  7. style

    @vasqu
    vasqu committed Nov 23, 2024
    Configuration menu
    Copy the full SHA
    0208905 View commit details
    Browse the repository at this point in the history

  8. handle dtype, replace torch where

    @vasqu
    vasqu committed Nov 23, 2024
    Configuration menu
    Copy the full SHA
    e689d28 View commit details
    Browse the repository at this point in the history