Fix nvexec sync_wait scheduler query

Author: fallintoplace

include/nvexec/stream/sync_wait.cuh

@@ -36,7 +36,7 @@ namespace nv::execution::_strm
     struct env
     {
       template <
-        __one_of<get_start_scheduler_t, get_start_scheduler_t, get_delegation_scheduler_t> _Query>
+        __one_of<get_scheduler_t, get_start_scheduler_t, get_delegation_scheduler_t> _Query>
       [[nodiscard]]
       constexpr auto query(_Query) const noexcept -> run_loop::scheduler
       {

test/nvexec/continues_on.cpp

@@ -32,4 +32,15 @@ namespace
 
     STDEXEC::sync_wait(std::move(sndr));
   }
+
+  TEST_CASE("nvexec sync_wait provides a scheduler", "[cuda][stream][sync_wait]")
+  {
+    nvexec::stream_context ctx;
+
+    auto sndr = STDEXEC::get_scheduler() | STDEXEC::continues_on(ctx.get_scheduler());
+
+    auto result = STDEXEC::sync_wait(std::move(sndr));
+
+    REQUIRE(result.has_value());
+  }
 }  // namespace