Skip to content

Commit f421f3c

Browse files
[CI/Build] Enabling kernels tests for AMD, ignoring some of then that fail (#8130)
1 parent 8c054b7 commit f421f3c

File tree

2 files changed

+24
-1
lines changed

2 files changed

+24
-1
lines changed

.buildkite/run-amd-test.sh

+23-1
Original file line numberDiff line numberDiff line change
@@ -71,13 +71,35 @@ mkdir -p ${HF_CACHE}
7171
HF_MOUNT="/root/.cache/huggingface"
7272

7373
commands=$@
74+
echo "Commands:$commands"
75+
#ignore certain kernels tests
76+
if [[ $commands == *" kernels "* ]]; then
77+
commands="${commands} \
78+
--ignore=kernels/test_attention.py \
79+
--ignore=kernels/test_attention_selector.py \
80+
--ignore=kernels/test_blocksparse_attention.py \
81+
--ignore=kernels/test_causal_conv1d.py \
82+
--ignore=kernels/test_cutlass.py \
83+
--ignore=kernels/test_encoder_decoder_attn.py \
84+
--ignore=kernels/test_flash_attn.py \
85+
--ignore=kernels/test_flashinfer.py \
86+
--ignore=kernels/test_int8_quant.py \
87+
--ignore=kernels/test_machete_gemm.py \
88+
--ignore=kernels/test_mamba_ssm.py \
89+
--ignore=kernels/test_marlin_gemm.py \
90+
--ignore=kernels/test_prefix_prefill.py \
91+
--ignore=kernels/test_rand.py \
92+
--ignore=kernels/test_sampler.py"
93+
fi
94+
7495
PARALLEL_JOB_COUNT=8
7596
# check if the command contains shard flag, we will run all shards in parallel because the host have 8 GPUs.
7697
if [[ $commands == *"--shard-id="* ]]; then
7798
for GPU in $(seq 0 $(($PARALLEL_JOB_COUNT-1))); do
7899
#replace shard arguments
79-
commands=${@//"--shard-id= "/"--shard-id=${GPU} "}
100+
commands=${commands//"--shard-id= "/"--shard-id=${GPU} "}
80101
commands=${commands//"--num-shards= "/"--num-shards=${PARALLEL_JOB_COUNT} "}
102+
echo "Shard ${GPU} commands:$commands"
81103
docker run \
82104
--device /dev/kfd --device /dev/dri \
83105
--network host \

.buildkite/test-pipeline.yaml

+1
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,7 @@ steps:
228228
parallelism: 4
229229

230230
- label: Kernels Test %N # 30min each
231+
mirror_hardwares: [amd]
231232
source_file_dependencies:
232233
- csrc/
233234
- vllm/attention

0 commit comments

Comments
 (0)