Skip to content

Commit

Permalink
tracking issue; styling
Browse files Browse the repository at this point in the history
  • Loading branch information
tarang-jain committed Apr 23, 2024
1 parent b61c1c3 commit 54061d0
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions cpp/test/neighbors/ann_cagra.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@
*/
#pragma once

#include "raft/neighbors/ivf_pq_types.hpp"
#undef RAFT_EXPLICIT_INSTANTIATE_ONLY // Search with filter instantiation

#include "../test_utils.cuh"
Expand All @@ -29,6 +28,7 @@
#include <raft/linalg/add.cuh>
#include <raft/neighbors/cagra.cuh>
#include <raft/neighbors/cagra_serialize.cuh>
#include <raft/neighbors/ivf_pq_types.hpp>
#include <raft/neighbors/sample_filter.cuh>
#include <raft/random/rng.cuh>
#include <raft/util/itertools.hpp>
Expand Down Expand Up @@ -246,7 +246,8 @@ class AnnCagraTest : public ::testing::TestWithParam<AnnCagraInputs> {
protected:
void testCagra()
{
// TODO (tarang-jain): remove when NN Descent index building support InnerProduct
// TODO (tarang-jain): remove when NN Descent index building support InnerProduct. Reference
// issue: https://github.com/rapidsai/raft/issues/2276
if (ps.metric == distance::InnerProduct && ps.build_algo == graph_build_algo::NN_DESCENT)
GTEST_SKIP();

Expand Down

0 comments on commit 54061d0

Please sign in to comment.