File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
projects/composablekernel Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ ARG BASE_DOCKER="rocm/pytorch:latest"
22FROM $BASE_DOCKER
33ARG AITER_BRANCH="main"
44ARG CK_AITER_BRANCH="develop"
5- RUN pip install pandas zmq einops ninja && \
5+ RUN pip install pandas zmq einops ninja tabulate && \
66 pip install numpy==1.26.2 && \
77 sudo mkdir /home/jenkins && \
88 sudo mkdir /home/jenkins/workspace && \
Original file line number Diff line number Diff line change @@ -1046,7 +1046,7 @@ def run_aiter_tests(Map conf=[:]){
10461046 sh " rocminfo"
10471047 sh " python3 --version"
10481048 sh " python3 /home/jenkins/workspace/aiter/op_tests/test_gemm_a8w8.py"
1049- // sh "python3 /home/jenkins/workspace/aiter/op_tests/test_gemm_a8w8_blockscale.py" //temporarily disable
1049+ sh " python3 /home/jenkins/workspace/aiter/op_tests/test_gemm_a8w8_blockscale.py"
10501050 sh " python3 /home/jenkins/workspace/aiter/op_tests/test_mha.py"
10511051 sh " python3 /home/jenkins/workspace/aiter/op_tests/test_mha_varlen.py"
10521052 sh " python3 /home/jenkins/workspace/aiter/op_tests/test_moe.py"
You can’t perform that action at this time.
0 commit comments