From acb2292b466fc228fce132ef33d8e68fe014b925 Mon Sep 17 00:00:00 2001 From: Illia Silin <98187287+illsilin@users.noreply.github.com> Date: Tue, 6 Jan 2026 09:36:54 -0800 Subject: [PATCH] add tabulate package to aiter docker (#3519) [ROCm/composable_kernel commit: 2ffbf7f476d99b6fc3db71480b49d221c602e071] --- Dockerfile.aiter | 2 +- Jenkinsfile | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile.aiter b/Dockerfile.aiter index 94591f9012..020afeccf4 100644 --- a/Dockerfile.aiter +++ b/Dockerfile.aiter @@ -2,7 +2,7 @@ ARG BASE_DOCKER="rocm/pytorch:latest" FROM $BASE_DOCKER ARG AITER_BRANCH="main" ARG CK_AITER_BRANCH="develop" -RUN pip install pandas zmq einops ninja && \ +RUN pip install pandas zmq einops ninja tabulate && \ pip install numpy==1.26.2 && \ sudo mkdir /home/jenkins && \ sudo mkdir /home/jenkins/workspace && \ diff --git a/Jenkinsfile b/Jenkinsfile index 78703ed9aa..7292d9b70c 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -1046,7 +1046,7 @@ def run_aiter_tests(Map conf=[:]){ sh "rocminfo" sh "python3 --version" 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" //temporarily disable + sh "python3 /home/jenkins/workspace/aiter/op_tests/test_gemm_a8w8_blockscale.py" sh "python3 /home/jenkins/workspace/aiter/op_tests/test_mha.py" sh "python3 /home/jenkins/workspace/aiter/op_tests/test_mha_varlen.py" sh "python3 /home/jenkins/workspace/aiter/op_tests/test_moe.py"