RPCSX/llvm
1
0
Fork 0
mirror of https://github.com/RPCSX/llvm.git synced 2025-01-08 04:52:50 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

AMDGPU/SI: Give up in promote alloca when a pointer may be captured.

Browse Source 
Differential Revision:
  http://reviews.llvm.org/D28970

Reviewer:
  Matt

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@292966 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Changpeng Fang 2017-01-24 19:06:28 +00:00
parent 60b4e66e58
commit 6562a033a3
2 changed files with 51 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/Target/AMDGPU/AMDGPUPromoteAlloca.cpp
View File

@ -21,6 +21,7 @@
#include "llvm/ADT/StringRef.h"
#include "llvm/ADT/Triple.h"
#include "llvm/ADT/Twine.h"
#include "llvm/Analysis/CaptureTracking.h"
#include "llvm/Analysis/ValueTracking.h"
#include "llvm/IR/Attributes.h"
#include "llvm/IR/BasicBlock.h"
@ -608,6 +609,9 @@ bool AMDGPUPromoteAlloca::collectUsesWithPtrTypes(
}
if (UseInst->getOpcode() == Instruction::AddrSpaceCast) {
// Give up if the pointer may be captured.
if (PointerMayBeCaptured(UseInst, true, true))
return false;
// Don't collect the users of this.
WorkList.push_back(User);
continue;

47
test/CodeGen/AMDGPU/addrspacecast-captured.ll Normal file
View File

@ -0,0 +1,47 @@
; RUN: opt -S -mtriple=amdgcn-unknown-amdhsa -mcpu=kaveri -amdgpu-promote-alloca < %s | FileCheck %s
; Nothing should be done if the addrspacecast is captured.
declare void @consume_ptr2int(i32) #0
; CHECK-LABEL: @addrspacecast_captured(
; CHECK: %data = alloca i32, align 4
; CHECK: %cast = addrspacecast i32* %data to i32 addrspace(4)*
; CHECK: %ptr2int = ptrtoint i32 addrspace(4)* %cast to i32
; CHECK: store i32 %ptr2int, i32 addrspace(1)* %out
define void @addrspacecast_captured(i32 addrspace(1)* %out) #0 {
entry:
%data = alloca i32, align 4
%cast = addrspacecast i32* %data to i32 addrspace(4)*
%ptr2int = ptrtoint i32 addrspace(4)* %cast to i32
store i32 %ptr2int, i32 addrspace(1)* %out
ret void
}
; CHECK-LABEL: @addrspacecast_captured_store(
; CHECK: %data = alloca i32, align 4
; CHECK: %cast = addrspacecast i32* %data to i32 addrspace(4)*
; CHECK: store i32 addrspace(4)* %cast, i32 addrspace(4)* addrspace(1)* %out
define void @addrspacecast_captured_store(i32 addrspace(4)* addrspace(1)* %out) #0 {
entry:
%data = alloca i32, align 4
%cast = addrspacecast i32* %data to i32 addrspace(4)*
store i32 addrspace(4)* %cast, i32 addrspace(4)* addrspace(1)* %out
ret void
}
; CHECK-LABEL: @addrspacecast_captured_call(
; CHECK: %data = alloca i32, align 4
; CHECK: %cast = addrspacecast i32* %data to i32 addrspace(4)*
; CHECK: %ptr2int = ptrtoint i32 addrspace(4)* %cast to i32
; CHECK: call void @consume_ptr2int(i32 %ptr2int)
define void @addrspacecast_captured_call() #0 {
entry:
%data = alloca i32, align 4
%cast = addrspacecast i32* %data to i32 addrspace(4)*
%ptr2int = ptrtoint i32 addrspace(4)* %cast to i32
call void @consume_ptr2int(i32 %ptr2int)
ret void
}
attributes #0 = { nounwind }