[SP] add SP deny list instead of allow #7887
kashif
wants to merge
18 commits into
deepspeedai:master
from
kashif:sp_attn_deny
add SP deny list instead of allow
aec2c90b
tohtana
commented
on 2026-03-06
hub-kernel lazy registration before validation and tests
49e0310e
Merge branch 'master' into sp_attn_deny
2f8e77c7
Merge branch 'master' into sp_attn_deny
ce69dc0e
Merge branch 'master' into sp_attn_deny
952c3ae8
Merge branch 'master' into sp_attn_deny
6e3f2cb3
position_ids generation and flex_attention BlockMask
874ec62f
refactor
7d0a136b
update comments
58681356
do not check for has_packed_samples
b0e05f0a
raise error instead of warning
89058fc8
cache BlockMask
463cb302
Merge branch 'master' into sp_attn_deny
5cf053df
Merge branch 'master' into sp_attn_deny
e2504921
flex_attention: lazy imports to fix pickle issue and add test
8f53e6cc
fix flex_attention BlockMask guard and add position_ids warning in fo…
d4c89533
flex_attention test: use local tiny model, fix _compile gradient issue
12bbb9ce
don't compile create_block_mask inside forward()
43862532
Login to write a write a comment.
Login via GitHub