Skip to content

refactor: rename BLOCK_EXPANDING/block_diffusion to BLOCK_EXTEND - #8

Merged
fdz-1999 merged 1 commit into
fdz-1999:feature/block-extendfrom
leon062112:feat/rename-block-extend
Jul 21, 2026
Merged

fdz-1999 merged 1 commit into
fdz-1999:feature/block-extendfrom
leon062112:feat/rename-block-extend

Conversation

@leon062112

Copy link
Copy Markdown

…k_extend

  • MaskMode.BLOCK_EXPANDING -> MaskMode.BLOCK_EXTEND (enum value)
  • kBlockExpanding -> kBlockExtend (C++ enum)
  • BLOCK_EXPANDING -> BLOCK_EXTEND (C++ template params)
  • block_expanding_prefill.cuh -> block_extend_prefill.cuh (file rename)
  • q/kv_block_expanding_offset -> q/kv_block_extend_offset (members)
  • block_diffusion -> block_extend (Python params/attrs/functions/files)
  • _block_diffusion.py -> _block_extend.py (file rename)
  • _block_diffusion_batch.py -> _block_extend_batch.py (file rename)
  • test_block_diffusion_attention.py -> test_block_extend_attention.py
$ /opt/conda/bin/python3 -m pytest tests/attention/test_block_extend_attention.py -v --no-header
==================================================================================== test session starts =====================================================================================
collected 4 items                                                                                                                                                                            

tests/attention/test_block_extend_attention.py::test_block_extend_batch_generator_routing PASSED                                                                                       [ 25%]
tests/attention/test_block_extend_attention.py::test_block_extend_single_prefill_matches_reference PASSED                                                                              [ 50%]
tests/attention/test_block_extend_attention.py::test_block_extend_ragged_wrapper_matches_reference PASSED                                                                              [ 75%]
tests/attention/test_block_extend_attention.py::test_block_extend_paged_wrapper_matches_reference PASSED                                                                               [100%]

====================================================================================== warnings summary ======================================================================================

…k_extend

- MaskMode.BLOCK_EXPANDING -> MaskMode.BLOCK_EXTEND (enum value)
- kBlockExpanding -> kBlockExtend (C++ enum)
- BLOCK_EXPANDING -> BLOCK_EXTEND (C++ template params)
- block_expanding_prefill.cuh -> block_extend_prefill.cuh (file rename)
- q/kv_block_expanding_offset -> q/kv_block_extend_offset (members)
- block_diffusion -> block_extend (Python params/attrs/functions/files)
- _block_diffusion.py -> _block_extend.py (file rename)
- _block_diffusion_batch.py -> _block_extend_batch.py (file rename)
- test_block_diffusion_attention.py -> test_block_extend_attention.py

15 files, pure rename — no logic changes.

Co-Authored-By: Claude <noreply@anthropic.com>
@fdz-1999
fdz-1999 merged commit 02e618e into fdz-1999:feature/block-extend Jul 21, 2026
2 of 4 checks passed
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.

2 participants