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.

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

Allow triton.code_gen.Binary to print asm. #89

Merged
merged 4 commits into from
Apr 23, 2021

Conversation

daadaada
Copy link
Contributor

An example can be found in triton/python/tutorical/01-vector-add.py

print(kernel.asm('ptx'))

@ptillet
Copy link
Collaborator

ptillet commented Apr 23, 2021

Hi, a few comments:

  • Can you rename asm('ir') to asm('ttir') or something? ir makes it a bit unclear that it is the Triton-IR rather than the LLVM-IR
  • Can you leave 01-vector-add.py ?I think in general returning a handle to the binary that was executed is not what we want users to do, so I'd like to document printing asm less aggressively for now. We should probably improve the API so that one can also do _add.asm(**meta) without having to actually run the op; we could then document that in the tutorial.

@daadaada daadaada changed the title Allow triton.code_gen.Binary to print asm Allow triton.code_gen.Binary to print asm. Apr 23, 2021
@ptillet ptillet merged commit 139fb98 into triton-lang:master Apr 23, 2021
dfukalov pushed a commit to dfukalov/triton that referenced this pull request Feb 9, 2023
…yernor_bwd_pass

Enable layernorm backward pass.
pingzhuu pushed a commit to siliconflow/triton that referenced this pull request Apr 2, 2024
oraluben pushed a commit to oraluben/triton that referenced this pull request Sep 11, 2024
pawelszczerbuk pushed a commit that referenced this pull request Nov 7, 2024
Change our BW experimental kernel to use TMA.
This only changes the forward pass for now as this is what we are
concentrating on for perf.
gglin001 pushed a commit to gglin001/triton that referenced this pull request Nov 13, 2024
stephen-huan pushed a commit to stephen-huan/triton that referenced this pull request Dec 24, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants