test

NVIDIA · Dec 10, 2024 · b863f58 · b863f58
1 parent 96ac8e8
commit b863f58
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cub/cub/device/dispatch/dispatch_merge_sort.cuh b/cub/cub/device/dispatch/dispatch_merge_sort.cuh
@@ -226,6 +226,7 @@ CUB_DETAIL_KERNEL_ATTRIBUTES void DeviceMergeSortPartitionKernel(
 {
   OffsetT partition_idx = blockDim.x * blockIdx.x + threadIdx.x;
 
+  NV_IF_TARGET(NV_PROVIDES_SM_90, cudaTriggerProgrammaticLaunchCompletion();)
   if (partition_idx < num_partitions)
   {
     AgentPartition<KeyIteratorT, OffsetT, CompareOpT, KeyT> agent(