llvm-project/llvm/test/CodeGen/SPIRV/pointers/pointer-addrspacecast.ll
Nathan Gauër 7d98b66e3d
[SPIR-V] Add InferAddrSpaces pass to the backend (#137766)
This commit enables a pass in the backend which propagates the addrspace
of the pointers down to the last use, making sure the addrspace remains
consistent, and thus stripping any addrspacecast. This is required to
lower LLVM-IR to logical SPIR-V, which does not support generic
pointers.

This is now required as HLSL emits several address spaces, and thus
addrspacecasts in some cases:

Example 1: resource access

```llvm
%handle = tail call target("spirv.VulkanBuffer", ...)
%rptr = @llvm.spv.resource.getpointer(%handle, ...);
%cptr = addrspacecast ptr addrspace(11) %rptr to ptr
%fptr = load i32, ptr %cptr
```

Example 2: object methods

```llvm
define void @objectMethod(ptr %this) {
}

define void @foo(ptr addrspace(11) %object) {
  call void @objectMethod(ptr addrspacecast(addrspace(11) %object to ptr));
}
```
2025-05-07 16:53:25 +02:00

37 lines
1.3 KiB
LLVM

; RUN: llc -verify-machineinstrs -O3 -mtriple=spirv-unknown-vulkan1.3-compute %s -o - | FileCheck %s
; RUN: %if spirv-tools %{ llc -O3 -mtriple=spirv-unknown-vulkan1.3-compute %s -o - -filetype=obj | spirv-val %}
; CHECK-DAG: %[[#uint:]] = OpTypeInt 32 0
; CHECK-DAG: %[[#uint_0:]] = OpConstant %[[#uint]] 0
; CHECK-DAG: %[[#ptr_uint:]] = OpTypePointer Private %[[#uint]]
; CHECK-DAG: %[[#var:]] = OpVariable %[[#ptr_uint]] Private %[[#uint_0]]
; CHECK-DAG: OpName %[[#func_simple:]] "simple"
; CHECK-DAG: OpName %[[#func_chain:]] "chain"
@global = internal addrspace(10) global i32 zeroinitializer
define void @simple() {
; CHECK: %[[#func_simple]] = OpFunction
entry:
%ptr = getelementptr i32, ptr addrspace(10) @global, i32 0
%casted = addrspacecast ptr addrspace(10) %ptr to ptr
%val = load i32, ptr %casted
; CHECK: %{{.*}} = OpLoad %[[#uint]] %[[#var]] Aligned 4
ret void
}
define void @chain() {
; CHECK: %[[#func_chain]] = OpFunction
entry:
%a = getelementptr i32, ptr addrspace(10) @global, i32 0
%b = addrspacecast ptr addrspace(10) %a to ptr
%c = getelementptr i32, ptr %b, i32 0
%d = addrspacecast ptr %c to ptr addrspace(10)
%e = addrspacecast ptr addrspace(10) %d to ptr
%val = load i32, ptr %e
; CHECK: %{{.*}} = OpLoad %[[#uint]] %[[#var]] Aligned 4
ret void
}