From 233401ae9fd981bc2c6cb2ea87e623eecafa0ffb Mon Sep 17 00:00:00 2001
From: ZhangLirong <lirzhang@amd.com>
Date: Tue, 7 Jul 2026 21:02:10 +0800
Subject: [PATCH] Revert "[ATOM][Fix] propagate dp_uniform_decode to TBO
 ubatches (#1474)"

This reverts commit 0c7d50f9853c7b10c242e0bb960a5297b2f2a1c8.
---
 atom/utils/tbo/ubatch_wrapper.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/atom/utils/tbo/ubatch_wrapper.py b/atom/utils/tbo/ubatch_wrapper.py
index e9842d1522..b21a7eca36 100644
--- a/atom/utils/tbo/ubatch_wrapper.py
+++ b/atom/utils/tbo/ubatch_wrapper.py
@@ -542,7 +542,6 @@ def _make_ubatch_context(
             batch_size=ub_num_reqs,
             graph_bs=graph_bs,
             is_draft=ctx.context.is_draft,
-            dp_uniform_decode=ctx.context.dp_uniform_decode,
         )
 
         return ForwardContext(