Index: llvm/lib/Linker/IRMover.cpp =================================================================== --- llvm/lib/Linker/IRMover.cpp +++ llvm/lib/Linker/IRMover.cpp @@ -628,7 +628,7 @@ SGVar->isConstant(), GlobalValue::ExternalLinkage, /*init*/ nullptr, SGVar->getName(), /*insertbefore*/ nullptr, SGVar->getThreadLocalMode(), - SGVar->getType()->getAddressSpace()); + SGVar->getAddressSpace()); NewDGV->setAlignment(MaybeAlign(SGVar->getAlignment())); NewDGV->copyAttributesFrom(SGVar); return NewDGV; @@ -654,9 +654,9 @@ Function *IRLinker::copyFunctionProto(const Function *SF) { // If there is no linkage to be performed or we are linking from the source, // bring SF over. - auto *F = - Function::Create(TypeMap.get(SF->getFunctionType()), - GlobalValue::ExternalLinkage, SF->getName(), &DstM); + auto *F = Function::Create(TypeMap.get(SF->getFunctionType()), + GlobalValue::ExternalLinkage, + SF->getAddressSpace(), SF->getName(), &DstM); F->copyAttributesFrom(SF); F->setAttributes(mapAttributeTypes(F->getContext(), F->getAttributes())); return F; @@ -671,11 +671,11 @@ auto *Ty = TypeMap.get(SGIS->getValueType()); GlobalIndirectSymbol *GIS; if (isa(SGIS)) - GIS = GlobalAlias::create(Ty, SGIS->getType()->getPointerAddressSpace(), + GIS = GlobalAlias::create(Ty, SGIS->getAddressSpace(), GlobalValue::ExternalLinkage, SGIS->getName(), &DstM); else - GIS = GlobalIFunc::create(Ty, SGIS->getType()->getPointerAddressSpace(), + GIS = GlobalIFunc::create(Ty, SGIS->getAddressSpace(), GlobalValue::ExternalLinkage, SGIS->getName(), nullptr, &DstM); GIS->copyAttributesFrom(SGIS); @@ -695,14 +695,15 @@ else if (SGV->getValueType()->isFunctionTy()) NewGV = Function::Create(cast(TypeMap.get(SGV->getValueType())), - GlobalValue::ExternalLinkage, SGV->getName(), &DstM); + GlobalValue::ExternalLinkage, SGV->getAddressSpace(), + SGV->getName(), &DstM); else - NewGV = new GlobalVariable( - DstM, TypeMap.get(SGV->getValueType()), - /*isConstant*/ false, GlobalValue::ExternalLinkage, - /*init*/ nullptr, SGV->getName(), - /*insertbefore*/ nullptr, SGV->getThreadLocalMode(), - SGV->getType()->getAddressSpace()); + NewGV = + new GlobalVariable(DstM, TypeMap.get(SGV->getValueType()), + /*isConstant*/ false, GlobalValue::ExternalLinkage, + /*init*/ nullptr, SGV->getName(), + /*insertbefore*/ nullptr, + SGV->getThreadLocalMode(), SGV->getAddressSpace()); } if (ForDefinition) @@ -918,7 +919,7 @@ GlobalVariable *NG = new GlobalVariable( DstM, NewType, SrcGV->isConstant(), SrcGV->getLinkage(), /*init*/ nullptr, /*name*/ "", DstGV, SrcGV->getThreadLocalMode(), - SrcGV->getType()->getAddressSpace()); + SrcGV->getAddressSpace()); NG->copyAttributesFrom(SrcGV); forceRenaming(NG, SrcGV->getName()); Index: llvm/test/Linker/addrspace.ll =================================================================== --- /dev/null +++ llvm/test/Linker/addrspace.ll @@ -0,0 +1,9 @@ +; RUN: llvm-link %s -S | FileCheck %s + +@G = addrspace(2) global i32 256 +; CHECK: @G = addrspace(2) global i32 + +define void @foo() addrspace(3) { +; CHECK: define void @foo() addrspace(3) + ret void +}