| 318 | } |
| 319 | |
| 320 | struct nearest_neighbors_params { |
| 321 | string testname_; |
| 322 | int k_, nfeat_, ntrain_, nquery_; |
| 323 | int feat_dim_; |
| 324 | dim4 qdims_, tdims_, idims_, ddims_; |
| 325 | vector<float> query_; |
| 326 | vector<float> train_; |
| 327 | vector<unsigned int> indices_; |
| 328 | vector<float> dists_; |
| 329 | |
| 330 | nearest_neighbors_params(string testname, int k, int feat_dim, array query, |
| 331 | array train, array indices, array dists) |
| 332 | : testname_(testname) |
| 333 | , k_(k) |
| 334 | , feat_dim_(feat_dim) |
| 335 | , query_(query.elements()) |
| 336 | , train_(train.elements()) |
| 337 | , indices_(indices.elements()) |
| 338 | , dists_(dists.elements()) { |
| 339 | qdims_ = query.dims(); |
| 340 | tdims_ = train.dims(); |
| 341 | idims_ = indices.dims(); |
| 342 | ddims_ = dists.dims(); |
| 343 | |
| 344 | query.host(query_.data()); |
| 345 | train.host(train_.data()); |
| 346 | indices.host(indices_.data()); |
| 347 | dists.host(dists_.data()); |
| 348 | } |
| 349 | }; |
| 350 | |
| 351 | template<typename TestClass> |
| 352 | string testNameGenerator( |
no outgoing calls
no test coverage detected