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.

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

[backend] 3/3 Triton 2 update #272

Merged
merged 5 commits into from
Apr 21, 2022
Merged

[backend] 3/3 Triton 2 update #272

merged 5 commits into from
Apr 21, 2022

Commits on Apr 21, 2022

  1. parent be72b26

    author Kashif Rasul <kashif.rasul@gmail.com> 1648069860 +0100
    committer Benjamin Lefaudeux <benjamin.lefaudeux@pm.me> 1650256563 -0700
    
    Move to Triton 2
    
    Author:    Kashif Rasul <kashif.rasul@gmail.com>
    Co-authored-by: Benjamin Lefaudeux <benjamin.lefaudeux@pm.me>
    
    Tentatively fixing layernorm
    
    - faster all around
    - bugfix
    
    better take on sparse tensors, put layout on the correct device
    update the pip packages, minor cleanup
    kashif authored and blefaudeux committed Apr 21, 2022
    Configuration menu
    Copy the full SHA
    713d6c6 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    3340f74 View commit details
    Browse the repository at this point in the history
  3. faster layernorm

    blefaudeux committed Apr 21, 2022
    Configuration menu
    Copy the full SHA
    e6c4046 View commit details
    Browse the repository at this point in the history
  4. Minor blocksparse refactoring, update block size restrictions, relax …

    …power of two constraint (#277)
    
    * Relax device size restrictions
    
    * Refactor device creation and run all tests
    
    * linting
    
    Co-authored-by: Cole Hawkins <colehawk@amazon.com>
    2 people authored and blefaudeux committed Apr 21, 2022
    Configuration menu
    Copy the full SHA
    7ec9726 View commit details
    Browse the repository at this point in the history
  5. code review, thanks @fmassa !

    blefaudeux committed Apr 21, 2022
    Configuration menu
    Copy the full SHA
    a9c6065 View commit details
    Browse the repository at this point in the history