fix old gcc version check (NVIDIA#2904)

trxcllnt · Nov 23, 2024 · b04fef5 · b04fef5
1 parent 820a362
commit b04fef5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cub/cub/device/dispatch/kernels/for_each_in_extents_kernel.cuh b/cub/cub/device/dispatch/kernels/for_each_in_extents_kernel.cuh
@@ -139,7 +139,7 @@ computation(IndexType id, IndexType, Func func, ExtendType, FastDivModArrayType,
  **********************************************************************************************************************/
 
 // GCC6/7/8/9 raises unused parameter warning
-#  if defined(_CCCL_COMPILER_GCC) && _CCCL_GCC_VERSION < 100000
+#  if _CCCL_COMPILER(GCC, <, 10)
 #    define _CUB_UNUSED_ATTRIBUTE __attribute__((unused))
 #  else
 #    define _CUB_UNUSED_ATTRIBUTE