Skip to content

Commit 9f0691f

Browse files
committed
uncomment test
Signed-off-by: Jennifer Chen <[email protected]>
1 parent 10e3e2b commit 9f0691f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/_test_utils/torch_quantization/quantize_common.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -172,12 +172,12 @@ def forward_loop(model):
172172
dist.ReduceOp.AVG,
173173
group=tp_group,
174174
)
175-
# TODO fc2 assert is failing
176-
"""
177175
_reduce_quantizer_attr(
178-
model.fc2.awq_lite, "act_scale", dist.ReduceOp.AVG, group=tp_group,
176+
model.fc2.awq_lite,
177+
"act_scale",
178+
dist.ReduceOp.AVG,
179+
group=tp_group,
179180
)
180-
"""
181181

182182
dist.destroy_process_group()
183183

0 commit comments

Comments
 (0)