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

Add daily AITER tests on gfx942. (#2639)

Browse Source 
* add option to select aiter branch, add tests on gfx942

[ROCm/composable_kernel commit: 7ac850ac72]
This commit is contained in:
Illia Silin
2025-08-08 09:30:46 -07:00
committed by GitHub
parent f4247d1996
commit af5a10cb7e
2 changed files with 21 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
Dockerfile.aiter
View File

@@ -1,16 +1,20 @@
ARG BASE_DOCKER="rocm/pytorch:latest" ARG BASE_DOCKER="rocm/pytorch:latest"
FROM $BASE_DOCKER FROM $BASE_DOCKER
RUN groupadd -f render && \ ARG AITER_BRANCH="main"
ARG CK_AITER_BRANCH="develop"
RUN groupadd -g 109 render && \
usermod -u 1001 jenkins && \
groupmod -g 1001 jenkins && \
pip install pandas zmq einops && \ pip install pandas zmq einops && \
pip install numpy==1.26.2 && \ pip install numpy==1.26.2 && \
sudo mkdir /home/jenkins && \ sudo mkdir /home/jenkins && \
sudo mkdir /home/jenkins/workspace && \ sudo mkdir /home/jenkins/workspace && \
cd /home/jenkins/workspace && \ cd /home/jenkins/workspace && \
rm -rf aiter && \ rm -rf aiter && \
git clone --recursive https://github.com/ROCm/aiter.git && \ git clone -b "$AITER_BRANCH" --recursive https://github.com/ROCm/aiter.git && \
cd aiter && \ cd aiter && \
rm -rf 3rdparty/composable_kernel/ && \ rm -rf 3rdparty/composable_kernel/ && \
git clone https://github.com/ROCm/composable_kernel.git 3rdparty/composable_kernel/ && \ git clone -b "$CK_AITER_BRANCH" https://github.com/ROCm/composable_kernel.git 3rdparty/composable_kernel/ && \
python3 setup.py develop && \ python3 setup.py develop && \
chown -R jenkins:jenkins /home/jenkins/workspace && \ chown -R jenkins:jenkins /home/jenkins/workspace && \
chmod -R a+rwx /home/jenkins/workspace && \ chmod -R a+rwx /home/jenkins/workspace && \

20
Jenkinsfile vendored
View File

@@ -190,7 +190,7 @@ def buildDocker(install_prefix){
} }
else if(params.RUN_AITER_TESTS){ else if(params.RUN_AITER_TESTS){
image_name = "rocm/composable_kernel:ck_aiter" image_name = "rocm/composable_kernel:ck_aiter"
dockerArgs = dockerArgs + " --no-cache -f Dockerfile.aiter . " dockerArgs = dockerArgs + " --no-cache -f Dockerfile.aiter --build-arg AITER_BRANCH='${params.aiter_branch}' --build-arg CK_AITER_BRANCH='${params.ck_aiter_branch}' . "
} }
else{ else{
dockerArgs = dockerArgs + " -f Dockerfile . " dockerArgs = dockerArgs + " -f Dockerfile . "
@@ -843,10 +843,10 @@ def run_aiter_tests(Map conf=[:]){
withDockerContainer(image: image, args: dockerOpts) { withDockerContainer(image: image, args: dockerOpts) {
timeout(time: 45, unit: 'MINUTES'){ timeout(time: 45, unit: 'MINUTES'){
try{ try{
sh "python3 --version"
sh "rocminfo" sh "rocminfo"
sh "python3 ../aiter/op_tests/test_gemm_a8w8_blockscale.py" sh "python3 --version"
//sh "python3 ../aiter/op_tests/test_mha.py" sh "python3 /home/jenkins/workspace/aiter/op_tests/test_gemm_a8w8.py"
sh "python3 /home/jenkins/workspace/aiter/op_tests/test_gemm_a8w8_blockscale.py"
} }
catch(e){ catch(e){
echo "Throwing error exception while running AITER tests" echo "Throwing error exception while running AITER tests"
@@ -1009,6 +1009,14 @@ pipeline {
name: "RUN_AITER_TESTS", name: "RUN_AITER_TESTS",
defaultValue: false, defaultValue: false,
description: "Run AITER tests with latest CK develop branch (default: OFF)") description: "Run AITER tests with latest CK develop branch (default: OFF)")
string(
name: 'aiter_branch',
defaultValue: 'main',
description: 'Specify which branch of AITER to use (default: main)')
string(
name: 'ck_aiter_branch',
defaultValue: 'develop',
description: 'Specify which branch of CK to test with AITER (default: develop)')
} }
environment{ environment{
dbuser = "${dbuser}" dbuser = "${dbuser}"
@@ -1093,13 +1101,13 @@ pipeline {
{ {
parallel parallel
{ {
stage("Run AITER Tests on gfx90a") stage("Run AITER Tests on gfx942")
{ {
when { when {
beforeAgent true beforeAgent true
expression { params.RUN_AITER_TESTS.toBoolean() } expression { params.RUN_AITER_TESTS.toBoolean() }
} }
agent{ label rocmnode("gfx90a")} agent{ label rocmnode("gfx942")}
steps{ steps{
run_aiter_tests() run_aiter_tests()
cleanWs() cleanWs()