xref: /llvm-project/llvm/test/CodeGen/SPIRV/transcoding/bitcast-diff-addrspace.ll (revision 9a737109a07c0c29176e617eed34e91b523b7e9b)
1; Check that bitcast for pointers implies that the address spaces must match.
2
3; RUN: llc -O0 -mtriple=spirv64-unknown-unknown %s --spirv-ext=+SPV_INTEL_usm_storage_classes -o - | FileCheck %s
4; TODO: %if spirv-tools %{ llc -O0 -mtriple=spirv64-unknown-unknown %s --spirv-ext=+SPV_INTEL_usm_storage_classes -o - -filetype=obj | spirv-val %}
5
6; CHECK: Capability USMStorageClassesINTEL
7; CHECK: OpExtension "SPV_INTEL_usm_storage_classes"
8; CHECK-DAG: OpName %[[#Bar:]] "bar"
9; CHECK-DAG: %[[#Void:]] = OpTypeVoid
10; CHECK-DAG: %[[#Char:]] = OpTypeInt 8 0
11; CHECK-DAG: %[[#GenPtrChar:]] = OpTypePointer Generic %[[#Char]]
12; CHECK-DAG: %[[#AtDevice:]] = OpTypePointer DeviceOnlyINTEL %[[#Char]]
13; CHECK-DAG: %[[#AtHost:]] = OpTypePointer HostOnlyINTEL %[[#Char]]
14; CHECK-DAG: %[[#PtrVarHost:]] = OpTypePointer Function %[[#AtHost]]
15; CHECK-DAG: %[[#PtrVarDevice:]] = OpTypePointer Function %[[#AtDevice]]
16; CHECK: OpFunction
17; CHECK: %[[#VarDevice:]] = OpVariable %[[#PtrVarDevice]] Function
18; CHECK: %[[#VarHost:]] = OpVariable %[[#PtrVarHost]] Function
19; CHECK: %[[#LoadedDevice:]] = OpLoad %[[#AtDevice]] %[[#VarDevice]]
20; CHECK: %[[#CastedFromDevice:]] = OpPtrCastToGeneric %[[#GenPtrChar]] %[[#LoadedDevice]]
21; CHECK: OpFunctionCall %[[#Void]] %[[#Bar]] %[[#CastedFromDevice]]
22; CHECK: %[[#LoadedHost:]] = OpLoad %[[#AtHost]] %[[#VarHost]]
23; CHECK: %[[#CastedFromHost:]] = OpPtrCastToGeneric %[[#GenPtrChar]] %[[#LoadedHost]]
24; CHECK: OpFunctionCall %[[#Void]] %[[#Bar]] %[[#CastedFromHost]]
25
26define spir_func void @foo() {
27entry:
28  %device_var = alloca ptr addrspace(5)
29  %host_var = alloca ptr addrspace(6)
30  %p1 = load ptr addrspace(5), ptr %device_var
31  %p2 = addrspacecast ptr addrspace(5) %p1 to ptr addrspace(4)
32  call spir_func void @bar(ptr addrspace(4) %p2)
33  %p3 = load ptr addrspace(6), ptr %host_var
34  %p4 = addrspacecast ptr addrspace(6) %p3 to ptr addrspace(4)
35  call spir_func void @bar(ptr addrspace(4) %p4)
36  ret void
37}
38
39define spir_func void @bar(ptr addrspace(4) %data) {
40entry:
41  %data.addr = alloca ptr addrspace(4)
42  store ptr addrspace(4) %data, ptr %data.addr
43  ret void
44}
45