feat(dpmodel): NeighborGraph support for dpa1 tebd_input_mode="strip" - #5747

Merged

OutisLi merged 7 commits into

deepmodeling:masterfrom

wanghan-iapcm:feat-graph-strip-tebd

Jul 8, 2026

Commits on Jul 7, 2026

feat(dpmodel): graph-native tebd_input_mode='strip' kernel for dpa1
Han Wang
committed
d21fd40
feat(dpmodel): route dpa1/se_atten_v2 strip through the graph lower (gate keeps exclude_types/compression dense)
Han Wang
committed
c070737
test(pt_expt): cover the strip graph kernel in dpa1 make_fx export tests
Han Wang
committed
4a68c86
docs(dpmodel): correct _call_dense docstring for strip/attention now graph-eligible
Han Wang
committed
c4b4162

Commits on Jul 8, 2026

test(dpmodel): add NaN guard before strip/attention parity assert_allclose
Han Wang
committed
e935668
test(pt_expt): exclude se_atten_v2 from the None-vs-explicit nlist dispatch test
Han Wang
committed
66089f1
test(pt_expt): assert bounded closeness instead of excluding se_atten_v2
Han Wang
committed
b774455