Unverified Commit 986ab5db authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

[CI Bugfix] Fix Kernels DeepGEMM Test (H100) (#29106)


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent dd39f91e
......@@ -554,7 +554,6 @@ steps:
timeout_in_minutes: 45
gpu: h100
num_gpus: 1
optional: true
source_file_dependencies:
- tools/install_deepgemm.sh
- vllm/utils/deep_gemm.py
......@@ -565,10 +564,10 @@ steps:
- tests/kernels/moe/test_batched_deepgemm.py
- tests/kernels/attention/test_deepgemm_attention.py
commands:
- pytest -v -s tests/kernels/quantization/test_block_fp8.py -k deep_gemm
- pytest -v -s tests/kernels/moe/test_deepgemm.py
- pytest -v -s tests/kernels/moe/test_batched_deepgemm.py
- pytest -v -s tests/kernels/attention/test_deepgemm_attention.py
- pytest -v -s kernels/quantization/test_block_fp8.py -k deep_gemm
- pytest -v -s kernels/moe/test_deepgemm.py
- pytest -v -s kernels/moe/test_batched_deepgemm.py
- pytest -v -s kernels/attention/test_deepgemm_attention.py
- label: Model Executor Test # 23min
timeout_in_minutes: 35
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment