Move check to TargetTransformInfo

Author: trenouf

llvm/lib/Analysis/TargetTransformInfo.cpp

@@ -288,6 +288,10 @@ bool TargetTransformInfo::hasBranchDivergence(const Function *F) const {
 }
 
 bool TargetTransformInfo::isSourceOfDivergence(const Value *V) const {
+  if (auto Call = dyn_cast<CallBase>(V)) {
+    if (Call->hasFnAttr(Attribute::NoDivergenceSource))
+      return false;
+  }
   return TTIImpl->isSourceOfDivergence(V);
 }
 

llvm/lib/Analysis/UniformityAnalysis.cpp

@@ -32,10 +32,7 @@ bool llvm::GenericUniformityAnalysisImpl<SSAContext>::markDefsDivergent(
 
 template <> void llvm::GenericUniformityAnalysisImpl<SSAContext>::initialize() {
   for (auto &I : instructions(F)) {
-    bool NoDivergenceSource = false;
-    if (auto Call = dyn_cast<CallBase>(&I))
-      NoDivergenceSource = Call->hasFnAttr(Attribute::NoDivergenceSource);
-    if (!NoDivergenceSource && TTI->isSourceOfDivergence(&I))
+    if (TTI->isSourceOfDivergence(&I))
       markDivergent(I);
     else if (TTI->isAlwaysUniform(&I))
       addUniformOverride(I);