[rocm-libraries] ROCm/rocm-libraries#7547 (commit 7e032ad)

[CK] fix daily builds for pytorch

## Motivation

This will restore the daily builds that test whether the latest pytorch
code can build with the latest CK code (pulled from the standalone CK
repo).

## Technical Details

<!-- Explain the changes along with any relevant GitHub links. -->

## Test Plan

<!-- Explain any relevant testing done to verify this PR. -->

## Test Result

<!-- Briefly summarize test outcomes. -->

## Submission Checklist

- [ ] Look over the contributing guidelines at
https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
This commit is contained in:
Illia Silin
2026-05-19 14:14:32 +00:00
committed by assistant-librarian[bot]
parent 457f153b69
commit 37950ea4eb
2 changed files with 11 additions and 15 deletions

View File

@@ -4,6 +4,7 @@ ARG CK_PYTORCH_BRANCH="develop"
RUN groupadd -g 109 render && \
usermod -u 1001 jenkins && \
groupmod -g 1001 jenkins && \
pip install --upgrade pandas && \
cd /tmp/pytorch && \
rm -rf build && \
cd /tmp/pytorch/third_party && \
@@ -18,16 +19,8 @@ RUN groupadd -g 109 render && \
cd /tmp/pytorch/third_party/flash-attention/csrc && \
rm -rf composable_kernel && \
git clone -b "$CK_PYTORCH_BRANCH" https://github.com/ROCm/composable_kernel.git && \
chown -R jenkins:jenkins /tmp/pytorch && \
chmod -R a+rwx /tmp/pytorch && \
sudo usermod -aG irc jenkins && \
#install hipblaslt
cd /tmp && \
git clone --no-checkout --filter=blob:none https://github.com/ROCm/rocm-libraries.git && \
cd rocm-libraries && \
git checkout develop && \
git sparse-checkout init --cone && \
git sparse-checkout set projects/hipblaslt shared/origami && \
cd projects/hipblaslt && \
git show --oneline -s && \
CPLUS_INCLUDE_PATH="/opt/amdgpu/include/" ./install.sh -idc --use-system-packages --architecture="gfx942;gfx950" -j 128 --skip_rocroller
mkdir -p /var/jenkins/workspace/pytorch && \
cp -r /tmp/pytorch/* /var/jenkins/workspace/pytorch/ && \
chown -R jenkins:jenkins /var/jenkins/workspace/pytorch && \
chmod -R a+rwx /var/jenkins/workspace/pytorch && \
sudo usermod -aG irc jenkins

7
Jenkinsfile vendored
View File

@@ -1248,8 +1248,11 @@ def run_downstream_tests(Map conf=[:]){
def getPytorchTestsCmds() {
return [
"python3 /tmp/pytorch/tools/amd_build/build_amd.py",
"USE_ROCM_CK_SDPA=1 PYTORCH_ROCM_ARCH=gfx942 python /tmp/pytorch/setup.py develop"
"mkdir pytorch",
"cp -r /var/jenkins/workspace/pytorch/* pytorch/",
"ls -ltr pytorch",
"python3 pytorch/tools/amd_build/build_amd.py",
"cd pytorch && USE_ROCM_CK_SDPA=1 PYTORCH_ROCM_ARCH=gfx942 python3 setup.py develop"
]
}
def getAiterTestsCmds() {