Restrict active histogram channels to channel count

Fixes: #1792
NVIDIA · May 31, 2024 · 67c7d32 · 67c7d32
1 parent 8b5bf66
commit 67c7d32
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cub/cub/device/dispatch/dispatch_histogram.cuh b/cub/cub/device/dispatch/dispatch_histogram.cuh
@@ -563,7 +563,8 @@ template <int NUM_CHANNELS,
 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:
   //---------------------------------------------------------------------