From 5ab4c420db16cb7fdd1bc5d82eb4d245318d1e2d Mon Sep 17 00:00:00 2001
From: tbqh <bzitelli@nvidia.com>
Date: Fri, 20 Feb 2026 08:55:14 -0800
Subject: [PATCH] Allow TMA inner reduction to test CI

---
 csrc/scheduler/reduction.cpp | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/csrc/scheduler/reduction.cpp b/csrc/scheduler/reduction.cpp
index e8617ff4044..7471f882c54 100644
--- a/csrc/scheduler/reduction.cpp
+++ b/csrc/scheduler/reduction.cpp
@@ -250,8 +250,7 @@ std::unique_ptr<HeuristicParams> ReductionScheduler::computeHeuristics(
   auto props = reduction_scheduler_utils::getFusionRuntimeProperties(
       fusion, runtime_info, data_cache);
 
-  bool use_tma =
-      mayUseTma(props) && isOptionEnabled(EnableOption::TmaReduction);
+  bool use_tma = mayUseTma(props);
 
   std::unique_ptr<HeuristicParams> rparams = nullptr;
   if (use_tma) {