forked from codeplaysoftware/syclacademy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
solution.cpp
74 lines (58 loc) · 1.98 KB
/
solution.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
/*
SYCL Academy (c)
SYCL Academy is licensed under a Creative Commons
Attribution-ShareAlike 4.0 International License.
You should have received a copy of the license along with this
work. If not, see <http://creativecommons.org/licenses/by-sa/4.0/>.
*/
#define CATCH_CONFIG_MAIN
#include <catch2/catch.hpp>
#include <sycl/sycl.hpp>
class scalar_add;
// Function device selector
int intel_gpu_selector1(const sycl::device &dev) {
if (dev.has(sycl::aspect::gpu)) {
auto vendorName = dev.get_info<sycl::info::device::vendor>();
if (vendorName.find("Intel") != std::string::npos) {
return 1;
}
}
return -1;
}
// Lambda device_selector
auto intel_gpu_selector2 = [](const sycl::device &dev) {
if (dev.has(sycl::aspect::gpu)) {
auto vendorName = dev.get_info<sycl::info::device::vendor>();
if (vendorName.find("Intel") != std::string::npos) {
return 1;
}
}
return -1;
};
TEST_CASE("intel_gpu_device_selector", "device_selectors_solution") {
int a = 18, b = 24, r = 0;
try {
auto defaultQueue1 = sycl::queue{intel_gpu_selector1};
auto defaultQueue2 = sycl::queue{intel_gpu_selector2};
std::cout << "Chosen device: "
<< defaultQueue1.get_device().get_info<sycl::info::device::name>()
<< std::endl;
{
auto bufA = sycl::buffer{&a, sycl::range{1}};
auto bufB = sycl::buffer{&b, sycl::range{1}};
auto bufR = sycl::buffer{&r, sycl::range{1}};
defaultQueue1
.submit([&](sycl::handler &cgh) {
auto accA = sycl::accessor{bufA, cgh, sycl::read_only};
auto accB = sycl::accessor{bufB, cgh, sycl::read_only};
auto accR = sycl::accessor{bufR, cgh, sycl::write_only};
cgh.single_task<scalar_add>([=]() { accR[0] = accA[0] + accB[0]; });
})
.wait();
}
defaultQueue1.throw_asynchronous();
} catch (const sycl::exception &e) {
std::cout << "Exception caught: " << e.what() << std::endl;
}
REQUIRE(r == 42);
}