File tree Expand file tree Collapse file tree 1 file changed +4
-12
lines changed Expand file tree Collapse file tree 1 file changed +4
-12
lines changed Original file line number Diff line number Diff line change 188
188
flastlinindices (x) = LinearIndices (x)[findlast (x)]
189
189
flastlinindices (f, x) = LinearIndices (x)[findlast (f, x)]
190
190
191
- @test ffirstlinindices (x) ≈ @jit (findfirst (x_ra)) skip = contains (
192
- string (Reactant. devices ()[1 ]), " TPU"
193
- )
194
- @test flastlinindices (x) ≈ @jit (findlast (x_ra)) skip = contains (
195
- string (Reactant. devices ()[1 ]), " TPU"
196
- )
191
+ @test ffirstlinindices (x) ≈ @jit (findfirst (x_ra))
192
+ @test flastlinindices (x) ≈ @jit (findlast (x_ra))
197
193
198
194
x = Int64[
199
195
3 5 7 9
202
198
]
203
199
x_ra = Reactant. to_rarray (x)
204
200
205
- @test ffirstlinindices (iseven, x) ≈ @jit (findfirst (iseven, x_ra)) skip = contains (
206
- string (Reactant. devices ()[1 ]), " TPU"
207
- )
208
- @test flastlinindices (iseven, x) ≈ @jit (findlast (iseven, x_ra)) skip = contains (
209
- string (Reactant. devices ()[1 ]), " TPU"
210
- )
201
+ @test ffirstlinindices (iseven, x) ≈ @jit (findfirst (iseven, x_ra))
202
+ @test flastlinindices (iseven, x) ≈ @jit (findlast (iseven, x_ra))
211
203
end
212
204
213
205
@testset " approx top k lowering" begin
You can’t perform that action at this time.
0 commit comments