ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 06:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a daily CI stage to test AITER with latest CK. (#2598)

Browse Source 
* add a CI stage for AITER testing
This commit is contained in:
Illia Silin
2025-08-01 07:55:51 -07:00
committed by GitHub
parent bb5c478295
commit e6104daecc
2 changed files with 94 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
Dockerfile.aiter Normal file
View File

@@ -0,0 +1,17 @@
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