Skip to content

Use rocmlirtriton as the compiler backend - #5215

Open
causten wants to merge 5 commits into
developfrom
use_rocmlirtriton
Open

Use rocmlirtriton as the compiler backend#5215
causten wants to merge 5 commits into
developfrom
use_rocmlirtriton

Conversation

@causten

@causten causten commented Aug 29, 2026

Copy link
Copy Markdown
Collaborator

Motivation

Swap out rocmlir with rocmlirTriton

Technical Details

Changelog Category

Add a CHANGELOG.md entry for any option other than Not Applicable

    • Added: New functionality.
    • Changed: Changes to existing functionality.
    • Removed: Functionality or support that has been removed. (Compared to a previous release)
    • Optimized: Component performance that has been optimized or improved.
    • Resolved Issues: Known issues from a previous version that have been resolved.
    • Not Applicable: This PR is not to be included in the changelog.

Follow the LLVM AI Tool Use Policy for contributions using AI.

@causten
causten requested review from pfultz2 and umangyadav and a lite review from Copilot and removed request for Copilot August 29, 2026 03:17
Comment thread tools/install_prereqs.sh Outdated
# on python3-packaging, and a dpkg-owned copy of packaging cannot be uninstalled
# by pip when requirements-py.txt pins a different version.
pip3 install setuptools wheel pipx
pip3 install setuptools wheel pipx pybind11

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

pybind should not be installed through pip. We already install pybind11 in the requirements.txt file. Also this should not be needed since we arent using the python bindings of triton.

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

In tools/install_prereqs.sh the install of py-requirements.txt happens after rbuild prepare , I'll try swapping the order

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Dont change the order. The py-requirements.txt is for python and rbuild and requirements.txt are for C++ dependencies.

Comment thread requirements.txt Outdated

@umangyadav umangyadav left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Comment thread tools/requirements-py.txt Outdated
Copilot AI lite review requested due to automatic review settings August 31, 2026 14:21

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Pull request overview

Updates the project’s dependency setup to use ROCm/rocmlirTriton in place of ROCm/rocMLIR, aligning the build/prereq tooling with the new compiler backend repository.

Changes:

  • Switch third-party dependency pin from ROCm/rocMLIR to ROCm/rocmlirTriton in requirements.txt.
  • Update rbuild.ini to ignore ROCm/rocmlirTriton in the same CI profiles that already ignore ROCm/rocMLIR.
  • Add pybind11 to Python requirements and reorder the rbuild prepare step in the prereq installer script.

Reviewed changes

Copilot reviewed 3 out of 4 changed files in this pull request and generated 1 comment.

File Description
tools/requirements-py.txt Adds a pinned pybind11 Python package requirement.
tools/install_prereqs.sh Reorders Python package installation vs rbuild prepare for prereqs setup.
requirements.txt Replaces the rocMLIR dependency pin with rocmlirTriton.
rbuild.ini Adds rocmlirTriton to the ignored deps list for selected rbuild profiles.

💡 Add a code-review agent skill or configure MCP servers for context-aware, tailored reviews. Learn more in the docs.

Comment thread tools/install_prereqs.sh Outdated
@causten
causten requested review from dhernandez0 and pfultz2 August 31, 2026 21:02
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.

5 participants