llvm/test/Transforms/InferAddressSpaces/AMDGPU/infer-addrspacecast.ll

   1 ; RUN: opt -S -mtriple=amdgcn-amd-amdhsa -infer-address-spaces %s | FileCheck %s
   2
   3 ; Test that pure addrspacecast instructions not directly connected to
   4 ; a memory operation are inferred.
   5
   6 ; CHECK-LABEL: @addrspacecast_gep_addrspacecast(
   7 ; CHECK: %gep0 = getelementptr i32, i32 addrspace(3)* %ptr, i64 9
   8 ; CHECK-NEXT: store i32 8, i32 addrspace(3)* %gep0, align 8
   9 ; CHECK-NEXT: ret void
  10 define void @addrspacecast_gep_addrspacecast(i32 addrspace(3)* %ptr) {
  11   %asc0 = addrspacecast i32 addrspace(3)* %ptr to i32*
  12   %gep0 = getelementptr i32, i32* %asc0, i64 9
  13   %asc1 = addrspacecast i32* %gep0 to i32 addrspace(3)*
  14   store i32 8, i32 addrspace(3)* %asc1, align 8
  15   ret void
  16 }
  17
  18 ; CHECK-LABEL: @addrspacecast_different_pointee_type(
  19 ; CHECK: [[GEP:%.*]] = getelementptr i32, i32 addrspace(3)* %ptr, i64 9
  20 ; CHECK: [[CAST:%.*]] = bitcast i32 addrspace(3)* [[GEP]] to i8 addrspace(3)*
  21 ; CHECK-NEXT: store i8 8, i8 addrspace(3)* [[CAST]], align 8
  22 ; CHECK-NEXT: ret void
  23 define void @addrspacecast_different_pointee_type(i32 addrspace(3)* %ptr) {
  24   %asc0 = addrspacecast i32 addrspace(3)* %ptr to i32*
  25   %gep0 = getelementptr i32, i32* %asc0, i64 9
  26   %asc1 = addrspacecast i32* %gep0 to i8 addrspace(3)*
  27   store i8 8, i8 addrspace(3)* %asc1, align 8
  28   ret void
  29 }
  30
  31 ; CHECK-LABEL: @addrspacecast_to_memory(
  32 ; CHECK: %gep0 = getelementptr i32, i32 addrspace(3)* %ptr, i64 9
  33 ; CHECK-NEXT: store volatile i32 addrspace(3)* %gep0, i32 addrspace(3)* addrspace(1)* undef
  34 ; CHECK-NEXT: ret void
  35 define void @addrspacecast_to_memory(i32 addrspace(3)* %ptr) {
  36   %asc0 = addrspacecast i32 addrspace(3)* %ptr to i32*
  37   %gep0 = getelementptr i32, i32* %asc0, i64 9
  38   %asc1 = addrspacecast i32* %gep0 to i32 addrspace(3)*
  39   store volatile i32 addrspace(3)* %asc1, i32 addrspace(3)* addrspace(1)* undef
  40   ret void
  41 }
  42
  43 ; CHECK-LABEL: @multiuse_addrspacecast_gep_addrspacecast(
  44 ; CHECK: %asc0 = addrspacecast i32 addrspace(3)* %ptr to i32*
  45 ; CHECK-NEXT: store volatile i32* %asc0, i32* addrspace(1)* undef
  46 ; CHECK-NEXT: %gep0 = getelementptr i32, i32 addrspace(3)* %ptr, i64 9
  47 ; CHECK-NEXT: store i32 8, i32 addrspace(3)* %gep0, align 8
  48 ; CHECK-NEXT: ret void
  49 define void @multiuse_addrspacecast_gep_addrspacecast(i32 addrspace(3)* %ptr) {
  50   %asc0 = addrspacecast i32 addrspace(3)* %ptr to i32*
  51   store volatile i32* %asc0, i32* addrspace(1)* undef
  52   %gep0 = getelementptr i32, i32* %asc0, i64 9
  53   %asc1 = addrspacecast i32* %gep0 to i32 addrspace(3)*
  54   store i32 8, i32 addrspace(3)* %asc1, align 8
  55   ret void
  56 }