Restrict active histogram channels to channel count

Fixes: NVIDIA#1792
bernhardmgruber · May 31, 2024 · d59e9e3 · d59e9e3
1 parent ca0ca5a
commit d59e9e3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/cub/cub/device/dispatch/dispatch_histogram.cuh b/cub/cub/device/dispatch/dispatch_histogram.cuh
@@ -564,6 +564,8 @@ struct DispatchHistogram : SelectedPolicy
 {
   static_assert(NUM_CHANNELS <= 4, "Histograms only support up to 4 channels");
   static_assert(NUM_ACTIVE_CHANNELS <= 4, "Histograms only support up to 4 active channels");
+  static_assert(NUM_ACTIVE_CHANNELS <= NUM_CHANNELS,
+                "Active channels must be at most the number of total channels of the input samples");
 
 public:
   //---------------------------------------------------------------------