fix comments

rapidsai · Jun 17, 2024 · 4b64c37 · 4b64c37
1 parent 7b0e44b
commit 4b64c37
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cpp/include/raft/distance/detail/distance_ops/dice.cuh b/cpp/include/raft/distance/detail/distance_ops/dice.cuh
@@ -36,7 +36,7 @@ struct dice_cutlass_op {
  *
  * It computes the following equation:
  *
- * d(x, y) = 1 - 2*(x ⋅ y) / ( ||x||_1 + ||y||_1 )
+ * d(x, y) = 1 - 2*(x ⋅ y) / ( Σ(x) + Σ(y) )
  */
 template <typename DataType, typename AccType, typename IdxType>
 struct dice_distance_op {

diff --git a/cpp/test/distance/distance_base.cuh b/cpp/test/distance/distance_base.cuh
@@ -120,7 +120,7 @@ RAFT_KERNEL naiveDiceDistanceKernel(
 
   int outidx = isRowMajor ? midx * n + nidx : midx + m * nidx;
 
-  // Use 1.0 - (dice similarity) to calc the distance
+  // Use 1.0 - (dice dissimilarity) to calc the distance
   dist[outidx] = (DataType)1.0 - (2 * acc_ab / ((acc_a) + (acc_b)));
 }