Skip to content

Commit 7f41b35

Browse files
committed
Edit comments
1 parent b117006 commit 7f41b35

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

cpp/src/neighbors/detail/cagra/cagra_build.cuh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2199,8 +2199,8 @@ index<T, IdxT> build(
21992199
}
22002200
}
22012201

2202-
// Guard against potential FP16 distance overflow for large-magnitude datasets. Applies to any
2203-
// IVF-PQ graph build (auto-selected above or explicitly set in params) -> fall back to FP32.
2202+
// Predict potential FP16 distance overflow for large-magnitude (e.g. unnormalized) datasets
2203+
// -> fall back to FP32.
22042204
if (auto* pq = std::get_if<cagra::graph_build_params::ivf_pq_params>(&knn_build_params)) {
22052205
const bool using_fp16_distance = pq->search_params.internal_distance_dtype == CUDA_R_16F ||
22062206
pq->search_params.coarse_search_dtype == CUDA_R_16F;

0 commit comments

Comments
 (0)