mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-04 21:51:28 +00:00
0ac908fb57
* add a stage to builf pytorch * add docker file for pytorch stage * call build scripts fro mthe default path * add a daily chron build for pytorcn stage
24 lines
1.0 KiB
Docker
24 lines
1.0 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 && \
|
|
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 && \
|
|
chown -R jenkins:jenkins /tmp/pytorch && \
|
|
chmod -R a+rwx /tmp/pytorch && \
|
|
sudo usermod -aG irc jenkins
|