Skip to content

Commit a886b09

Browse files
committed
use cuda::std::span instead of std::span in nvexec bulk test
1 parent c7ae38c commit a886b09

1 file changed

Lines changed: 7 additions & 5 deletions

File tree

test/nvexec/bulk.cpp

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
#include "nvexec/stream_context.cuh"
55
#include "common.cuh"
66

7-
#include <span>
7+
#include <cuda/std/span>
88

99
namespace 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

0 commit comments

Comments
 (0)