Skip to content

Commit a1061af

Browse files
author
Han Wang
committed
test(pt_expt): pin smooth off in neighbor-list dpa1 fixture (route parity)
neighbor_list=None now takes the carry-all graph default for eligible attention models; explicit World-1 builders take the legacy dense route. With smooth attention the two routes differ by design (PR-D), so the route-equivalence tests pin smooth_type_embedding=False.
1 parent 480cd61 commit a1061af

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

source/tests/pt_expt/utils/test_neighbor_list.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,11 @@
113113
"temperature": 1.0,
114114
"set_davg_zero": True,
115115
"type_one_side": True,
116+
# smooth attention diverges between the carry-all graph default
117+
# (neighbor_list=None) and the explicit World-1 builders by design
118+
# (NeighborGraph PR-D: dense keeps sel-padding in the attention
119+
# softmax denominator); pin smooth off so all routes are exact.
120+
"smooth_type_embedding": False,
116121
"seed": 1,
117122
},
118123
"fitting_net": {"neuron": [8, 8], "resnet_dt": True, "seed": 1},

0 commit comments

Comments
 (0)