Type *ResolverFTy =
GlobalIFunc::getResolverFunctionType(GI->getValueType());
// Transparently fix up the type for compatibility with older bitcode
- GI->setResolver(
- ConstantExpr::getBitCast(C, ResolverFTy->getPointerTo()));
+ GI->setResolver(ConstantExpr::getBitCast(
+ C, ResolverFTy->getPointerTo(GI->getAddressSpace())));
} else {
return error("Expected an alias or an ifunc");
}
const Type *ResolverTy = GI.getResolver()->getType();
const Type *ResolverFuncTy =
GlobalIFunc::getResolverFunctionType(GI.getValueType());
- Check(ResolverTy == ResolverFuncTy->getPointerTo(),
+ Check(ResolverTy == ResolverFuncTy->getPointerTo(GI.getAddressSpace()),
"IFunc resolver has incorrect type", &GI);
}
--- /dev/null
+; RUN: llvm-as < %s | llvm-dis | FileCheck %s
+
+target datalayout = "P1"
+
+; CHECK: @ifunc_as0 = ifunc void (), ptr @resolver_as0
+@ifunc_as0 = ifunc void (), ptr @resolver_as0
+
+; CHECK: @ifunc_as1 = ifunc void (), ptr addrspace(1) @resolver_as1
+@ifunc_as1 = ifunc void (), ptr addrspace(1) @resolver_as1
+
+; CHECK: define ptr @resolver_as0() addrspace(0) {
+define ptr @resolver_as0() addrspace(0) {
+ ret ptr null
+}
+
+; CHECK: define ptr @resolver_as1() addrspace(1) {
+define ptr @resolver_as1() addrspace(1) {
+ ret ptr null
+}
+
+; CHECK: define void @call_ifunc_as0() addrspace(1) {
+; CHECK-NEXT: call addrspace(0) void @ifunc_as0()
+define void @call_ifunc_as0() addrspace(1) {
+ call addrspace(0) void @ifunc_as0()
+ ret void
+}
+
+; CHECK: define void @call_ifunc_as1() addrspace(1) {
+; CHECK-NEXT: call addrspace(1) void @ifunc_as1()
+define void @call_ifunc_as1() addrspace(1) {
+ call addrspace(1) void @ifunc_as1()
+ ret void
+}