Skip to content

[tests] refactor the pipeline-level attention processor tests - #14701

Open
sayakpaul wants to merge 4 commits into
mainfrom
attn-fusion-tests
Open

[tests] refactor the pipeline-level attention processor tests#14701
sayakpaul wants to merge 4 commits into
mainfrom
attn-fusion-tests

Conversation

@sayakpaul

Copy link
Copy Markdown
Member

Currently, we have got a bunch of QKV fusion tests for pipelines, but these tests don't exercise any pipeline-level features at all.

On the other hand, we have got a handful of QKV fusion tests in the model-level attention test suite. So, this PR removes these pipeline-level tests and adds to the existing model-level QKV fusion tests so that we can retain coverage.

@github-actions github-actions Bot added size/L PR with diff > 200 LOC tests labels Sep 4, 2026
Comment thread tests/models/testing_utils/attention.py Outdated
@sayakpaul

Copy link
Copy Markdown
Member Author

/diffusers-bot pytest tests/models -m attention -k test_fuse_unfuse_qkv_projections

@sayakpaul
sayakpaul marked this pull request as ready for review September 4, 2026 05:55
@github-actions

github-actions Bot commented Sep 4, 2026

Copy link
Copy Markdown
Contributor

pytest tests/models -m attention -k test_fuse_unfuse_qkv_projections passed on GPU — view logs.

@sayakpaul
sayakpaul requested a review from DN6 September 4, 2026 06:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

size/L PR with diff > 200 LOC tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant