[BUG] ivf_pq::search mr argument ignored

[tfeher]

Describe the bug
The IVF-PQ method has an argument to specify memor allocator used for temporary buffers. https://github.com/rapidsai/raft/blob/branch-23.08/cpp/include/raft/neighbors/ivf_pq-inl.cuh#L407

This allocator is ignored for some of the internal allocations (See #1557), and that decreases the performance.

Expected behavior

We plan to deprecate explicit MR argument for IVF-PQ search. Ideally the workspace allocator shall control the temporary allocations.

Workaround

rmm::mr::pool_memory_resource<rmm::mr::device_memory_resource> mr(rmm::mr::get_current_device_resource(), 1024 * 1024 * 1024ull);
rmm::mr::set_current_device_resource(&mr);
ivf_pq::search(res, search_params, index, queries, neighbors_view, distances_view);