mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 18:17:44 +00:00
566b5351a3
* 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
[ROCm/composable_kernel commit: 0ac908fb57]
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
|