cuvs-bench-cpu: avoid 'mkl' dependency on aarch64

[jameslamb]

Fixes rapidsai/docker#739

#260 introduced a runtime dependency on mkl for the cuvs-bench-cpu conda package. There are not aarch64 packages for mkl on conda-forge, so this makes cuvs-bench-cpu impossible to install on aarch64.

This fixes that, by applying the same "only add on x86_64" guard used for mkl everywhere else in this project, e.g. like this:

cuvs/conda/recipes/cuvs-bench-cpu/meta.yaml

Line 51 in 89b0349

- mkl-devel=2023 # [linux64]

It also proposes adding import tests to the cuvs-bench-cpu conda recipe, so issues like this can be caught in CI in the future.

Notes for Reviewers

I searched for references like this

git grep -i mkl

How I tested this

Saw lines like this in conda-python-build logs:

BUILD START: ['cuvs-bench-cpu-25.04.00a96-py312_250305_g94340bc_96.conda']
...
TEST START: /tmp/conda-bld-output/linux-aarch64/cuvs-bench-cpu-25.04.00a96-py312_250305_g94340bc_96.conda
...
import: 'cuvs_bench'
import: 'cuvs_bench.generate_groundtruth'
import: 'cuvs_bench.get_dataset'
import: 'cuvs_bench.plot'
import: 'cuvs_bench.run'
import: 'cuvs_bench.run.data_export'
import: 'cuvs_bench.split_groundtruth'
import: 'cuvs_bench'
import: 'cuvs_bench.generate_groundtruth'
import: 'cuvs_bench.get_dataset'
import: 'cuvs_bench.plot'
import: 'cuvs_bench.run'
import: 'cuvs_bench.run.data_export'
import: 'cuvs_bench.split_groundtruth'
...
TEST END: /tmp/conda-bld-output/linux-aarch64/cuvs-bench-cpu-25.04.00a96-py312_250305_g94340bc_96.conda

(build link)

[jameslamb]

I think these tests indirectly caught a mistake in the docs! When I set up the import tests, I did something like this to find which modules were referenced in documentation:

git grep -E 'cuvs_bench\.'

That showed me these references to cuvs_bench.data_export.

However, the import tests failed:

Traceback (most recent call last):
  File "/opt/conda/conda-bld/test_tmp/run_test.py", line 5, in <module>
    import cuvs_bench.data_export
ModuleNotFoundError: No module named 'cuvs_bench.data_export'
...
CondaBuildUserError: TESTS FAILED: cuvs-bench-cpu-25.04.00a95-py310_250305_g0208914_95.conda
...
import: 'cuvs_bench'
import: 'cuvs_bench.data_export'

(build link)

And it really does look to me like this needs to be cuvs_bench.run.data_export... there's no top-level __init__.py doing remapping in https://github.com/rapidsai/cuvs/tree/branch-25.04/python/cuvs_bench/cuvs_bench, and data_export is down in cuvs_bench/run/: https://github.com/rapidsai/cuvs/blob/89b03493b487910d2125fde6680590adde8e2a95/python/cuvs_bench/cuvs_bench/run/data_export.py

[divyegala]

It is now an option that is true by default. However, for this PR please just update the line to cuvs_bench.run --data-export

[jameslamb]

Ah ok, sure! Thanks for the explanation.

Just pushed 669e169 with that change.

[gforsyth]

Nice catch on the import, @jameslamb ! Yay tests!

[divyegala]

Approved pending doc update

[divyegala]

It is now an option that is true by default. However, for this PR please just update the line to cuvs_bench.run --data-export

[jameslamb]

Saw one test failure for wheels:

>       assert recall > 0.7
E       AssertionError

/pyenv/versions/3.12.9/lib/python3.12/site-packages/cuvs/tests/ann_utils.py:125: AssertionError
=========================== short test summary info ============================
FAILED python/cuvs/cuvs/tests/test_ivf_flat.py::test_filtered_ivf_flat[0.5] - AssertionError

(build link)

Strongly suspect that that's a flaky test. I just triggered a re-run.

[jameslamb]

/merge