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

[Triton SMEM] Add not-yet-landed usage of Triton SMEM feature with autotuning #72

Draft
wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

[Triton SMEM] Add not-yet-landed usage of Triton SMEM feature with autotuning #72

wants to merge 1 commit into from

Commits on Aug 22, 2024

  1. [Triton SMEM] Add not-yet-landed usage of Triton SMEM feature with au… 

    …totuning

The following change requires a private patchset that is not yet
available outside of plotfi/triton#4

This patch adds usage of shared memory using the tl.local_copy and
tl.gather operations for the TW (time bias) and PW (position bias)
tensors for the forward pass kernel.

Autotuning is also hooked up to the usage of these shared memory operators
    @plotfi
    plotfi committed Aug 22, 2024
    Configuration menu
    Copy the full SHA
    f57284a View commit details
    Browse the repository at this point in the history