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 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 rparams = nullptr; if (use_tma) {