diff --git a/polly/lib/CodeGen/IslExprBuilder.cpp b/polly/lib/CodeGen/IslExprBuilder.cpp index ed2641fbd2df..b502e7a07182 100644 --- a/polly/lib/CodeGen/IslExprBuilder.cpp +++ b/polly/lib/CodeGen/IslExprBuilder.cpp @@ -115,9 +115,12 @@ Value *IslExprBuilder::createAccessAddress(isl_ast_expr *Expr) { assert(Base->getType()->isPointerTy() && "Access base should be a pointer"); StringRef BaseName = Base->getName(); - if (Base->getType() != SAI->getElementType()->getPointerTo()) - Base = Builder.CreateBitCast(Base, SAI->getElementType()->getPointerTo(), - "polly.access.cast." + BaseName); + auto PointerTy = PointerType::get(SAI->getElementType(), + Base->getType()->getPointerAddressSpace()); + if (Base->getType() != PointerTy) { + Base = + Builder.CreateBitCast(Base, PointerTy, "polly.access.cast." + BaseName); + } IndexOp = nullptr; for (unsigned u = 1, e = isl_ast_expr_get_op_n_arg(Expr); u < e; u++) { diff --git a/polly/test/Isl/CodeGen/MemAccess/codegen_address_space.ll b/polly/test/Isl/CodeGen/MemAccess/codegen_address_space.ll new file mode 100644 index 000000000000..ad465eda931f --- /dev/null +++ b/polly/test/Isl/CodeGen/MemAccess/codegen_address_space.ll @@ -0,0 +1,44 @@ +;RUN: opt %loadPolly -polly-detect-unprofitable -polly-no-early-exit -polly-import-jscop -polly-import-jscop-dir=%S -polly-import-jscop-postfix=transformed -polly-codegen -instnamer < %s -S | FileCheck %s + +;int A[100]; +; +;int codegen_simple () { +; for (int i = 0; i < 12; i++) +; A[13] = A[i] + A[i-1]; +; +; return 0; +;} + +target datalayout = "e-p:32:32:32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-f32:32:32-f64:32:64-v64:64:64-v128:128:128-a0:0:64-f80:32:32-n8:16:32" + +define i32 @codegen_simple([100 x i32] addrspace(5)* %A) nounwind { +entry: + %arrayidx.write = getelementptr [100 x i32], [100 x i32] addrspace(5)* %A, i32 0, i32 13 + br label %for.cond + +for.cond: ; preds = %for.inc, %entry + %tmp1 = phi i32 [ 0, %entry ], [ %inc, %for.inc ] + %tmp = add i32 %tmp1, -1 + %arrayidx4 = getelementptr [100 x i32], [100 x i32] addrspace(5)* %A, i32 0, i32 %tmp + %arrayidx = getelementptr [100 x i32], [100 x i32] addrspace(5)* %A, i32 0, i32 %tmp1 + %exitcond = icmp ne i32 %tmp1, 12 + br i1 %exitcond, label %for.body, label %for.end + +for.body: ; preds = %for.cond + %tmp2 = load i32, i32 addrspace(5)* %arrayidx, align 4 + %tmp5 = load i32, i32 addrspace(5)* %arrayidx4, align 4 + %add = add nsw i32 %tmp2, %tmp5 + store i32 %add, i32 addrspace(5)* %arrayidx.write + br label %for.inc + +for.inc: ; preds = %for.body + %inc = add nsw i32 %tmp1, 1 + br label %for.cond + +for.end: ; preds = %for.cond + ret i32 0 +} + +; CHECK: %polly.access.cast.A = bitcast [100 x i32] addrspace(5)* %A to i32 addrspace(5)* +; CHECK: %polly.access.A = getelementptr i32, i32 addrspace(5)* %polly.access.cast.A, i64 0 +; CHECK: %tmp2_p_scalar_ = load i32, i32 addrspace(5)* %polly.access.A, align 4, !alias.scope !0, !noalias !2