Sayak Paul
c2798cc785
[tests] refactor the pipeline-level attention processor tests ( #14701 )
...
* refactor the pipeline-level attention processor tests
* simplify stuff.
* remove utils.py
2026-10-05 15:38:23 +05:30
Sayak Paul
fef717ffb0
[core] propagate sage attention updates. ( #14584 )
...
* propagate sage attention updates.
* add sage blackwell.
* up
* style.
* address feedback
* style.
2026-09-29 18:47:23 +05:30
Sayak Paul
5f33428194
update licensing for others to 2026. ( #14128 )
...
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com >
2026-07-08 07:40:20 +05:30
Cheung Ka Wai
5be68ced49
fix _flash_3_varlen_hub mask handling ( #14115 )
...
* fix fa3 mask
* add test case
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2026-07-06 10:33:11 +05:30
Sayak Paul
cbdedbaf03
[tests] add attention backend tests. ( #13174 )
...
* add attention backend tests.
* remove existing tests/others/test_attention_backends.py file
* modify generate_model_tests.py
* remove native.
* account for _keep_in_fp32_modules
* don't skip when exception is raised.
* use is_kernels_available()
* mark with compile.
* move rtol and atol to methods as defaults.
* Apply suggestions from code review
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
* up
* up
2026-05-12 17:00:00 +09:00
Dhruv Nair
0b76728e27
Refactor Model Tests ( #12822 )
...
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
* update
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2026-02-02 18:51:44 +05:30
