ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix AITER tests. (#3106)

Browse Source 
* change base docker image for aiter

* do not add group irc to aiter docker

* add user and group jenkins

* pip install ninja

* update permissions for /home/jenkins

[ROCm/composable_kernel commit: e02b1e7caf]
This commit is contained in:
Illia Silin
2025-10-27 20:59:21 -07:00
committed by GitHub
parent 71fef01df6
commit 689ae06dbc
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Dockerfile.aiter
View File

@@ -1,9 +1,8 @@
ARG BASE_DOCKER="rocm/composable_kernel-private:ck_aiter_base"
ARG BASE_DOCKER="rocm/pytorch:latest"
FROM $BASE_DOCKER
ARG AITER_BRANCH="main"
ARG CK_AITER_BRANCH="develop"
RUN groupadd irc && \
pip install pandas zmq einops && \
RUN pip install pandas zmq einops ninja && \
pip install numpy==1.26.2 && \
sudo mkdir /home/jenkins && \
sudo mkdir /home/jenkins/workspace && \
@@ -14,6 +13,8 @@ RUN groupadd irc && \
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 && \
groupadd -g 1001 jenkins && \
useradd -u 1001 -g 1001 -m -s /bin/bash jenkins && \
chown -R jenkins:jenkins /home/jenkins && \
chmod -R a+rwx /home/jenkins && \
sudo usermod -aG irc jenkins