[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,9 @@ Value *coro::Shape::emitAlloc(IRBuilder<> &Builder, Value *Size,
         Args.push_back(TypeId);
     }
     auto *Call = Builder.CreateCall(Alloc, Args);
+    if (ABI == coro::ABI::RetconOnceDynamic) {
+      Call->addParamAttr(1, Attribute::SwiftCoro);
+    }
     propagateCallAttrsFromCallee(Call, Alloc);
     addCallToCallGraph(CG, Call, Alloc);
     return Call;
@@ -598,6 +601,9 @@ void coro::Shape::emitDealloc(IRBuilder<> &Builder, Value *Ptr,
         Ptr, Dealloc->getFunctionType()->getParamType(allocationParamIndex));
     Args.push_back(Ptr);
     auto *Call = Builder.CreateCall(Dealloc, Args);
+    if (ABI == coro::ABI::RetconOnceDynamic) {
+      Call->addParamAttr(1, Attribute::SwiftCoro);
+    }
     propagateCallAttrsFromCallee(Call, Dealloc);
     addCallToCallGraph(CG, Call, Dealloc);
     return;

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

@@ -187,12 +187,12 @@ declare { ptr, ptr } @allocating_something_else(ptr noalias %frame, ptr swiftcor
 ; 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:    )
 ; CHECK:         call swiftcorocc ptr @allocate_frame(
 ; CHECK-SAME:      ptr %frame,
-; CHECK-SAME:      ptr %allocator,
+; CHECK-SAME:      ptr swiftcoro %allocator,
 ; CHECK-SAME:      i32
 ; CHECK-SAME:    )
 ; CHECK-LABEL: @allocating_something.resume.0(
@@ -202,12 +202,12 @@ declare { ptr, ptr } @allocating_something_else(ptr noalias %frame, ptr swiftcor
 ; CHECK-SAME:  {
 ; CHECK:         call void @deallocate_frame(
 ; CHECK-SAME:      ptr %0,
-; CHECK-SAME:      ptr %1,
+; CHECK-SAME:      ptr swiftcoro %1,
 ; CHECK-SAME:      ptr
 ; CHECK-SAME:    )
 ; CHECK:         call void @deallocate(
 ; CHECK-SAME:      ptr %0,
-; CHECK-SAME:      ptr %1,
+; CHECK-SAME:      ptr swiftcoro %1,
 ; CHECK-SAME:      ptr
 ; CHECK-SAME:    )
 @allocating_something_cfp = constant <{ i32, i32 }>