File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 44#include " nvexec/stream_context.cuh"
55#include " common.cuh"
66
7- #include < span>
7+ #include < cuda/std/ span>
88
99namespace ex = stdexec;
1010
@@ -170,10 +170,12 @@ namespace {
170170 cudaMallocManaged (&inout, nelems * sizeof (double ));
171171
172172 auto task =
173- stdexec::transfer_just (ctx.get_scheduler (), std::span<double >{inout, nelems})
174- | stdexec::bulk (nelems, [](std::size_t i, std::span<double > out) { out[i] = i; })
175- | stdexec::let_value ([](std::span<double > out) { return stdexec::just (out); })
176- | stdexec::bulk (nelems, [](std::size_t i, std::span<double > out) { out[i] = 2.0 * out[i]; });
173+ stdexec::transfer_just (ctx.get_scheduler (), cuda::std::span<double >{inout, nelems})
174+ | stdexec::bulk (nelems, [](std::size_t i, cuda::std::span<double > out) { out[i] = i; })
175+ | stdexec::let_value ([](cuda::std::span<double > out) { return stdexec::just (out); })
176+ | stdexec::bulk (nelems, [](std::size_t i, cuda::std::span<double > out) {
177+ out[i] = 2.0 * out[i];
178+ });
177179
178180 stdexec::sync_wait (std::move (task)).value ();
179181
You can’t perform that action at this time.
0 commit comments