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

[ROCM] Enable ROCM Backend #1: Empty Kernel #1312

Merged
merged 42 commits into from
Mar 25, 2023

Conversation

micmelesse
Copy link
Contributor

This PR is a first in a series of PRs to import the changes that we have made to enable ROCM on our fork of triton.

The PR contains the major changes to the python frontend and enough changes to the c++ backend to allow compilation and running of the empty kernel. We use the ROCM ci added a few weeks ago to verify things.

add scripts

functional empty kernel

remove scripts

clean up

clean more
This is a combination of 3 commits.

Fix Issues related to CI

This is a combination of 19 commits.

add ci

test empty kernel

fix isort

fix autopep8

clang fix

skip lit test for rocm

verbose install

install rocm torch on rocm nodes

remove verbose

explicitly install rocm torch

verbose install on rocm

fix header issue

uninstall torch

uninteractive uninstall

uninstall torchvision and torchaudio

remove torch package

no cache dir

pin version

verbose test output

don't assume env var

run only empty kernel test
@micmelesse micmelesse changed the title [ROCM] Enable Empty Kernel #1 [ROCM] Enable ROCM Backend PR 1: Empty Kernel Mar 15, 2023
@micmelesse micmelesse changed the title [ROCM] Enable ROCM Backend PR 1: Empty Kernel [ROCM] Enable ROCM Backend #1: Empty Kernel Mar 17, 2023
lib/Target/HSACO/HSACOTranslation.cpp Outdated Show resolved Hide resolved
lib/Target/HSACO/HSACOTranslation.cpp Show resolved Hide resolved
python/triton/compiler.py Outdated Show resolved Hide resolved
python/triton/compiler.py Outdated Show resolved Hide resolved
python/triton/compiler.py Outdated Show resolved Hide resolved
python/triton/compiler.py Outdated Show resolved Hide resolved
python/triton/compiler.py Outdated Show resolved Hide resolved
micmelesse and others added 15 commits March 22, 2023 14:01
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
Co-authored-by: Ronan Keryell <ronan@keryell.fr>
include/triton/Conversion/CMakeLists.txt Outdated Show resolved Hide resolved
lib/Target/HSACO/HSACOTranslation.cpp Outdated Show resolved Hide resolved
lib/Target/HSACO/HSACOTranslation.cpp Outdated Show resolved Hide resolved
@ptillet
Copy link
Collaborator

ptillet commented Mar 25, 2023

@micmelesse I'm going to merge this since I'm planning a refactor of the frontend over the weekend, but could you address @keryell 's remaining comment (and any pending work that you're doing) in a separate PR? thanks

@micmelesse
Copy link
Contributor Author

micmelesse commented Mar 28, 2023

@ptillet @keryell I have addressed all the comments here. For those that require code change see. Let me know if there is anything else.

ptillet pushed a commit that referenced this pull request Mar 29, 2023
#1434)

This PR address the remaing issues from #1312. It does the following
*  LLVM String Join
* adds comment to GCNBuilder Class

---------

Co-authored-by: Rahul Batra <rahbatra@amd.com>
@keryell
Copy link
Contributor

keryell commented Mar 29, 2023

@ptillet @keryell I have addressed all the comments here. For those that require code change see. Let me know if there is anything else.

Thanks! I realize that most of my comments depend on some possible code factorizing across back-ends and are probably a bigger task.

@micmelesse micmelesse deleted the upstream_main_pr_1 branch January 11, 2024 19:10
pingzhuu pushed a commit to siliconflow/triton that referenced this pull request Apr 2, 2024
This PR is a first in a series of PRs to import the changes that we have
made to enable ROCM on [our
fork](https://github.com/ROCmSoftwarePlatform/triton) of triton.

The PR contains the major changes to the python frontend and enough
changes to the c++ backend to allow compilation and running of the empty
kernel. We use the ROCM ci added a few weeks ago to verify things.

---------

Co-authored-by: Ronan Keryell <ronan@keryell.fr>
pingzhuu pushed a commit to siliconflow/triton that referenced this pull request Apr 2, 2024
…on-lang#1312 (triton-lang#1434)

This PR address the remaing issues from triton-lang#1312. It does the following
*  LLVM String Join
* adds comment to GCNBuilder Class

---------

Co-authored-by: Rahul Batra <rahbatra@amd.com>
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.

3 participants