| // RUN: %clang_cc1 -triple amdgpu-amd-amdhsa -fclangir -emit-cir %s -o %t.cir |
| // RUN: FileCheck --input-file=%t.cir %s -check-prefix=CIR |
| // RUN: %clang_cc1 -triple amdgpu-amd-amdhsa -fclangir -emit-llvm %s -o %t.ll |
| // RUN: FileCheck --input-file=%t.ll %s -check-prefix=LLVM |
| // RUN: %clang_cc1 -triple amdgpu-amd-amdhsa -emit-llvm %s -o %t.ll |
| // RUN: FileCheck --input-file=%t.ll %s -check-prefix=OGCG |
| |
| // Test that address spaces are preserved through array-to-pointer decay |
| // and array element access on AMDGPU, where globals are in AS 1. |
| |
| int globalArr[10] = {0}; |
| |
| void takes_ptr(int *p); |
| |
| // The array_to_ptrdecay cast must preserve the address space of the base |
| // pointer, followed by an address_space cast. |
| |
| // CIR-LABEL: cir.func{{.*}} @_Z17pass_global_arrayv() |
| // CIR: %[[ARR:.*]] = cir.get_global @globalArr : !cir.ptr<!cir.array<!s32i x 10>, target_address_space(1)> |
| // CIR-NEXT: %[[DECAY:.*]] = cir.cast array_to_ptrdecay %[[ARR]] : !cir.ptr<!cir.array<!s32i x 10>, target_address_space(1)> -> !cir.ptr<!s32i, target_address_space(1)> |
| // CIR-NEXT: %[[FLAT:.*]] = cir.cast address_space %[[DECAY]] : !cir.ptr<!s32i, target_address_space(1)> -> !cir.ptr<!s32i> |
| // CIR-NEXT: cir.call @_Z9takes_ptrPi(%[[FLAT]]) |
| |
| // LLVM-LABEL: define{{.*}} void @_Z17pass_global_arrayv() |
| // LLVM: call void @_Z9takes_ptrPi(ptr noundef addrspacecast (ptr addrspace(1) @globalArr to ptr)) |
| |
| // OGCG-LABEL: define{{.*}} void @_Z17pass_global_arrayv() |
| // OGCG: call void @_Z9takes_ptrPi(ptr noundef addrspacecast (ptr addrspace(1) @globalArr to ptr)) |
| void pass_global_array() { |
| takes_ptr(globalArr); |
| } |
| |
| // The get_element op must preserve the address space of the base pointer |
| // so that the subsequent load uses the correct address space. |
| |
| // CIR-LABEL: cir.func{{.*}} @_Z18index_global_arrayi |
| // CIR: %[[ARR:.*]] = cir.get_global @globalArr : !cir.ptr<!cir.array<!s32i x 10>, target_address_space(1)> |
| // CIR-NEXT: %[[ELEM:.*]] = cir.get_element %[[ARR]][%{{.*}} : !s64i] : !cir.ptr<!cir.array<!s32i x 10>, target_address_space(1)> -> !cir.ptr<!s32i, target_address_space(1)> |
| // CIR-NEXT: %{{.*}} = cir.load align(4) %[[ELEM]] : !cir.ptr<!s32i, target_address_space(1)>, !s32i |
| |
| // LLVM-LABEL: define{{.*}} i32 @_Z18index_global_arrayi |
| // LLVM: %[[GEP:.*]] = getelementptr [10 x i32], ptr addrspace(1) @globalArr, i32 0, i64 %{{.*}} |
| // LLVM-NEXT: %{{.*}} = load i32, ptr addrspace(1) %[[GEP]], align 4 |
| |
| // OGCG-LABEL: define{{.*}} i32 @_Z18index_global_arrayi |
| // OGCG: getelementptr inbounds [10 x i32], ptr addrspacecast (ptr addrspace(1) @globalArr to ptr) |
| // OGCG: load i32, ptr %{{.*}}, align 4 |
| int index_global_array(int i) { |
| return globalArr[i]; |
| } |
