From 307811d22ceac77ddbf1711838999a5e9dafee14 Mon Sep 17 00:00:00 2001
From: Dante Gama Dessavre <danteg@nvidia.com>
Date: Wed, 12 Feb 2025 19:29:08 -0600
Subject: [PATCH] FIX set n_items to 1 in FIL to avoid the BATCH_TREE_REORG
 issue on H100

---
 cpp/src/fil/fil.cu | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cpp/src/fil/fil.cu b/cpp/src/fil/fil.cu
index 44ea2a3c23..0345968a35 100644
--- a/cpp/src/fil/fil.cu
+++ b/cpp/src/fil/fil.cu
@@ -116,7 +116,9 @@ struct forest {
       // if n_items was not provided, try from 1 to MAX_N_ITEMS. Otherwise, use as-is.
       int min_n_items = ssp.n_items == 0 ? 1 : ssp.n_items;
       int max_n_items =
-        ssp.n_items == 0 ? (algo_ == algo_t::BATCH_TREE_REORG ? MAX_N_ITEMS : 1) : ssp.n_items;
+        // we force this to 1 to avoid running into the BATCH_TREE_REORG issue
+        // ssp.n_items == 0 ? (algo_ == algo_t::BATCH_TREE_REORG ? MAX_N_ITEMS : 1) : ssp.n_items;
+        ssp.n_items == 0 ? 1 : ssp.n_items;
       for (bool cols_in_shmem : {false, true}) {
         ssp.cols_in_shmem = cols_in_shmem;
         for (ssp.n_items = min_n_items; ssp.n_items <= max_n_items; ++ssp.n_items) {