Skip to content

GitLab

"tests/compile/distributed/__init__.py" did not exist on "e575df33b1963212f79b2efddd45d53225fa5c45"
Unverified Commit ab3a85fd authored by Gregory Shtrasberg's avatar Gregory Shtrasberg Committed by GitHub
Browse files

[ROCm][CI/Build] Fix triton version to one that has triton_kernels required... 


[ROCm][CI/Build] Fix triton version to one that has triton_kernels required for gpt-oss to run (#31159)
Signed-off-by: default avatarGregory Shtrasberg <Gregory.Shtrasberg@amd.com>
parent 8dd0db68
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
docker/Dockerfile.rocm_base
View file @ ab3a85fd
ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete
ARG TRITON_BRANCH="a272dfa8"
ARG TRITON_BRANCH="57c693b6"
ARG TRITON_REPO="https://github.com/ROCm/triton.git"
ARG PYTORCH_BRANCH="89075173"
ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment