search

NVIDIA / spark-rapids-ml

Spark RAPIDS MLlib – accelerate Apache Spark MLlib with GPUs
https://nvidia.github.io/spark-rapids-ml/
Apache License 2.0
67 stars 30 forks source link

test `test_ivfflat_wide_matrix` failed in CI #697

Closed thirtiseven closed 3 months ago

thirtiseven commented 3 months ago

A test test_ivfflat_wide_matrix added by https://github.com/NVIDIA/spark-rapids-ml/pull/691 failed in CI:


[2024-07-25T05:29:20.685Z] =========================== short test summary info ============================

[2024-07-25T05:29:20.685Z] FAILED tests/test_approximate_nearest_neighbors.py::test_ivfflat_wide_matrix[float32-(4000, 3000)-combo0] - assert 0.05323904405853512 == 0.053238988191376575 ± 5.3e-08

[2024-07-25T05:29:20.685Z]   

[2024-07-25T05:29:20.685Z]   comparison failed

[2024-07-25T05:29:20.685Z]   Obtained: 0.05323904405853512

[2024-07-25T05:29:20.685Z]   Expected: 0.053238988191376575 ± 5.3e-08

[2024-07-25T05:29:20.685Z] FAILED tests/test_approximate_nearest_neighbors.py::test_ivfflat_wide_matrix[float32-(4000, 3000)-combo1] - assert 0.2333125547209036 == 0.23331224220430977 ± 2.3e-07

[2024-07-25T05:29:20.685Z]   

[2024-07-25T05:29:20.685Z]   comparison failed

[2024-07-25T05:29:20.685Z]   Obtained: 0.2333125547209036

[2024-07-25T05:29:20.685Z]   Expected: 0.23331224220430977 ± 2.3e-07

[2024-07-25T05:29:20.686Z] FAILED tests/test_approximate_nearest_neighbors.py::test_ivfflat_wide_matrix[float32-(4000, 3000)-combo2] - assert 0.9490436035531464 == 0.9490454080897135 ± 9.5e-07

[2024-07-25T05:29:20.686Z]   

[2024-07-25T05:29:20.686Z]   comparison failed

[2024-07-25T05:29:20.686Z]   Obtained: 0.9490436035531464

[2024-07-25T05:29:20.686Z]   Expected: 0.9490454080897135 ± 9.5e-07

[2024-07-25T05:29:20.686Z] ========== 3 failed, 732 passed, 9616 warnings in 8707.87s (2:25:07) ===========
eordentlich commented 3 months ago

Fixed by https://github.com/NVIDIA/spark-rapids-ml/pull/698