From 9c34f87ab93a83c3f401c8f72371dc9f47407490 Mon Sep 17 00:00:00 2001 From: Ramil Bakhshyiev Date: Fri, 6 Sep 2024 14:09:10 -0700 Subject: [PATCH] Re-enable Query_L2_MMCodeDistance and Query_IP_MMCodeDistance tests for ROCm (#3838) Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/3838 This fixes 3787 now that we do not install CUDA for ROCm builds. Reviewed By: mengdilin Differential Revision: D62283662 fbshipit-source-id: e5c736296b1c6d7b9a6b9f60161ffe3b5cb1c699 --- faiss/gpu/test/TestGpuIndexIVFPQ.cpp | 3 --- 1 file changed, 3 deletions(-) diff --git a/faiss/gpu/test/TestGpuIndexIVFPQ.cpp b/faiss/gpu/test/TestGpuIndexIVFPQ.cpp index 73de101946..9cc52bc788 100644 --- a/faiss/gpu/test/TestGpuIndexIVFPQ.cpp +++ b/faiss/gpu/test/TestGpuIndexIVFPQ.cpp @@ -307,8 +307,6 @@ 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); } @@ -316,7 +314,6 @@ TEST(TestGpuIndexIVFPQ, Query_L2_MMCodeDistance) { TEST(TestGpuIndexIVFPQ, Query_IP_MMCodeDistance) { testMMCodeDistance(faiss::MetricType::METRIC_INNER_PRODUCT); } -#endif // USE_AMD_ROCM TEST(TestGpuIndexIVFPQ, Float16Coarse) { Options opt;