Skip to content

Commit 9a0e07e

Browse files
committed
fix CUDA build
1 parent 0e615bc commit 9a0e07e

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

include/nvexec/stream/let_xxx.cuh

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -306,34 +306,35 @@ namespace nv::execution::_strm
306306
STDEXEC_HOST_DEVICE_DEDUCTION_GUIDE
307307
let_sender(Sender, Fun, SetTag) -> let_sender<Sender, Fun, SetTag>;
308308

309-
template <class SetTag>
309+
template <class LetTag>
310310
struct _transform_let_sender
311311
{
312+
using _set_tag = __t<LetTag>;
313+
312314
template <class Env, class Fun, class Sender>
313315
auto operator()(Env const &, __ignore, Fun fn, Sender&& sndr) const
314316
{
315317
if constexpr (stream_completing_sender<Sender, Env>)
316318
{
317-
return let_sender{static_cast<Sender&&>(sndr), static_cast<Fun&&>(fn), SetTag{}};
319+
return let_sender{static_cast<Sender&&>(sndr), static_cast<Fun&&>(fn), _set_tag{}};
318320
}
319321
else
320322
{
321-
using _let_t = decltype(STDEXEC::__let::__let_from_set<SetTag>);
322-
return _strm::_no_stream_scheduler_in_env<_let_t, Sender, Env>();
323+
return _strm::_no_stream_scheduler_in_env<LetTag, Sender, Env>();
323324
}
324325
}
325326
};
326327

327328
template <>
328-
struct transform_sender_for<STDEXEC::let_value_t> : _transform_let_sender<set_value_t>
329+
struct transform_sender_for<STDEXEC::let_value_t> : _transform_let_sender<STDEXEC::let_value_t>
329330
{};
330331

331332
template <>
332-
struct transform_sender_for<STDEXEC::let_error_t> : _transform_let_sender<set_error_t>
333+
struct transform_sender_for<STDEXEC::let_error_t> : _transform_let_sender<STDEXEC::let_error_t>
333334
{};
334335

335336
template <>
336-
struct transform_sender_for<STDEXEC::let_stopped_t> : _transform_let_sender<set_stopped_t>
337+
struct transform_sender_for<STDEXEC::let_stopped_t> : _transform_let_sender<STDEXEC::let_stopped_t>
337338
{};
338339
} // namespace nv::execution::_strm
339340

0 commit comments

Comments
 (0)