mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-04-19 22:39:03 +00:00
18 lines
687 B
Docker
18 lines
687 B
Docker
ARG BASE_DOCKER="rocm/pytorch:latest"
|
|
FROM $BASE_DOCKER
|
|
RUN groupadd -f render && \
|
|
pip install pandas zmq einops && \
|
|
pip install numpy==1.26.2 && \
|
|
sudo mkdir /home/jenkins && \
|
|
sudo mkdir /home/jenkins/workspace && \
|
|
cd /home/jenkins/workspace && \
|
|
rm -rf aiter && \
|
|
git clone --recursive https://github.com/ROCm/aiter.git && \
|
|
cd aiter && \
|
|
rm -rf 3rdparty/composable_kernel/ && \
|
|
git clone https://github.com/ROCm/composable_kernel.git 3rdparty/composable_kernel/ && \
|
|
python3 setup.py develop && \
|
|
chown -R jenkins:jenkins /home/jenkins/workspace && \
|
|
chmod -R a+rwx /home/jenkins/workspace && \
|
|
sudo usermod -aG irc jenkins
|