Cheung Ka Wai
e1e7d58a4a
Fix Ulysses SP backward with SDPA ( #13328 )
...
* add UT for backward
* fix SDPA attention backward
2026-03-30 15:15:27 +05:30
Cheung Ka Wai
afdda57f61
Fix the attention mask in ulysses SP for QwenImage ( #13278 )
...
* fix mask in SP
* change the modification to qwen specific
* drop xfail since qwen-image mask is fixed
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2026-03-24 02:12:50 -07:00
Cheung Ka Wai
9d4c9dcf21
change QwenImageTransformer UT to batch inputs ( #13312 )
...
* UT expands to batch inputs
* update according to suggestion
* update according to suggestion 2
* fix CI
* update according to suggestion 3
* clean line
2026-03-24 08:56:40 +05:30
Wang, Yi
9677859ebf
fix parallelism case failure in xpu ( #13270 )
...
* fix parallelism case failure in xpu
Signed-off-by: Wang, Yi <yi.a.wang@intel.com >
* updated
Signed-off-by: Wang, Yi <yi.a.wang@intel.com >
---------
Signed-off-by: Wang, Yi <yi.a.wang@intel.com >
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2026-03-17 08:52:15 +05:30
Dhruv Nair
07c5ba8eee
[Context Parallel] Add support for custom device mesh ( #13064 )
...
* add custom mesh support
* update
---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com >
2026-03-11 16:42:11 +05:30
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
