mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 18:17:44 +00:00
8c0cdebe63
* enable aiter and fmha test stages on gfx950
* use newer compiler for gfx950
* make sure gfx950 runs correct docker
* fix typo
* upgrade base docker for aiter
* change base docker for aiter tests
* do not add group render to ck_aiter image
* add group irc in ck_aiter docker
* do not fix the irc group id to 39
* do not set jenkins uid and gid
* skip group irc for aiter tests
* fix syntax error in dockerfile
* change the base docker for aiter tests
* add irc group back to ck_aiter docker
[ROCm/composable_kernel commit: b9d69d32a8]
20 lines
801 B
Docker
20 lines
801 B
Docker
ARG BASE_DOCKER="rocm/composable_kernel-private:ck_aiter_base"
|
|
FROM $BASE_DOCKER
|
|
ARG AITER_BRANCH="main"
|
|
ARG CK_AITER_BRANCH="develop"
|
|
RUN groupadd irc && \
|
|
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 -b "$AITER_BRANCH" --recursive https://github.com/ROCm/aiter.git && \
|
|
cd aiter && \
|
|
rm -rf 3rdparty/composable_kernel/ && \
|
|
git clone -b "$CK_AITER_BRANCH" 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
|