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

[XLA:GPU] Improve memory bandwidth utilization of column reduction #11018

Closed
wants to merge 47 commits into from
Closed

[XLA:GPU] Improve memory bandwidth utilization of column reduction #11018

wants to merge 47 commits into from

Commits on Mar 28, 2024

  1. optimize column reduction

    @lingzhi98
    lingzhi98 committed Mar 28, 2024
    Configuration menu
    Copy the full SHA
    cd60cb7 View commit details
    Browse the repository at this point in the history

  2. missing file

    @lingzhi98
    lingzhi98 committed Mar 28, 2024
    Configuration menu
    Copy the full SHA
    b8f3770 View commit details
    Browse the repository at this point in the history

  3. typo

    @lingzhi98
    lingzhi98 committed Mar 28, 2024
    Configuration menu
    Copy the full SHA
    d67859b View commit details
    Browse the repository at this point in the history

  4. disable tiling adjustment for mlir reduction codegen and gpu performa… 

    …nce estimate
    @lingzhi98
    lingzhi98 committed Mar 28, 2024
    Configuration menu
    Copy the full SHA
    e364b0c View commit details
    Browse the repository at this point in the history

  5. format

    @lingzhi98
    lingzhi98 committed Mar 28, 2024
    Configuration menu
    Copy the full SHA
    b33507c View commit details
    Browse the repository at this point in the history

  6. fix reduction mlir test build error

    @lingzhi98
    lingzhi98 committed Mar 28, 2024
    Configuration menu
    Copy the full SHA
    37474bf View commit details
    Browse the repository at this point in the history

Commits on Mar 29, 2024

  1. refine

    @lingzhi98
    lingzhi98 committed Mar 29, 2024
    Configuration menu
    Copy the full SHA
    ad8fe37 View commit details
    Browse the repository at this point in the history

Commits on Mar 30, 2024

  1. comments

    @lingzhi98
    lingzhi98 committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    90cbfa6 View commit details
    Browse the repository at this point in the history

  2. fix filecheck error

    @lingzhi98
    lingzhi98 committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    1a725c6 View commit details
    Browse the repository at this point in the history

  3. reduce the number of syncs

    @lingzhi98
    lingzhi98 committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    a510ce7 View commit details
    Browse the repository at this point in the history

  4. remove unused variable

    @lingzhi98
    lingzhi98 committed Mar 30, 2024
    Configuration menu
    Copy the full SHA
    08169b7 View commit details
    Browse the repository at this point in the history

Commits on Apr 12, 2024

  1. add tests

    @lingzhi98
    lingzhi98 committed Apr 12, 2024
    Configuration menu
    Copy the full SHA
    1453d73 View commit details
    Browse the repository at this point in the history

  2. modify tests

    @lingzhi98
    lingzhi98 committed Apr 12, 2024
    Configuration menu
    Copy the full SHA
    0caba39 View commit details
    Browse the repository at this point in the history

Commits on Apr 16, 2024

  1. estimate compute time of column reduction fusion correctly

    @lingzhi98
    lingzhi98 committed Apr 16, 2024
    Configuration menu
    Copy the full SHA
    7b465eb View commit details
    Browse the repository at this point in the history

  2. modify comment

    @lingzhi98
    lingzhi98 committed Apr 16, 2024
    Configuration menu
    Copy the full SHA
    4ac609d View commit details
    Browse the repository at this point in the history

Commits on Apr 18, 2024

  1. estimate compute time correctly for kernels which use shared cache

    @lingzhi98
    lingzhi98 committed Apr 18, 2024
    Configuration menu
    Copy the full SHA
    b75bbba View commit details
    Browse the repository at this point in the history

  2. make block counts larger than core counts

    @lingzhi98
    lingzhi98 committed Apr 18, 2024
    Configuration menu
    Copy the full SHA
    8c0e306 View commit details
    Browse the repository at this point in the history

Commits on Apr 23, 2024

  1. update reduction mlir emitter

    @lingzhi98
    lingzhi98 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    2605755 View commit details
    Browse the repository at this point in the history

  2. modify comment

    @lingzhi98
    lingzhi98 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    87bb413 View commit details
    Browse the repository at this point in the history

  3. correct the condition of using shared cache

    @lingzhi98
    lingzhi98 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    6976ee6 View commit details
    Browse the repository at this point in the history

  4. update compute time calculation in indexing performance model

    @lingzhi98
    lingzhi98 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    8c63413 View commit details
    Browse the repository at this point in the history

  5. fix merge conflict

    @lingzhi98
    lingzhi98 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    45904ee View commit details
    Browse the repository at this point in the history

  6. remove unused variable

    @lingzhi98
    lingzhi98 committed Apr 23, 2024
    Configuration menu
    Copy the full SHA
    4caf6d4 View commit details
    Browse the repository at this point in the history

Commits on Apr 24, 2024

  1. conflict

    @lingzhi98
    lingzhi98 committed Apr 24, 2024
    Configuration menu
    Copy the full SHA
    5e68018 View commit details
    Browse the repository at this point in the history

  2. fix reduction base test

    @lingzhi98
    lingzhi98 committed Apr 24, 2024
    Configuration menu
    Copy the full SHA
    24f71c8 View commit details
    Browse the repository at this point in the history

  3. format

    @lingzhi98
    lingzhi98 committed Apr 24, 2024
    Configuration menu
    Copy the full SHA
    c7e8ca2 View commit details
    Browse the repository at this point in the history

Commits on Apr 26, 2024

  1. use alloca

    @lingzhi98
    lingzhi98 committed Apr 26, 2024
    Configuration menu
    Copy the full SHA
    f4d7713 View commit details
    Browse the repository at this point in the history

Commits on May 1, 2024

  1. rotate iter args

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    4fe58f2 View commit details
    Browse the repository at this point in the history

  2. conflict

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    6d08bc5 View commit details
    Browse the repository at this point in the history

  3. hlo change

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    a9a48d6 View commit details
    Browse the repository at this point in the history

  4. fix

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    444f36c View commit details
    Browse the repository at this point in the history

  5. fix build error

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    9ac2a26 View commit details
    Browse the repository at this point in the history

  6. restore comments

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    7f4200c View commit details
    Browse the repository at this point in the history

  7. remove unused variable

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    496be7f View commit details
    Browse the repository at this point in the history

  8. format

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    954cbf1 View commit details
    Browse the repository at this point in the history

  9. remove

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    a166784 View commit details
    Browse the repository at this point in the history

  10. keep usage consistent

    @lingzhi98
    lingzhi98 committed May 1, 2024
    Configuration menu
    Copy the full SHA
    77ca02e View commit details
    Browse the repository at this point in the history

Commits on May 6, 2024

  1. conflict

    @lingzhi98
    lingzhi98 committed May 6, 2024
    Configuration menu
    Copy the full SHA
    5e71fba View commit details
    Browse the repository at this point in the history

  2. fix build error

    @lingzhi98
    lingzhi98 committed May 6, 2024
    Configuration menu
    Copy the full SHA
    5a481e6 View commit details
    Browse the repository at this point in the history

  3. remove redundant comments

    @lingzhi98
    lingzhi98 committed May 6, 2024
    Configuration menu
    Copy the full SHA
    5463bbe View commit details
    Browse the repository at this point in the history

  4. remove legacy emitter related change

    @lingzhi98
    lingzhi98 committed May 6, 2024
    Configuration menu
    Copy the full SHA
    bdba6a6 View commit details
    Browse the repository at this point in the history

  5. Merge remote-tracking branch 'origin'

    @lingzhi98
    lingzhi98 committed May 6, 2024
    Configuration menu
    Copy the full SHA
    b1c53f2 View commit details
    Browse the repository at this point in the history

Commits on May 7, 2024

  1. move cost model change to another pull request

    @lingzhi98
    lingzhi98 committed May 7, 2024
    Configuration menu
    Copy the full SHA
    a35ef76 View commit details
    Browse the repository at this point in the history

  2. check thread indexing of vectorize column reduction for mlir reductio… 

    …n emitter
    @lingzhi98
    lingzhi98 committed May 7, 2024
    Configuration menu
    Copy the full SHA
    c0cf49c View commit details
    Browse the repository at this point in the history

  3. fix hlo filecheck error

    @lingzhi98
    lingzhi98 committed May 7, 2024
    Configuration menu
    Copy the full SHA
    b248c32 View commit details
    Browse the repository at this point in the history

Commits on May 8, 2024

  1. remove unnecessary constraint check

    @lingzhi98
    lingzhi98 committed May 8, 2024
    Configuration menu
    Copy the full SHA
    2554006 View commit details
    Browse the repository at this point in the history

  2. add comments

    @lingzhi98
    lingzhi98 committed May 8, 2024
    Configuration menu
    Copy the full SHA
    d3d0679 View commit details
    Browse the repository at this point in the history