address comments

Author: AlexMaclean

llvm/lib/Transforms/IPO/OpenMPOpt.cpp

@@ -5922,17 +5922,16 @@ KernelSet llvm::omp::getDeviceKernels(Module &M) {
 
   DenseSet<const Function *> SeenKernels;
   auto ProcessKernel = [&](Function &KF) {
-    if (SeenKernels.contains(&KF))
-      return;
-    SeenKernels.insert(&KF);
-
-    // We are only interested in OpenMP target regions. Others, such as kernels
-    // generated by CUDA but linked together, are not interesting to this pass.
-    if (isOpenMPKernel(KF)) {
-      ++NumOpenMPTargetRegionKernels;
-      Kernels.insert(&KF);
-    } else
-      ++NumNonOpenMPTargetRegionKernels;
+    if (SeenKernels.insert(&KF).second) {
+      // We are only interested in OpenMP target regions. Others, such as
+      // kernels generated by CUDA but linked together, are not interesting to
+      // this pass.
+      if (isOpenMPKernel(KF)) {
+        ++NumOpenMPTargetRegionKernels;
+        Kernels.insert(&KF);
+      } else
+        ++NumNonOpenMPTargetRegionKernels;
+    }
   };
 
   if (NamedMDNode *MD = M.getNamedMetadata("nvvm.annotations"))