the streams created by cudax::stream should not synchronize with th…

…e null stream (#3167)
NVIDIA · Dec 14, 2024 · 0aa0b37 · 0aa0b37
1 parent 3a7f2cd
commit 0aa0b37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cudax/include/cuda/experimental/__stream/stream.cuh b/cudax/include/cuda/experimental/__stream/stream.cuh
@@ -48,7 +48,7 @@ struct stream : stream_ref
   {
     [[maybe_unused]] __ensure_current_device __dev_setter(__dev);
     _CCCL_TRY_CUDA_API(
-      ::cudaStreamCreateWithPriority, "Failed to create a stream", &__stream, cudaStreamDefault, __priority);
+      ::cudaStreamCreateWithPriority, "Failed to create a stream", &__stream, cudaStreamNonBlocking, __priority);
   }
 
   //! @brief Constructs a stream on a specified logical device and with specified priority
@@ -61,7 +61,7 @@ struct stream : stream_ref
   {
     [[maybe_unused]] __ensure_current_device __dev_setter(__dev);
     _CCCL_TRY_CUDA_API(
-      ::cudaStreamCreateWithPriority, "Failed to create a stream", &__stream, cudaStreamDefault, __priority);
+      ::cudaStreamCreateWithPriority, "Failed to create a stream", &__stream, cudaStreamNonBlocking, __priority);
   }
 
   //! @brief Constructs a stream on the default device