Skip to content

Commit a33077b

Browse files
committed
gpu rawdata return false, reduce cpu usage with gpu index
Signed-off-by: yusheng.ma <yusheng.ma@zilliz.com>
1 parent 5699751 commit a33077b

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

src/index/gpu_raft/gpu_raft.h

+2-6
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141

4242
namespace knowhere {
4343

44-
auto static constexpr cuda_concurrent_size = std::uint32_t{32};
44+
auto static constexpr cuda_concurrent_size = std::uint32_t{16};
4545

4646
template <raft_proto::raft_index_kind K>
4747
struct KnowhereConfigType {};
@@ -150,11 +150,7 @@ struct GpuRaftIndexNode : public IndexNode {
150150

151151
bool
152152
HasRawData(const std::string& metric_type) const override {
153-
if constexpr (index_kind == raft_proto::raft_index_kind::ivf_flat) {
154-
return !IsMetricType(metric_type, metric::COSINE);
155-
} else {
156-
return false;
157-
}
153+
return false;
158154
}
159155

160156
Status

0 commit comments

Comments
 (0)