taskflow/examples/sycl/sycl_rebind.cpp at master · ModuleWorks/taskflow · GitHub

50 lines (33 loc) · 1.07 KB

// This program demonstrates how to rebind syclFlow tasks

// to different device operations.

#include <taskflow/syclflow.hpp>

int main() {

size_t N = 10000;

sycl::queue queue;

auto data = sycl::malloc_shared<int>(N, queue);

tf::syclFlow syclflow(queue);

// fill data with -1

std::cout << "filling data with -1 ...\n";

tf::syclTask task = syclflow.fill(data, -1, N);

syclflow.offload();

for(size_t i=0; i<N; i++) {

if(data[i] != -1) {

throw std::runtime_error("unexpected result after fill");

}

std::cout << "correct result after fill\n";

// rebind the task to for-each task setting each element to 100

// You can rebind a syclTask to any other task type.

std::cout << "rebind task to for_each task setting each element to 100\n";

syclflow.for_each(

task, data, data+N, [](int& i){ i = 100; }

);

syclflow.offload();

for(size_t i=0; i<N; i++) {

if(data[i] != 100) {

throw std::runtime_error("unexpected result after for_each");

}

std::cout << "correct result after updating for_each\n";

return 0;

}

Expand file tree