Skip to content

Update vector_addition for new device selector #26

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
joeatodd merged 3 commits into from
Nov 10, 2022
Merged

Update vector_addition for new device selector #26

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
a0610ff
Update vector_addition for new device selector
joeatodd Nov 10, 2022
95caac1
Replace CUDASelector function w/ lambda
joeatodd Nov 10, 2022
66ee339
clang-format the lambdas
joeatodd Nov 10, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 9 additions & 13 deletions examples/vector_addition/vector_addition.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,18 +24,6 @@

#include <CL/sycl.hpp>

class CUDASelector : public sycl::device_selector {
public:
int operator()(const sycl::device &device) const override {
if(device.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda){
std::cout << " CUDA device found " << std::endl;
return 1;
} else{
return -1;
}
}
};

int main(int argc, char *argv[]) {
constexpr const size_t N = 100000;
const sycl::range VecSize{N};
Expand All @@ -56,7 +44,15 @@ int main(int argc, char *argv[]) {
}
}

sycl::queue myQueue{CUDASelector()};
auto CUDASelector = [](sycl::device const &dev) {
if (dev.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda) {
std::cout << " CUDA device found " << std::endl;
return 1;
} else {
return -1;
}
};
sycl::queue myQueue{CUDASelector};

// Command Group creation
auto cg = [&](sycl::handler &h) {
Expand Down
22 changes: 9 additions & 13 deletions examples/vector_addition/vector_addition_usm.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,23 +24,19 @@

#include <CL/sycl.hpp>

class CUDASelector : public sycl::device_selector {
public:
int operator()(const sycl::device &device) const override {
if(device.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda){
std::cout << " CUDA device found " << std::endl;
return 1;
} else{
return -1;
}
}
};

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

// Create a sycl queue with our CUDASelector
sycl::queue myQueue{CUDASelector()};
auto CUDASelector = [](sycl::device const &dev) {
if (dev.get_platform().get_backend() == sycl::backend::ext_oneapi_cuda) {
std::cout << " CUDA device found " << std::endl;
return 1;
} else {
return -1;
}
};
sycl::queue myQueue{CUDASelector};

// Host input vectors
double *h_a;
Expand Down