[Transforms] Fix Coroutine transform on non-default addressspaces

Summary:
This pass tries to get a pointer from the global. Some targets, like
AMDGPU, emit globals to a qualified address space. This would result in
an invalid cast. To fix this, we simply apply an addrspace cast, which
is a no-op if none exit.

Fixes: #102095

Author: jhuber6

llvm/lib/Transforms/Coroutines/CoroEarly.cpp

@@ -121,8 +121,8 @@ static void buildDebugInfoForNoopResumeDestroyFunc(Function *NoopFn) {
 }
 
 void Lowerer::lowerCoroNoop(IntrinsicInst *II) {
+  LLVMContext &C = Builder.getContext();
   if (!NoopCoro) {
-    LLVMContext &C = Builder.getContext();
     Module &M = *II->getModule();
 
     // Create a noop.frame struct type.
@@ -152,7 +152,8 @@ void Lowerer::lowerCoroNoop(IntrinsicInst *II) {
   }
 
   Builder.SetInsertPoint(II);
-  auto *NoopCoroVoidPtr = Builder.CreateBitCast(NoopCoro, Int8Ptr);
+  auto *NoopCoroVoidPtr =
+      Builder.CreateAddrSpaceCast(NoopCoro, PointerType::getUnqual(C));
   II->replaceAllUsesWith(NoopCoroVoidPtr);
   II->eraseFromParent();
 }

llvm/test/Transforms/Coroutines/coro-noop.ll

@@ -1,15 +1,18 @@
 ; Tests that CoroEarly pass correctly lowers coro.noop
-; RUN: opt < %s -S -passes=coro-early | FileCheck %s
+; RUN: opt -S -passes=coro-early < %s | FileCheck %s
+; RUN: opt -mtriple=amdgcn-- -S -passes=coro-early < %s | FileCheck %s --check-prefix=AMDGPU
 
 ; CHECK: %NoopCoro.Frame = type { ptr, ptr }
 ; CHECK: @NoopCoro.Frame.Const = private constant %NoopCoro.Frame { ptr @__NoopCoro_ResumeDestroy, ptr @__NoopCoro_ResumeDestroy }
+; AMDGPU: @NoopCoro.Frame.Const = private addrspace(1) constant %NoopCoro.Frame { ptr @__NoopCoro_ResumeDestroy, ptr @__NoopCoro_ResumeDestroy }
 
 
 ; CHECK-LABEL: @noop(
 define ptr @noop() {
 ; CHECK-NEXT: entry
 entry:
 ; CHECK-NEXT: ret ptr @NoopCoro.Frame.Const
+; AMDGPU: ret ptr addrspacecast (ptr addrspace(1) @NoopCoro.Frame.Const to ptr)
   %n = call ptr @llvm.coro.noop()
   ret ptr %n
 }