diff --git a/lib/Transforms/IPO/Attributor.cpp b/lib/Transforms/IPO/Attributor.cpp index 94ed7003c96..b9f943e4772 100644 --- a/lib/Transforms/IPO/Attributor.cpp +++ b/lib/Transforms/IPO/Attributor.cpp @@ -874,12 +874,17 @@ ChangeStatus AAReturnedValuesImpl::manifest(Attributor &A) { if (Function *F = dyn_cast(&AnchorValue)) { for (const Use &U : F->uses()) if (CallBase *CB = dyn_cast(U.getUser())) - if (CB->isCallee(&U)) - Changed = ReplaceCallSiteUsersWith(*CB, *RVC) | Changed; + if (CB->isCallee(&U)) { + Constant *RVCCast = + ConstantExpr::getTruncOrBitCast(RVC, CB->getType()); + Changed = ReplaceCallSiteUsersWith(*CB, *RVCCast) | Changed; + } } else { assert(isa(AnchorValue) && "Expcected a function or call base anchor!"); - Changed = ReplaceCallSiteUsersWith(cast(AnchorValue), *RVC); + Constant *RVCCast = + ConstantExpr::getTruncOrBitCast(RVC, AnchorValue.getType()); + Changed = ReplaceCallSiteUsersWith(cast(AnchorValue), *RVCCast); } if (Changed == ChangeStatus::CHANGED) STATS_DECLTRACK(UniqueConstantReturnValue, FunctionReturn, diff --git a/test/Transforms/FunctionAttrs/arg_returned.ll b/test/Transforms/FunctionAttrs/arg_returned.ll index 6e57475c579..95871ff7d34 100644 --- a/test/Transforms/FunctionAttrs/arg_returned.ll +++ b/test/Transforms/FunctionAttrs/arg_returned.ll @@ -827,6 +827,17 @@ define i32 @exact(i32* %a) { ret i32 %add3 } +@G = external global i8 +define i32* @ret_const() #0 { + %bc = bitcast i8* @G to i32* + ret i32* %bc +} +define i32* @use_const() #0 { + %c = call i32* @ret_const() + ; CHECK: ret i32* bitcast (i8* @G to i32*) + ret i32* %c +} + attributes #0 = { noinline nounwind uwtable } ; BOTH-NOT: attributes #