Skip to content

Commit 06ece86

Browse files
committed
remove prints
Signed-off-by: Kyle Sayers <[email protected]>
1 parent 25e54a5 commit 06ece86

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

tests/llmcompressor/modifiers/calibration/test_observers.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -260,8 +260,6 @@ def test_static_weight_quantization(
260260
input = torch.eye(input_size, dtype=torch.bfloat16)
261261
output = linear(input)
262262

263-
print(output.T)
264-
print(torch.nn.functional.mse_loss(output.T, linear.weight))
265263
assert torch.allclose(output.T, exp_quant.to(output.dtype))
266264
assert torch.nn.functional.mse_loss(output.T, linear.weight) <= exp_loss
267265

@@ -376,8 +374,5 @@ def test_static_activation_quantization(
376374
assert torch.equal(observer.max_val[key], exp_max_val[key])
377375

378376
# check forward pass
379-
print(args.strategy)
380-
print(output)
381-
print(torch.nn.functional.mse_loss(output, input))
382377
assert torch.allclose(output, exp_quant.to(output.dtype))
383378
assert torch.nn.functional.mse_loss(output, input) <= exp_loss

0 commit comments

Comments
 (0)