Changing digits

rapidsai · Oct 16, 2023 · d90c09e · d90c09e
1 parent 4ca9c0d
commit d90c09e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/include/raft/distance/detail/distance_ops/l2_exp.cuh b/cpp/include/raft/distance/detail/distance_ops/l2_exp.cuh
@@ -34,7 +34,7 @@ struct l2_exp_cutlass_op {
     // outVal could be negative due to numerical instability, especially when
     // calculating self distance.
     // clamp to 0 to avoid potential NaN in sqrt
-    outVal = outVal * (raft::abs(outVal) >= DataT(1e-12));
+    outVal = outVal * (raft::abs(outVal) >= DataT(1e-6));
     return sqrt ? raft::sqrt(outVal) : outVal;
   }