Fix code distance C++ unit tests for ROCm on MI250 #3787
Description
Summary
TestGpuIndexIVFPQ.Query_L2_MMCodeDistance
TestGpuIndexIVFPQ.Query_IP_MMCodeDistance
Platform
ROCm
Running on:
- CPU
- GPU
Interface:
- C++
- Python
Reproduction instructions
GitHub CI