[Bugfix] Fix broken ViT attention selection for Blackwell device (#30731)
Signed-off-by:Isotr0py <mozf@mail2.sysu.edu.cn> (cherry picked from commit e94384bb)
Showing
Please register or sign in to comment
Signed-off-by:Isotr0py <mozf@mail2.sysu.edu.cn> (cherry picked from commit e94384bb)