Skip to content

Commit e7aed34

Browse files
committed
fix: removed perCh from stddev var names
Signed-off-by: Brandon Groth <[email protected]>
1 parent 561f144 commit e7aed34

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/models/test_save_aiu.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,11 +85,11 @@ def test_large_outlier_bert(
8585
# Loaded model w/ recomputed SAWB should have widened channel quantization stdev
8686
for k, v in state_dict.items():
8787
if k.endswith(".weight") and any(n in k for n in bert_linear_names):
88-
perCh_stdev_model = layer2stdev.get(k)
89-
perCh_stdev_loaded = v.to(torch.float32).std(dim=stddev_dim)
88+
stddev_model = layer2stdev.get(k)
89+
stddev_loaded = v.to(torch.float32).std(dim=stddev_dim)
9090

9191
# SAWB stddev should be at least as good as Qmax stddev w/ outlier
92-
assert torch.all(perCh_stdev_loaded >= perCh_stdev_model)
92+
assert torch.all(stddev_loaded >= stddev_model)
9393

9494

9595
def test_clip_vals_zero_bert(

0 commit comments

Comments
 (0)