1409e1a5e5
This PR adds overrides in `SPIRVTTIImpl` for `collectFlatAddressOperands` and `rewriteIntrinsicWithAddressSpace` to enable `InferAddressSpacesPass` to rewrite the `llvm.spv.generic.cast.to.ptr.explicit` intrinsic (corresponding to `OpGenericCastToPtrExplicit`) when the address space of the argument can be inferred. When the destination address space of the cast matches the inferred address space of the argument, the call is replaced with that argument. When they do not match, the cast is replaced with a constant null pointer.
103 lines
4.1 KiB
LLVM
103 lines
4.1 KiB
LLVM
; This test checks that the address space casts for SPIR-V generic pointer casts
|
|
; are lowered correctly by the infer-address-spaces pass.
|
|
; RUN: opt < %s -passes=infer-address-spaces -S --mtriple=spirv64-unknown-unknown | FileCheck %s
|
|
|
|
; Casting a global pointer to a global pointer.
|
|
; The uses of c2 will be replaced with %global.
|
|
; CHECK: @kernel1(ptr addrspace(1) %global)
|
|
define i1 @kernel1(ptr addrspace(1) %global) {
|
|
%c1 = addrspacecast ptr addrspace(1) %global to ptr addrspace(4)
|
|
%c2 = call ptr addrspace(1) @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr addrspace(1) %global, null
|
|
%b1 = icmp eq ptr addrspace(1) %c2, null
|
|
ret i1 %b1
|
|
}
|
|
|
|
; Casting a global pointer to a local pointer.
|
|
; The uses of c2 will be replaced with null.
|
|
; CHECK: @kernel2(ptr addrspace(1) %global)
|
|
define i1 @kernel2(ptr addrspace(1) %global) {
|
|
%c1 = addrspacecast ptr addrspace(1) %global to ptr addrspace(4)
|
|
%c2 = call ptr addrspace(3) @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr addrspace(3) null, null
|
|
%b1 = icmp eq ptr addrspace(3) %c2, null
|
|
ret i1 %b1
|
|
}
|
|
|
|
; Casting a global pointer to a private pointer.
|
|
; The uses of c2 will be replaced with null.
|
|
; CHECK: @kernel3(ptr addrspace(1) %global)
|
|
define i1 @kernel3(ptr addrspace(1) %global) {
|
|
%c1 = addrspacecast ptr addrspace(1) %global to ptr addrspace(4)
|
|
%c2 = call ptr @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr null, null
|
|
%b1 = icmp eq ptr %c2, null
|
|
ret i1 %b1
|
|
}
|
|
|
|
; Casting a local pointer to a local pointer.
|
|
; The uses of c2 will be replaced with %local.
|
|
; CHECK: @kernel4(ptr addrspace(3) %local)
|
|
define i1 @kernel4(ptr addrspace(3) %local) {
|
|
%c1 = addrspacecast ptr addrspace(3) %local to ptr addrspace(4)
|
|
%c2 = call ptr addrspace(3) @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr addrspace(3) %local, null
|
|
%b1 = icmp eq ptr addrspace(3) %c2, null
|
|
ret i1 %b1
|
|
}
|
|
|
|
; Casting a local pointer to a global pointer.
|
|
; The uses of c2 will be replaced with null.
|
|
; CHECK: @kernel5(ptr addrspace(3) %local)
|
|
define i1 @kernel5(ptr addrspace(3) %local) {
|
|
%c1 = addrspacecast ptr addrspace(3) %local to ptr addrspace(4)
|
|
%c2 = call ptr addrspace(1) @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr addrspace(1) null, null
|
|
%b1 = icmp eq ptr addrspace(1) %c2, null
|
|
ret i1 %b1
|
|
}
|
|
|
|
; Casting a local pointer to a private pointer.
|
|
; The uses of c2 will be replaced with null.
|
|
; CHECK: @kernel6(ptr addrspace(3) %local)
|
|
define i1 @kernel6(ptr addrspace(3) %local) {
|
|
%c1 = addrspacecast ptr addrspace(3) %local to ptr addrspace(4)
|
|
%c2 = call ptr @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr null, null
|
|
%b1 = icmp eq ptr %c2, null
|
|
ret i1 %b1
|
|
}
|
|
|
|
; Casting a private pointer to a private pointer.
|
|
; The uses of c2 will be replaced with %private.
|
|
; CHECK: @kernel7(ptr %private)
|
|
define i1 @kernel7(ptr %private) {
|
|
%c1 = addrspacecast ptr %private to ptr addrspace(4)
|
|
%c2 = call ptr @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr %private, null
|
|
%b1 = icmp eq ptr %c2, null
|
|
ret i1 %b1
|
|
}
|
|
|
|
; Casting a private pointer to a global pointer.
|
|
; The uses of c2 will be replaced with null.
|
|
; CHECK: @kernel8(ptr %private)
|
|
define i1 @kernel8(ptr %private) {
|
|
%c1 = addrspacecast ptr %private to ptr addrspace(4)
|
|
%c2 = call ptr addrspace(1) @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr addrspace(1) null, null
|
|
%b1 = icmp eq ptr addrspace(1) %c2, null
|
|
ret i1 %b1
|
|
}
|
|
|
|
; Casting a private pointer to a local pointer.
|
|
; The uses of c2 will be replaced with null.
|
|
; CHECK: @kernel9(ptr %private)
|
|
define i1 @kernel9(ptr %private) {
|
|
%c1 = addrspacecast ptr %private to ptr addrspace(4)
|
|
%c2 = call ptr addrspace(3) @llvm.spv.generic.cast.to.ptr.explicit(ptr addrspace(4) %c1)
|
|
; CHECK: %b1 = icmp eq ptr addrspace(3) null, null
|
|
%b1 = icmp eq ptr addrspace(3) %c2, null
|
|
ret i1 %b1
|
|
}
|