Вы не можете выбрать более 25 тем
Темы должны начинаться с буквы или цифры, могут содержать дефисы(-) и должны содержать не более 35 символов.
53 строки
1.3 KiB
53 строки
1.3 KiB
// Patric Zhao: patric.zhao@intel.com
|
|
|
|
#include <CL/sycl.hpp>
|
|
#include <iostream>
|
|
using namespace sycl;
|
|
|
|
constexpr int N = 10;
|
|
|
|
int main() {
|
|
queue my_gpu_queue( gpu_selector{} );
|
|
|
|
std::cout << "Selected GPU device: " <<
|
|
my_gpu_queue.get_device().get_info<info::device::name>() << "\n";
|
|
|
|
int *host_mem = malloc_host<int>(N, my_gpu_queue);
|
|
int *device_mem = malloc_device<int>(N, my_gpu_queue);
|
|
|
|
// Init CPU data
|
|
for(int i = 0; i < N; i++) {
|
|
host_mem[i] = i;
|
|
}
|
|
|
|
// Copy from host(CPU) to device(GPU)
|
|
my_gpu_queue.memcpy(device_mem, host_mem, N * sizeof(int)).wait();
|
|
|
|
// submit the content to the queue for execution
|
|
my_gpu_queue.submit([&](handler& h) {
|
|
|
|
// Parallel Computation
|
|
h.parallel_for(range{N}, [=](id<1> item) {
|
|
device_mem[item] *= 2;
|
|
});
|
|
|
|
});
|
|
|
|
// wait the computation done
|
|
my_gpu_queue.wait();
|
|
|
|
// Copy back from GPU to CPU
|
|
my_gpu_queue.memcpy(host_mem, device_mem, N * sizeof(int)).wait();
|
|
|
|
printf("\nData Result\n");
|
|
for(int i = 0; i < N; i++) {
|
|
printf("%d, ", host_mem[i]);
|
|
}
|
|
printf("\nTask Done!\n");
|
|
|
|
free(host_mem, my_gpu_queue);
|
|
free(device_mem, my_gpu_queue);
|
|
|
|
return 0;
|
|
}
|
|
|