[Coro] RetconOnceDynamic: Annotate alloc param.

Mark the allocator parameter passed to the allocation functions
swiftcoro.

Author: nate-chandler

llvm/lib/Transforms/Coroutines/Coroutines.cpp

@@ -559,6 +559,7 @@ Value *coro::Shape::emitAlloc(IRBuilder<> &Builder, Value *Size,
         Args.push_back(TypeId);
     }
     auto *Call = Builder.CreateCall(Alloc, Args);
+    Call->addParamAttr(1, Attribute::SwiftCoro);
     propagateCallAttrsFromCallee(Call, Alloc);
     addCallToCallGraph(CG, Call, Alloc);
     return Call;
@@ -598,6 +599,7 @@ void coro::Shape::emitDealloc(IRBuilder<> &Builder, Value *Ptr,
         Ptr, Dealloc->getFunctionType()->getParamType(allocationParamIndex));
     Args.push_back(Ptr);
     auto *Call = Builder.CreateCall(Dealloc, Args);
+    Call->addParamAttr(1, Attribute::SwiftCoro);
     propagateCallAttrsFromCallee(Call, Dealloc);
     addCallToCallGraph(CG, Call, Dealloc);
     return;

llvm/test/Transforms/Coroutines/coro-retcon-once-dynamic.ll

@@ -174,7 +174,7 @@ declare i32 @getSize()
 ; CHECK:         %size = call i32 @getSize()
 ; CHECK:         call swiftcorocc ptr @allocate(
 ; CHECK-SAME:      ptr %frame,
-; CHECK-SAME:      ptr %allocator,
+; CHECK-SAME:      ptr swiftcoro %allocator,
 ; CHECK-SAME:      i32 %size
 ; CHECK-SAME:    )
 define { ptr, ptr } @allocating_something(ptr noalias %frame, ptr swiftcoro %allocator) {