From 5cbff67c7b8e0086d17466e1fecca1c3799c6430 Mon Sep 17 00:00:00 2001 From: Gergely Szilvasy Date: Fri, 26 Apr 2024 06:32:34 -0700 Subject: [PATCH] fix raft log spew Summary: Remove debugging log lines Reviewed By: mlomeli1 Differential Revision: D56626636 fbshipit-source-id: 2721b84e4e1359d1372df2b2c95cc668c6a75c3f --- faiss/gpu/GpuDistance.cu | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/faiss/gpu/GpuDistance.cu b/faiss/gpu/GpuDistance.cu index a235404b14..38a62f03bb 100644 --- a/faiss/gpu/GpuDistance.cu +++ b/faiss/gpu/GpuDistance.cu @@ -327,7 +327,7 @@ void bfKnn(GpuResourcesProvider* prov, const GpuDistanceParams& args) { int64_t, raft::col_major>> index_vec = {index.view()}; - RAFT_LOG_INFO("Invoking flat bfknn"); + brute_force::knn( handle, index_vec, @@ -354,10 +354,7 @@ void bfKnn(GpuResourcesProvider* prov, const GpuDistanceParams& args) { [] __device__(const float& a) { return powf(a, 2); }); } - RAFT_LOG_INFO("Done."); - handle.sync_stream(); - RAFT_LOG_INFO("All synced."); } else #else if (should_use_raft(args)) {