Use CallBase::hasFnAttr

Author: trenouf

llvm/lib/Analysis/UniformityAnalysis.cpp

@@ -33,12 +33,8 @@ bool llvm::GenericUniformityAnalysisImpl<SSAContext>::markDefsDivergent(
 template <> void llvm::GenericUniformityAnalysisImpl<SSAContext>::initialize() {
   for (auto &I : instructions(F)) {
     bool NoDivergenceSource = false;
-    if (auto Call = dyn_cast<CallInst>(&I)) {
-      if (Function *Callee = Call->getCalledFunction()) {
-        if (Callee->hasFnAttribute(Attribute::NoDivergenceSource))
-          NoDivergenceSource = true;
-      }
-    }
+    if (auto Call = dyn_cast<CallBase>(&I))
+      NoDivergenceSource = Call->hasFnAttr(Attribute::NoDivergenceSource);
     if (!NoDivergenceSource && TTI->isSourceOfDivergence(&I))
       markDivergent(I);
     else if (TTI->isAlwaysUniform(&I))