From c62efcbb709d35ddc5315225b880bad490d15cf8 Mon Sep 17 00:00:00 2001
From: Bernhard Manfred Gruber <bernhardmgruber@gmail.com>
Date: Thu, 6 Feb 2025 13:08:48 +0100
Subject: [PATCH] Guard i128

---
 cub/cub/device/dispatch/tuning/tuning_select_if.cuh | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/cub/cub/device/dispatch/tuning/tuning_select_if.cuh b/cub/cub/device/dispatch/tuning/tuning_select_if.cuh
index 4017aee53e5..2bd4e298a89 100644
--- a/cub/cub/device/dispatch/tuning/tuning_select_if.cuh
+++ b/cub/cub/device/dispatch/tuning/tuning_select_if.cuh
@@ -1152,8 +1152,8 @@ struct sm100_tuning<Input,
   using delay_constructor                            = exponential_backoff_constructor_t<0, 1200>;
 };
 
-// because we introduced cases for when offset is I64 this leads to regressions if not
-// defaulted explicitly
+#if CUB_IS_INT128_ENABLED
+// because we introduced cases for when offset is I64 this leads to regressions if not defaulted explicitly
 template <distinct_partitions DistinctPartitions>
 struct sm100_tuning<__int128_t,
                     flagged::no,
@@ -1166,6 +1166,7 @@ struct sm100_tuning<__int128_t,
     : sm90_tuning<__int128_t, flagged::no, keep_rejects::yes, offset_size::_4, primitive::no, input_size::_16>
 // ^^^^^ this base is wrong and leads to regressions ^^^^^
 {};
+#endif // CUB_IS_INT128_ENABLED
 
 // partition::flagged
 template <class Input>