Skip to content

Commit c2a41f1

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 274c0a1 commit c2a41f1

1 file changed

Lines changed: 1 addition & 4 deletions

File tree

tests/pytorch/attention/test_linear_mxfp8_attention.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -268,10 +268,7 @@ def test_accuracy(self, batch_size: int, seq_len: int) -> None:
268268
assert out_mxfp8.float().abs().max() > 0, "MXFP8 output is all zeros"
269269

270270
max_abs_qkv, rms_qkv = _compute_errors(qkv_bf16, qkv_mxfp8)
271-
print(
272-
f"\n[QKV] b={batch_size} s={seq_len}: "
273-
f"max_abs={max_abs_qkv:.6f} rms={rms_qkv:.6f}"
274-
)
271+
print(f"\n[QKV] b={batch_size} s={seq_len}: max_abs={max_abs_qkv:.6f} rms={rms_qkv:.6f}")
275272
torch.testing.assert_close(
276273
qkv_mxfp8,
277274
qkv_bf16,

0 commit comments

Comments
 (0)