Function test_edge_case_large_values

tests/test_compress.py:129–133

()

def test_edge_case_large_values():
    engine = TurboQuantEngine(head_dim=128, total_bits=3, device="cpu")
    K = torch.randn(4, 128).half() * 100
    compressed = engine.compress_keys_pytorch(K)
    assert not torch.isnan(compressed["k_mse"]).any()