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

Atomic Load and Store operations for Triton (tl.atomic_store/tl.atomic_load) #5187

Draft
wants to merge 2 commits into
base: main
Choose a base branch
from
Draft

Atomic Load and Store operations for Triton (tl.atomic_store/tl.atomic_load) #5187

wants to merge 2 commits into from
+435 −54

Commits on Nov 19, 2024

  1. [FRONTEND] Implement load and store sem and scope 

    PTX ld and st support scopes { .cta, .cluster, .gpu, .sys } and
sub-opcodes relaxed and acquire.

This patch adds frontend support for them.

(cherry picked from commit ec9fc96)
(cherry picked from commit 7d33470)
    @plotfi
    plotfi committed Nov 19, 2024
    Configuration menu
    Copy the full SHA
    848fbac View commit details
    Browse the repository at this point in the history

  2. [BACKEND] Support for tl.atomic_load and tl.atomic_store codegen for … 

    …PTX ld/st

This change implements codegen for MemSemantic flags and Scope flags for atomic
load and store ops.

(cherry picked from commit b512a43)
(cherry picked from commit 871e8f7)
    @plotfi
    plotfi committed Nov 19, 2024
    Configuration menu
    Copy the full SHA
    5984863 View commit details
    Browse the repository at this point in the history