Skip to content

Comments

Fix compute_block_sparsity import in benchmark_mask_mod#2190

Open
blueberrycongee wants to merge 1 commit intoDao-AILab:mainfrom
blueberrycongee:fix-benchmark-mask-mod-import
Open

Fix compute_block_sparsity import in benchmark_mask_mod#2190
blueberrycongee wants to merge 1 commit intoDao-AILab:mainfrom
blueberrycongee:fix-benchmark-mask-mod-import

Conversation

@blueberrycongee
Copy link

Fixes #2184.

Changes:

  • Import compute_block_sparsity from flash_attn.cute.compute_block_sparsity in tests/cute/benchmark_mask_mod.py.

Notes:

  • Matches maintainer guidance in the issue.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[broken benchmark] benchmark_mask_mod.py is broken with ImportError: cannot import name 'compute_block_sparsity' from 'flash_attn.cute.block_sparsity'

1 participant