Skip to content

Commit a0610ff

Browse files
committed
Update vector_addition for new device selector
1 parent 380fe4f commit a0610ff

File tree

2 files changed

+12
-18
lines changed

2 files changed

+12
-18
lines changed

examples/vector_addition/vector_addition.cpp

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -24,17 +24,14 @@
2424

2525
#include <CL/sycl.hpp>
2626

27-
class CUDASelector : public sycl::device_selector {
28-
public:
29-
int operator()(const sycl::device &device) const override {
30-
if(device.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda){
27+
int CUDASelector(sycl::device const & dev){
28+
if(dev.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda){
3129
std::cout << " CUDA device found " << std::endl;
3230
return 1;
3331
} else{
3432
return -1;
3533
}
36-
}
37-
};
34+
}
3835

3936
int main(int argc, char *argv[]) {
4037
constexpr const size_t N = 100000;
@@ -56,7 +53,7 @@ int main(int argc, char *argv[]) {
5653
}
5754
}
5855

59-
sycl::queue myQueue{CUDASelector()};
56+
sycl::queue myQueue{CUDASelector};
6057

6158
// Command Group creation
6259
auto cg = [&](sycl::handler &h) {

examples/vector_addition/vector_addition_usm.cpp

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -24,23 +24,20 @@
2424

2525
#include <CL/sycl.hpp>
2626

27-
class CUDASelector : public sycl::device_selector {
28-
public:
29-
int operator()(const sycl::device &device) const override {
30-
if(device.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda){
31-
std::cout << " CUDA device found " << std::endl;
32-
return 1;
33-
} else{
34-
return -1;
35-
}
27+
int CUDASelector(sycl::device const &dev) {
28+
if (dev.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda) {
29+
std::cout << " CUDA device found " << std::endl;
30+
return 1;
31+
} else {
32+
return -1;
3633
}
37-
};
34+
}
3835

3936
int main(int argc, char *argv[]) {
4037
constexpr const size_t n = 100000;
4138

4239
// Create a sycl queue with our CUDASelector
43-
sycl::queue myQueue{CUDASelector()};
40+
sycl::queue myQueue{CUDASelector};
4441

4542
// Host input vectors
4643
double *h_a;

0 commit comments

Comments
 (0)