Skip to content

Commit 95caac1

Browse files
committed
Replace CUDASelector function w/ lambda
1 parent a0610ff commit 95caac1

File tree

2 files changed

+24
-17
lines changed

2 files changed

+24
-17
lines changed

examples/vector_addition/vector_addition.cpp

Lines changed: 12 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -24,15 +24,6 @@
2424

2525
#include <CL/sycl.hpp>
2626

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;
33-
}
34-
}
35-
3627
int main(int argc, char *argv[]) {
3728
constexpr const size_t N = 100000;
3829
const sycl::range VecSize{N};
@@ -53,6 +44,18 @@ int main(int argc, char *argv[]) {
5344
}
5445
}
5546

47+
auto CUDASelector = [](sycl::device const &dev)
48+
{
49+
if (dev.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda)
50+
{
51+
std::cout << " CUDA device found " << std::endl;
52+
return 1;
53+
}
54+
else
55+
{
56+
return -1;
57+
}
58+
};
5659
sycl::queue myQueue{CUDASelector};
5760

5861
// Command Group creation

examples/vector_addition/vector_addition_usm.cpp

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

2525
#include <CL/sycl.hpp>
2626

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;
33-
}
34-
}
3527

3628
int main(int argc, char *argv[]) {
3729
constexpr const size_t n = 100000;
3830

3931
// Create a sycl queue with our CUDASelector
32+
auto CUDASelector = [](sycl::device const &dev)
33+
{
34+
if (dev.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda)
35+
{
36+
std::cout << " CUDA device found " << std::endl;
37+
return 1;
38+
}
39+
else
40+
{
41+
return -1;
42+
}
43+
};
4044
sycl::queue myQueue{CUDASelector};
4145

4246
// Host input vectors

0 commit comments

Comments
 (0)