1; RUN: llc --mtriple=wasm32-unknown-unknown -asm-verbose=false -mattr=+reference-types < %s | FileCheck %s
2
3%externref = type ptr addrspace(10) ;; addrspace 10 is nonintegral
4
5@externref_table = local_unnamed_addr addrspace(1) global [0 x %externref] undef
6
7define void @set_externref_table(%externref %g, i32 %i) {
8; CHECK-LABEL: set_externref_table:
9; CHECK-NEXT: .functype       set_externref_table (externref, i32) -> ()
10; CHECK-NEXT: local.get      1
11; CHECK-NEXT: local.get      0
12; CHECK-NEXT: table.set     externref_table
13; CHECK-NEXT: end_function
14
15;; this generates a table.set of @externref_table
16  %p = getelementptr [0 x %externref], ptr addrspace (1) @externref_table, i32 0, i32 %i
17  store %externref %g, ptr addrspace(1) %p
18  ret void
19}
20
21define void @set_externref_table_const(%externref %g) {
22; CHECK-LABEL: set_externref_table_const:
23; CHECK-NEXT:  .functype      set_externref_table_const (externref) -> ()
24; CHECK-NEXT:  i32.const      0
25; CHECK-NEXT:  local.get      0
26; CHECK-NEXT:  table.set      externref_table
27; CHECK-NEXT:  end_function
28  %p = getelementptr [0 x %externref], ptr addrspace (1) @externref_table, i32 0, i32 0
29  store %externref %g, ptr addrspace(1) %p
30  ret void
31}
32
33define void @set_externref_table_with_offset(%externref %g, i32 %i) {
34; CHECK-LABEL: set_externref_table_with_offset:
35; CHECK-NEXT:  .functype       set_externref_table_with_offset (externref, i32) -> ()
36; CHECK-NEXT:  local.get       1
37; CHECK-NEXT:  i32.const       2
38; CHECK-NEXT:  i32.add
39; CHECK-NEXT:  local.get       0
40; CHECK-NEXT:  table.set       externref_table
41; CHECK-NEXT:  end_function
42  %off = add nsw i32 %i, 2
43  %p = getelementptr [0 x %externref], ptr addrspace (1) @externref_table, i32 0, i32 %off
44  store %externref %g, ptr addrspace(1) %p
45  ret void
46}
47
48define void @set_externref_table_with_var_offset(%externref %g, i32 %i, i32 %j) {
49; CHECK-LABEL: set_externref_table_with_var_offset:
50; CHECK-NEXT:  .functype       set_externref_table_with_var_offset (externref, i32, i32) -> ()
51; CHECK-NEXT:  local.get       1
52; CHECK-NEXT:  local.get       2
53; CHECK-NEXT:  i32.add
54; CHECK-NEXT:  local.get       0
55; CHECK-NEXT:  table.set       externref_table
56; CHECK-NEXT:  end_function
57  %off = add nsw i32 %i, %j
58  %p = getelementptr [0 x %externref], ptr addrspace (1) @externref_table, i32 0, i32 %off
59  store %externref %g, ptr addrspace(1) %p
60  ret void
61}
62
63declare i32 @set_offset()
64
65define void @set_externref_table_with_var_offset2(%externref %g, i32 %i) {
66; CHECK-LABEL: set_externref_table_with_var_offset2:
67; CHECK-NEXT:  .functype       set_externref_table_with_var_offset2 (externref, i32) -> ()
68; CHECK-NEXT:  local.get       1
69; CHECK-NEXT:  call    set_offset
70; CHECK-NEXT:  i32.add
71; CHECK-NEXT:  local.get       0
72; CHECK-NEXT:  table.set       externref_table
73; CHECK-NEXT:  end_function
74  %j = call i32 @set_offset()
75  %off = add nsw i32 %i, %j
76  %p = getelementptr [0 x %externref], ptr addrspace (1) @externref_table, i32 0, i32 %off
77  store %externref %g, ptr addrspace(1) %p
78  ret void
79}
80
81declare i32 @get_table_slot() local_unnamed_addr
82
83define void @set_externref_table_with_id_from_call(%externref %g) {
84; CHECK-LABEL: set_externref_table_with_id_from_call:
85; CHECK-NEXT:  .functype       set_externref_table_with_id_from_call (externref) -> ()
86; CHECK-NEXT:  call    get_table_slot
87; CHECK-NEXT:  local.get       0
88; CHECK-NEXT:  table.set       externref_table
89; CHECK-NEXT:  end_function
90  %id = call i32 @get_table_slot()
91  %p = getelementptr [0 x %externref], ptr addrspace (1) @externref_table, i32 0, i32 %id
92  store %externref %g, ptr addrspace(1) %p
93  ret void
94}
95
96;       CHECK: .tabletype externref_table, externref
97; CHECK-LABEL: externref_table:
98