test/CodeGen/PTX/st.ll

   1 ; RUN: llc < %s -march=ptx | FileCheck %s
   2
   3 ;CHECK: .extern .global .s32 array[];
   4 @array = external global [10 x i32]
   5
   6 ;CHECK: .extern .const .s32 array_constant[];
   7 @array_constant = external addrspace(1) constant [10 x i32]
   8
   9 ;CHECK: .extern .local .s32 array_local[];
  10 @array_local = external addrspace(2) global [10 x i32]
  11
  12 ;CHECK: .extern .shared .s32 array_shared[];
  13 @array_shared = external addrspace(4) global [10 x i32]
  14
  15 define ptx_device void @t1(i32* %p, i32 %x) {
  16 entry:
  17 ;CHECK: st.global.s32 [r1], r2;
  18   store i32 %x, i32* %p
  19   ret void
  20 }
  21
  22 define ptx_device void @t2(i32* %p, i32 %x) {
  23 entry:
  24 ;CHECK: st.global.s32 [r1+4], r2;
  25   %i = getelementptr i32* %p, i32 1
  26   store i32 %x, i32* %i
  27   ret void
  28 }
  29
  30 define ptx_device void @t3(i32* %p, i32 %q, i32 %x) {
  31 ;CHECK: .reg .s32 r0;
  32 entry:
  33 ;CHECK: shl.b32 r0, r2, 2;
  34 ;CHECK: add.s32 r0, r1, r0;
  35 ;CHECK: st.global.s32 [r0], r3;
  36   %i = getelementptr i32* %p, i32 %q
  37   store i32 %x, i32* %i
  38   ret void
  39 }
  40
  41 define ptx_device void @t4_global(i32 %x) {
  42 entry:
  43 ;CHECK: st.global.s32 [array], r1;
  44   %i = getelementptr [10 x i32]* @array, i32 0, i32 0
  45   store i32 %x, i32* %i
  46   ret void
  47 }
  48
  49 define ptx_device void @t4_local(i32 %x) {
  50 entry:
  51 ;CHECK: st.local.s32 [array_local], r1;
  52   %i = getelementptr [10 x i32] addrspace(2)* @array_local, i32 0, i32 0
  53   store i32 %x, i32 addrspace(2)* %i
  54   ret void
  55 }
  56
  57 define ptx_device void @t4_shared(i32 %x) {
  58 entry:
  59 ;CHECK: st.shared.s32 [array_shared], r1;
  60   %i = getelementptr [10 x i32] addrspace(4)* @array_shared, i32 0, i32 0
  61   store i32 %x, i32 addrspace(4)* %i
  62   ret void
  63 }
  64
  65 define ptx_device void @t5(i32 %x) {
  66 entry:
  67 ;CHECK: st.global.s32 [array+4], r1;
  68   %i = getelementptr [10 x i32]* @array, i32 0, i32 1
  69   store i32 %x, i32* %i
  70   ret void
  71 }