From b863f582247b1a620982aca47efd7c22a0f3911d Mon Sep 17 00:00:00 2001
From: Bernhard Manfred Gruber <bernhardmgruber@gmail.com>
Date: Tue, 10 Dec 2024 19:02:10 +0100
Subject: [PATCH] test

---
 cub/cub/device/dispatch/dispatch_merge_sort.cuh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cub/cub/device/dispatch/dispatch_merge_sort.cuh b/cub/cub/device/dispatch/dispatch_merge_sort.cuh
index 6689e4cf7dc..e1535e20305 100644
--- 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(