Skip to content
This repository was archived by the owner on Mar 28, 2023. It is now read-only.

[SYCL] Adopt test for CUDA backend #240

Merged
merged 2 commits into from
Apr 21, 2021
Merged

[SYCL] Adopt test for CUDA backend #240

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
ee91895
[SYCL] Adopt test for CUDA backend
vladimirlaz Apr 20, 2021
823c226
Apply clang format
vladimirlaz Apr 20, 2021
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
5 changes: 3 additions & 2 deletions SYCL/USM/memadvise.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,14 +29,15 @@ int main() {
queue q;
auto dev = q.get_device();
auto ctxt = q.get_context();
const int mem_advice = ((dev.get_backend() == backend::cuda) ? 1 : 0);
if (!dev.get_info<info::device::usm_shared_allocations>())
return 0;

Node *s_head = (Node *)malloc_shared(sizeof(Node), dev, ctxt);
if (s_head == nullptr) {
return -1;
}
q.mem_advise(s_head, sizeof(Node), (pi_mem_advice)0);
q.mem_advise(s_head, sizeof(Node), (pi_mem_advice)mem_advice);
Node *s_cur = s_head;

for (int i = 0; i < numNodes; i++) {
Expand All @@ -47,7 +48,7 @@ int main() {
if (s_cur->pNext == nullptr) {
return -1;
}
q.mem_advise(s_cur->pNext, sizeof(Node), (pi_mem_advice)0);
q.mem_advise(s_cur->pNext, sizeof(Node), (pi_mem_advice)mem_advice);
} else {
s_cur->pNext = nullptr;
}
Expand Down