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