Lines Matching refs:runtime
1 // RUN: mlir-opt %s -async-runtime-ref-counting | FileCheck %s
15 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
21 // CHECK: %[[TOKEN:.*]] = async.runtime.create : !async.token
22 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
23 %0 = async.runtime.create : !async.token
30 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
38 // CHECK: async.runtime.add_ref %[[TOKEN]] {count = 1 : i64} : !async.token
40 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64} : !async.token
46 // CHECK: %[[TOKEN:.*]] = async.runtime.create : !async.token
47 %0 = async.runtime.create : !async.token
48 // CHECK: async.runtime.add_ref %[[TOKEN]] {count = 1 : i64} : !async.token
50 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
66 // CHECK: async.runtime.await %[[TOKEN]]
67 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
68 async.runtime.await %arg0 : !async.token
74 // CHECK: %[[TOKEN:.*]] = async.runtime.create : !async.token
75 %token = async.runtime.create : !async.token
81 // CHECK-NOT: async.runtime.drop_ref
82 // CHECK-NOT: async.runtime.add_ref
84 %token = async.runtime.create : !async.token
88 async.runtime.resume %hdl
103 %token = async.runtime.create : !async.token
107 // CHECK: async.runtime.await_and_resume %[[TOKEN]]
108 async.runtime.await_and_resume %arg0, %hdl : !async.token
109 // CHECK-NEXT: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
124 %token = async.runtime.create : !async.token
128 // CHECK: async.runtime.await_and_resume %[[VALUE]]
129 async.runtime.await_and_resume %arg0, %hdl : !async.value<f32>
135 // CHECK: %[[LOADED:.*]] = async.runtime.load %[[VALUE]]
136 // CHECK: async.runtime.drop_ref %[[VALUE]] {count = 1 : i64}
137 %0 = async.runtime.load %arg0 : !async.value<f32>
152 %0 = async.runtime.create : !async.token
156 async.runtime.resume %2
162 async.runtime.await_and_resume %arg0, %2 : !async.token
163 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
168 // CHECK: async.runtime.set_available
169 async.runtime.set_available %0 : !async.token
190 // CHECK: async.runtime.await %[[TOKEN]]
191 async.runtime.await %token : !async.token
194 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
206 // CHECK: async.runtime.await %[[TOKEN]]
207 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
208 async.runtime.await %token : !async.token
225 // CHECK: async.runtime.await %[[TOKEN]]
226 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
228 async.runtime.await %token : !async.token
231 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
247 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
252 // CHECK: async.runtime.await %[[TOKEN]]
253 // CHECK: async.runtime.drop_ref %[[TOKEN]] {count = 1 : i64}
255 async.runtime.await %token : !async.token
273 // CHECK: async.runtime.drop_ref %[[TOKEN1]] {count = 1 : i64}
274 // CHECK: async.runtime.await %[[TOKEN0]]
275 // CHECK: async.runtime.drop_ref %[[TOKEN0]] {count = 1 : i64}
277 async.runtime.await %token0 : !async.token
281 // CHECK: async.runtime.drop_ref %[[TOKEN0]] {count = 1 : i64}
282 // CHECK: async.runtime.await %[[TOKEN1]]
283 // CHECK: async.runtime.drop_ref %[[TOKEN1]] {count = 1 : i64}
285 async.runtime.await %token1 : !async.token