Unverified Commit 188defbd authored by Stefano Castagnetta's avatar Stefano Castagnetta Committed by GitHub
Browse files

[CI] Add flashinfer.py to attention test source deps (#38792)


Signed-off-by: default avatarStefano Castagnetta <scastagnetta@nvidia.com>
Co-authored-by: default avatarLuka Govedič <ProExpertProg@users.noreply.github.com>
parent 08ed2b96
......@@ -29,6 +29,7 @@ steps:
- vllm/v1/attention
# TODO: remove this dependency (https://github.com/vllm-project/vllm/issues/32267)
- vllm/model_executor/layers/attention
- vllm/utils/flashinfer.py
- tests/kernels/attention
commands:
- pytest -v -s kernels/attention --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
......
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