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