llvm/test/Transforms/InferAddressSpaces/masked-gather-scatter.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 2
   2 ; RUN: opt -S -passes=infer-address-spaces -assume-default-is-flat-addrspace %s | FileCheck %s
   3
   4 define <4 x i32> @masked_gather_inferas(ptr addrspace(1) %out, <4 x i64> %index) {
   5 ; CHECK-LABEL: define <4 x i32> @masked_gather_inferas
   6 ; CHECK-SAME: (ptr addrspace(1) [[OUT:%.*]], <4 x i64> [[INDEX:%.*]]) {
   7 ; CHECK-NEXT:  entry:
   8 ; CHECK-NEXT:    [[PTRS:%.*]] = getelementptr inbounds i32, ptr addrspace(1) [[OUT]], <4 x i64> [[INDEX]]
   9 ; CHECK-NEXT:    [[VALUE:%.*]] = tail call <4 x i32> @llvm.masked.gather.v4i32.v4p1(<4 x ptr addrspace(1)> [[PTRS]], i32 4, <4 x i1> <i1 true, i1 true, i1 true, i1 true>, <4 x i32> poison)
  10 ; CHECK-NEXT:    ret <4 x i32> [[VALUE]]
  11 ;
  12 entry:
  13   %out.1 = addrspacecast ptr addrspace(1) %out to ptr
  14   %ptrs = getelementptr inbounds i32, ptr %out.1, <4 x i64> %index
  15   %value = tail call <4 x i32> @llvm.masked.gather.v4i32.v4p0(<4 x ptr> %ptrs, i32 4, <4 x i1> <i1 true, i1 true, i1 true, i1 true>, <4 x i32> poison)
  16   ret <4 x i32> %value
  17 }
  18
  19 define void @masked_scatter_inferas(ptr addrspace(1) %out, <4 x i64> %index, <4 x i32> %value) {
  20 ; CHECK-LABEL: define void @masked_scatter_inferas
  21 ; CHECK-SAME: (ptr addrspace(1) [[OUT:%.*]], <4 x i64> [[INDEX:%.*]], <4 x i32> [[VALUE:%.*]]) {
  22 ; CHECK-NEXT:  entry:
  23 ; CHECK-NEXT:    [[PTRS:%.*]] = getelementptr inbounds i32, ptr addrspace(1) [[OUT]], <4 x i64> [[INDEX]]
  24 ; CHECK-NEXT:    tail call void @llvm.masked.scatter.v4i32.v4p1(<4 x i32> [[VALUE]], <4 x ptr addrspace(1)> [[PTRS]], i32 4, <4 x i1> <i1 true, i1 true, i1 true, i1 true>)
  25 ; CHECK-NEXT:    ret void
  26 ;
  27 entry:
  28   %out.1 = addrspacecast ptr addrspace(1) %out to ptr
  29   %ptrs = getelementptr inbounds i32, ptr %out.1, <4 x i64> %index
  30   tail call void @llvm.masked.scatter.v4i32.v4p0(<4 x i32> %value, <4 x ptr> %ptrs, i32 4, <4 x i1> <i1 true, i1 true, i1 true, i1 true>)
  31   ret void
  32 }
  33
  34 declare <4 x i32> @llvm.masked.gather.v4i32.v4p0(<4 x ptr>, i32 immarg, <4 x i1>, <4 x i32>)
  35
  36 declare void @llvm.masked.scatter.v4i32.v4p0(<4 x i32>, <4 x ptr>, i32 immarg, <4 x i1>)