 Tong WUandcoderabbitai[bot]
|
3ab93cd76b
|
[Enhancement] Keep max score attention across blocks in FlashAttention for better numerical stablity (#1269)
* Implement max score retention across blocks in FlashAttention for improved stability
* fix manual pipeline parameters
* Update examples/flash_attention/example_gqa_fwd_varlen.py
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
* fix typo
* more
* fix a previous typo
---------
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
|
2025-11-17 21:53:19 +08:00 |
|
|
Tong WU
|
0c7e74196e
|
[Cleanup] Remove tilelang.disable_cache() calls from examples and tests (#1088)
* [Cleanup] Remove `tilelang.disable_cache()` calls from example scripts
* lint
* lint
|
2025-10-21 16:34:30 +08:00 |
|
|
|
8ce2778221
|
[CI][Refactor] Merge test CI workflow files into one (#973)
* refactor: merge test CI workflow files into one
* chore: set `UV_INDEX_STRATEGY=unsafe-best-match`
* feat: add AST test with Python 3.8
* feat: implement manual caching mechanism for self-hosted runners
* refactor: simplify cache logic for self-hosted runners
* chore: clear uv cache on failure
* chore: print format.sh output to logs
* chore: improve uv caching
* chore: disable parallel test
* chore: use `PYTHONDEVMODE=1` in CI
* feat: enable coredump generation
* fix: fix perfbench condition
* Revert "feat: enable coredump generation"
This reverts commit c52da65cb5.
* chore: move example CI down
* Revert "chore: move example CI down"
This reverts commit 9d8e65055e.
* chore: skip example `test_example_mha_sink_bwd_bhsd`
* chore: skip example `test_example_gqa_sink_bwd_bhsd`
* fix: fix example argument passing
* fix: loosen test criteria
* chore: rename `CMAKE_CONFIGURE_OPTIONS` -> `CLANG_TIDY_CMAKE_OPTIONS` for clarity
* feat: enable parallel testings
* chore: update pytest options
* remove skipped test as now been resolved
* chore: empty commit to re-trigger ci
* test for n 1
* chore: remove ` --numprocesses=1` option in example
* chore: disable failfast
* chore: update cibw selection
* fix: fix git submodule clone
* chore: update cibw commands
* fix: fix yapf multiprocessing
* chore: setup ccache for CIBW on macOS only
* chore: update comments
* chore: update artifact listing
* fix: do not fail if not found nvcc in PATH
* fix: fix flash-attn installation
* chore: update dist workflow trigger
* chore: remove outdated comments
* chore(workflows/dist): simplify build matrix strategy
* fix: fix CUDA path finding
* fix: fix CUDA path finding
* chore: imcrease CI timeout
* ci: disable failfast
* fix: hide path prefix
* chore: more verbose
* chore: disable PR trigger for dist workflow
* fix: seed for tests
* fix: use nightly torch for ROCm tests
* chore: enable PR trigger for dist workflow
* chore: stop uploading debug wheels as artifacts in PR
* chore: do not run workflows in forks
* chore: housekeep requirements
* chore: use Nightly-ROCm-6.3 for CI
* chore: use Nightly-ROCm-6.4 for CI
* Update ROCm toolkit version to 7.0
* chore: restore previous rocm-ci.yml for test
* fix: cleanup PYTHONPATH
* chore: remove previous rocm-ci.yml
* ci fix
* chore: remove previous rocm-ci.yml
* chore: enable parallel example run
---------
Co-authored-by: LeiWang1999 <leiwang1999@outlook.com>
Co-authored-by: alex_xiao <xinyuxiao2024@gmail.com>
|
2025-10-15 22:12:41 +08:00 |
|
|
Yichen Yan
|
5e52952201
|
[Lint] Add ruff config to check for useless spaces (#807)
* update lint config
* Remove spaces for blank line
* update
|
2025-09-13 21:06:25 +08:00 |
|
|
Wenhao Xie
|
013adca01a
|
[Bugfix] Fix incorrect synchronization bug in minference example (#786)
* fix
* lint
|
2025-09-05 17:47:15 +08:00 |
|
|
Wenhao Xie
|
37051417b7
|
[Example] Add vertical slash sparse attention pattern (#762)
* upd sparse attn
* lint
* rename
* update test file
* update benchmark
* lint
* update benchmark
|
2025-08-28 14:34:17 +08:00 |
|