[dtensor][fix] fix _scaled_dot_product_flash_attention sharding#148125
Closed
XilunWu wants to merge 2 commits intogh/XilunWu/120/basefrom
Closed
[dtensor][fix] fix _scaled_dot_product_flash_attention sharding#148125XilunWu wants to merge 2 commits intogh/XilunWu/120/basefrom
XilunWu wants to merge 2 commits intogh/XilunWu/120/basefrom