format

Author: ZhangLirong-amd

atom/model_engine/engine_core.py

@@ -407,7 +407,6 @@ def _execute_dummy_prefill(self, num_tokens: int, num_reqs: int = 1):
             "dummy_prefill_execution", num_tokens, num_reqs, wait_out=True
         )
 
-
     def _sync_dp_state(
         self,
         local_is_prefill: bool,

atom/utils/dbo/__init__.py

@@ -1,7 +1,11 @@
 # SPDX-License-Identifier: MIT
 # Copyright (C) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
 
-from .ubatch_splitting import UBatchSlice, maybe_create_ubatch_slices, split_attn_metadata
+from .ubatch_splitting import (
+    UBatchSlice,
+    maybe_create_ubatch_slices,
+    split_attn_metadata,
+)
 from .ubatch_wrapper import UBatchWrapper
 from .ubatching import (
     TBOContext,

atom/utils/dbo/ubatch_wrapper.py

@@ -92,7 +92,11 @@ def _run_ubatches(
             else:
                 padded_bs = full_graph_bs - (full_graph_bs // N) * (N - 1)
             ub_ctx = self._make_ubatch_context(
-                original_ctx, ub_slice, padded_bs, i, ub_num_reqs,
+                original_ctx,
+                ub_slice,
+                padded_bs,
+                i,
+                ub_num_reqs,
             )
             forward_contexts.append(ub_ctx)
             ub_inputs.append(