mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-20 21:09:08 +00:00
37950ea4eb
[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.
27 lines
1.2 KiB
Docker
27 lines
1.2 KiB
Docker
ARG BASE_DOCKER="rocm/pytorch-nightly:latest"
|
|
FROM $BASE_DOCKER
|
|
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 && \
|
|
rm -rf composable_kernel && \
|
|
git clone -b "$CK_PYTORCH_BRANCH" https://github.com/ROCm/composable_kernel.git && \
|
|
cd /tmp/pytorch/third_party/aiter/3rdparty && \
|
|
rm -rf composable_kernel && \
|
|
git clone -b "$CK_PYTORCH_BRANCH" https://github.com/ROCm/composable_kernel.git && \
|
|
cd /tmp/pytorch/third_party/fbgemm/external && \
|
|
rm -rf composable_kernel && \
|
|
git clone -b "$CK_PYTORCH_BRANCH" https://github.com/ROCm/composable_kernel.git && \
|
|
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 && \
|
|
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
|