OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
12cb11a268
pytorch/torch/testing
History
jianan-gu 12cb11a268 [Inductor UT] Refactor FlexAttention UT and add CPU tests (#144953) 
This PR extends and refines all rest UTs for CPU and more devices in `test/inductor/test_flex_attention.py`  and `test/inductor/test_flex_decoding.py`, as a follow-up to https://github.com/pytorch/pytorch/pull/141453

Pull Request resolved: https://github.com/pytorch/pytorch/pull/144953
Approved by: https://github.com/drisspg
2025-04-15 12:44:49 +00:00
..
_internal [Inductor UT] Refactor FlexAttention UT and add CPU tests (#144953) 2025-04-15 12:44:49 +00:00
__init__.py
_comparison.py Fix code descriptions in the test package. (#148145) 2025-03-04 19:14:41 +00:00
_creation.py PEP585 update - torch/testing (#145200) 2025-01-20 22:42:42 +00:00
_utils.py Fix code descriptions in the test package. (#148145) 2025-03-04 19:14:41 +00:00