Skip to content

Commit fa33949

Browse files
committed
update correct slices
1 parent a98a184 commit fa33949

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/quantization/torchao/test_torchao.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -772,7 +772,7 @@ def get_dummy_inputs(self, device: torch.device, seed: int = 0):
772772

773773
def test_transformer_int8wo(self):
774774
# fmt: off
775-
expected_slice = np.array([0.0505, 0.0742, 0.1367, 0.0429, 0.0585, 0.1386, 0.0585, 0.0703, 0.1367, 0.0566, 0.0703, 0.1464, 0.0546, 0.0703, 0.1425, 0.0546, 0.3535, 0.7578, 0.5000, 0.4062, 0.7656, 0.5117, 0.4121, 0.7656, 0.5117, 0.3984, 0.7578, 0.5234, 0.4023, 0.7382, 0.5390, 0.4570])
775+
expected_slice = np.array([0.0566, 0.0781, 0.1426, 0.0488, 0.0684, 0.1504, 0.0625, 0.0781, 0.1445, 0.0625, 0.0781, 0.1562, 0.0547, 0.0723, 0.1484, 0.0566, 0.5703, 0.8867, 0.7266, 0.5742, 0.875, 0.7148, 0.5586, 0.875, 0.7148, 0.5547, 0.8633, 0.7109, 0.5469, 0.8398, 0.6992, 0.5703])
776776
# fmt: on
777777

778778
# This is just for convenience, so that we can modify it at one place for custom environments and locally testing

0 commit comments

Comments
 (0)