aboutsummaryrefslogtreecommitdiff
path: root/llvm/lib/Target/AMDGPU/AMDGPUPropagateAttributes.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'llvm/lib/Target/AMDGPU/AMDGPUPropagateAttributes.cpp')
-rw-r--r--llvm/lib/Target/AMDGPU/AMDGPUPropagateAttributes.cpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/llvm/lib/Target/AMDGPU/AMDGPUPropagateAttributes.cpp b/llvm/lib/Target/AMDGPU/AMDGPUPropagateAttributes.cpp
index cd71c7a16c73..0e4c26170a8f 100644
--- a/llvm/lib/Target/AMDGPU/AMDGPUPropagateAttributes.cpp
+++ b/llvm/lib/Target/AMDGPU/AMDGPUPropagateAttributes.cpp
@@ -249,7 +249,11 @@ bool AMDGPUPropagateAttributes::process() {
if (!I)
continue;
CallBase *CI = dyn_cast<CallBase>(I);
- if (!CI)
+ // Only propagate attributes if F is the called function. Specifically,
+ // do not propagate attributes if F is passed as an argument.
+ // FIXME: handle bitcasted callee, e.g.
+ // %retval = call i8* bitcast (i32* ()* @f to i8* ()*)()
+ if (!CI || CI->getCalledOperand() != &F)
continue;
Function *Caller = CI->getCaller();
if (!Caller || !Visited.insert(CI).second)