Merge jakirkham/use_sqrt_moc into jakirkham/cuda-12

rapidsai · Jul 14, 2023 · bb56ef7 · bb56ef7
2 parents eabbd77 + 3c40509
commit bb56ef7
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/python/cucim/src/cucim/skimage/measure/_colocalization.py b/python/cucim/src/cucim/skimage/measure/_colocalization.py
@@ -176,12 +176,6 @@ def manders_coloc_coeff(image0, image1_mask, mask=None):
  return cp.sum(image0 * image1_mask) / img_sum
 
 
-@cp.fuse()
-def _get_manders_overlap_coeff(image0, image1):
- denom = cp.linalg.norm(image0) * cp.linalg.norm(image1)
- return cp.sum(cp.multiply(image0, image1)) / denom
-
-
 def manders_overlap_coeff(image0, image1, mask=None):
  r"""Manders' overlap coefficient
 
@@ -261,7 +255,8 @@ def manders_overlap_coeff(image0, image1, mask=None):
  if image1.min() < 0:
  raise ValueError("image1 contains negative values")
 
- return _get_manders_overlap_coeff(image0, image1)
+ denom = cp.linalg.norm(image0) * cp.linalg.norm(image1)
+ return cp.vdot(image0, image1) / denom
 
 
 def intersection_coeff(image0_mask, image1_mask, mask=None):