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

[CK_TILE] LDS prefetch pipeline support for FlashAttention #1349

Open
wants to merge 21 commits into
base: develop
Choose a base branch
from
Open

[CK_TILE] LDS prefetch pipeline support for FlashAttention #1349

wants to merge 21 commits into from
+1,324 −178

Commits on Jun 18, 2024

  1. Merge pull request #1 from ROCm/develop 

    upstream update
    @ramjana
    ramjana committed Jun 18, 2024
    Configuration menu
    Copy the full SHA
    3ba573d View commit details
    Browse the repository at this point in the history

  2. LDS prefetch pipeline support

    Raman committed Jun 18, 2024
    Configuration menu
    Copy the full SHA
    4321800 View commit details
    Browse the repository at this point in the history

Commits on Aug 13, 2024

  1. Merge remote-tracking branch 'upstream/develop' into develop

    Raman committed Aug 13, 2024
    Configuration menu
    Copy the full SHA
    917680d View commit details
    Browse the repository at this point in the history

  2. AMD fa3-fwd prototype +600Tflops

    Raman committed Aug 13, 2024
    Configuration menu
    Copy the full SHA
    ce50297 View commit details
    Browse the repository at this point in the history

Commits on Aug 14, 2024

  1. Merge branch 'develop' into develop

    @poyenc
    poyenc committed Aug 14, 2024
    Configuration menu
    Copy the full SHA
    e84338b View commit details
    Browse the repository at this point in the history

Commits on Aug 15, 2024

  1. Re-format files

    @poyenc
    poyenc committed Aug 15, 2024
    Configuration menu
    Copy the full SHA
    85725f0 View commit details
    Browse the repository at this point in the history

Commits on Aug 20, 2024

  1. Merge branch 'develop' into develop

    @poyenc
    poyenc committed Aug 20, 2024
    Configuration menu
    Copy the full SHA
    39993c0 View commit details
    Browse the repository at this point in the history

Commits on Aug 21, 2024

  1. Merge branch 'develop' into develop

    @poyenc
    poyenc committed Aug 21, 2024
    Configuration menu
    Copy the full SHA
    bb19358 View commit details
    Browse the repository at this point in the history

Commits on Aug 27, 2024

  1. Remove duplicated function

    @poyenc
    poyenc committed Aug 27, 2024
    Configuration menu
    Copy the full SHA
    69d01f6 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'develop' into develop

    @poyenc
    poyenc committed Aug 27, 2024
    Configuration menu
    Copy the full SHA
    5a9702f View commit details
    Browse the repository at this point in the history

  3. Merge branch 'develop' of github.com:ramjana/composable_kernel into d… 

    …evelop
    @poyenc
    poyenc committed Aug 27, 2024
    Configuration menu
    Copy the full SHA
    f176a13 View commit details
    Browse the repository at this point in the history

Commits on Aug 29, 2024

  1. Merge branch 'develop' into develop

    @poyenc
    poyenc committed Aug 29, 2024
    Configuration menu
    Copy the full SHA
    0490ba2 View commit details
    Browse the repository at this point in the history

Commits on Aug 30, 2024

  1. Merge branch 'develop' into develop

    @poyenc
    poyenc committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    fd61c76 View commit details
    Browse the repository at this point in the history

  2. Comment out fail-to-compile pipelines

    @poyenc
    poyenc committed Aug 30, 2024
    Configuration menu
    Copy the full SHA
    dc31a9a View commit details
    Browse the repository at this point in the history

Commits on Aug 31, 2024

  1. Only use 'qr' pipeline for fmha_fwd() api

    @poyenc
    poyenc committed Aug 31, 2024
    Configuration menu
    Copy the full SHA
    49fb890 View commit details
    Browse the repository at this point in the history

Commits on Sep 1, 2024

  1. Fix LDS write sync logics

    @poyenc
    poyenc committed Sep 1, 2024
    Configuration menu
    Copy the full SHA
    242cdcd View commit details
    Browse the repository at this point in the history

  2. At most using 256 threads to load tile

    @poyenc
    poyenc committed Sep 1, 2024
    Configuration menu
    Copy the full SHA
    3aa5cb2 View commit details
    Browse the repository at this point in the history

Commits on Sep 2, 2024

  1. Duplicate V tile data for wave groups

    @poyenc
    poyenc committed Sep 2, 2024
    Configuration menu
    Copy the full SHA
    4dd640a View commit details
    Browse the repository at this point in the history

Commits on Sep 3, 2024

  1. Update license & macro usage

    @poyenc
    poyenc committed Sep 3, 2024
    Configuration menu
    Copy the full SHA
    8141046 View commit details
    Browse the repository at this point in the history

  2. fa3 prototype with correct QKV tiling-sizes; head-dim=256 requires ch… 

    …anges to pipeline
    Raman committed Sep 3, 2024
    Configuration menu
    Copy the full SHA
    d4b2d94 View commit details
    Browse the repository at this point in the history

Commits on Sep 4, 2024

  1. Merge branch 'develop' into develop

    @poyenc
    poyenc committed Sep 4, 2024
    Configuration menu
    Copy the full SHA
    695fa56 View commit details
    Browse the repository at this point in the history