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

Add support for 3-5D TMA to allow loading non-matmul operands #5207

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

Add support for 3-5D TMA to allow loading non-matmul operands #5207

wants to merge 12 commits into from
+178 −78

Commits on Nov 18, 2024

  1. Add support for higher dim TMA

    Masahiro Masuda committed Nov 18, 2024
    Configuration menu
    Copy the full SHA
    1640404 View commit details
    Browse the repository at this point in the history

  2. fix

    Masahiro Masuda committed Nov 18, 2024
    Configuration menu
    Copy the full SHA
    8332845 View commit details
    Browse the repository at this point in the history

Commits on Nov 20, 2024

  1. disable swizzling and hasLeadingOffset for > 2D TMA

    Masahiro Masuda committed Nov 20, 2024
    Configuration menu
    Copy the full SHA
    517ea83 View commit details
    Browse the repository at this point in the history

  2. parametrize test

    Masahiro Masuda committed Nov 20, 2024
    Configuration menu
    Copy the full SHA
    028000c View commit details
    Browse the repository at this point in the history

  3. add comment on disabling swizzling

    Masahiro Masuda committed Nov 20, 2024
    Configuration menu
    Copy the full SHA
    04f48c5 View commit details
    Browse the repository at this point in the history

  4. run ruff

    Masahiro Masuda committed Nov 20, 2024
    Configuration menu
    Copy the full SHA
    d228adf View commit details
    Browse the repository at this point in the history

  5. remove rank <= 2 assert

    Masahiro Masuda committed Nov 20, 2024
    Configuration menu
    Copy the full SHA
    d5b5b1b View commit details
    Browse the repository at this point in the history

Commits on Nov 21, 2024

  1. add comment on test

    Masahiro Masuda committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    38d6cd4 View commit details
    Browse the repository at this point in the history

  2. precommit

    Masahiro Masuda committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    a9f4e30 View commit details
    Browse the repository at this point in the history

  3. remove ttgir test

    Masahiro Masuda committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    e19f439 View commit details
    Browse the repository at this point in the history

  4. Merge branch 'main' into tma-higher-dim

    Masahiro Masuda committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    591ad5d View commit details
    Browse the repository at this point in the history

Commits on Nov 22, 2024

  1. Merge branch 'main' into tma-higher-dim

    Masahiro Masuda committed Nov 22, 2024
    Configuration menu
    Copy the full SHA
    e3f0282 View commit details
    Browse the repository at this point in the history