Skip to content

Commit

Permalink
Re-enable Query_L2_MMCodeDistance and Query_IP_MMCodeDistance tests f…
Browse files Browse the repository at this point in the history
…or ROCm (#3838)

Summary:
Pull Request resolved: #3838

This fixes 3787 now that we do not install CUDA for ROCm builds.

Reviewed By: mengdilin

Differential Revision: D62283662

fbshipit-source-id: e5c736296b1c6d7b9a6b9f60161ffe3b5cb1c699
  • Loading branch information
ramilbakhshyiev authored and facebook-github-bot committed Sep 6, 2024
1 parent 753833c commit 3f41161
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions faiss/gpu/test/TestGpuIndexIVFPQ.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -307,16 +307,13 @@ void testMMCodeDistance(faiss::MetricType mt) {
}
}

// FIXME: https://github.com/facebookresearch/faiss/issues/3787
#ifndef USE_AMD_ROCM
TEST(TestGpuIndexIVFPQ, Query_L2_MMCodeDistance) {
testMMCodeDistance(faiss::MetricType::METRIC_L2);
}

TEST(TestGpuIndexIVFPQ, Query_IP_MMCodeDistance) {
testMMCodeDistance(faiss::MetricType::METRIC_INNER_PRODUCT);
}
#endif // USE_AMD_ROCM

TEST(TestGpuIndexIVFPQ, Float16Coarse) {
Options opt;
Expand Down

0 comments on commit 3f41161

Please sign in to comment.