Change other LLVM modes to atomic increments

1 files changed, 5 insertions, 1 deletions

diff --git a/instrumentation/SanitizerCoverageLTO.so.cc b/instrumentation/SanitizerCoverageLTO.so.cc
index 6dd390e6..cd6b1939 100644
--- a/instrumentation/SanitizerCoverageLTO.so.cc
+++ b/instrumentation/SanitizerCoverageLTO.so.cc
@@ -1496,7 +1496,11 @@ void ModuleSanitizerCoverage::InjectCoverageAtBlock(Function &F, BasicBlock &BB,
     }
 
     /* Update bitmap */
+#if 1 /* Atomic */
+    IRB.CreateAtomicRMW(llvm::AtomicRMWInst::BinOp::Add, MapPtrIdx, One, 
+        llvm::AtomicOrdering::Monotonic);
 
+#else
     LoadInst *Counter = IRB.CreateLoad(MapPtrIdx);
     Counter->setMetadata(Mo->getMDKindID("nosanitize"), MDNode::get(*Ct, None));
 
@@ -1512,7 +1516,7 @@ void ModuleSanitizerCoverage::InjectCoverageAtBlock(Function &F, BasicBlock &BB,
 
     IRB.CreateStore(Incr, MapPtrIdx)
         ->setMetadata(Mo->getMDKindID("nosanitize"), MDNode::get(*Ct, None));
-
+#endif
     // done :)
 
     inst++;