Skip to content

GitLab

Unverified Commit 4d396509 authored by Hongxia Yang's avatar Hongxia Yang Committed by GitHub
Browse files

[ROCm] update triton branch to support gpt-oss models for gfx11xx devices (#34032) 


Signed-off-by: default avatarHongxia Yang <hongxia.yang@amd.com>
parent 8fd31f62
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
docker/Dockerfile.rocm_base
View file @ 4d396509
ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:7.0-complete
ARG TRITON_BRANCH="57c693b6" ARG TRITON_BRANCH="f332c492"
ARG TRITON_REPO="https://github.com/ROCm/triton.git" ARG TRITON_REPO="https://github.com/ROCm/triton.git"
ARG PYTORCH_BRANCH="89075173" ARG PYTORCH_BRANCH="89075173"
ARG PYTORCH_REPO="https://github.com/ROCm/pytorch.git" 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