Skip to content

Commit e3768c5

Browse files
authored
[Executor] Fix bug of logger.debug (#2778)
1 parent 1f28bdf commit e3768c5

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

fastdeploy/model_executor/graph_optimization/cudagraph_piecewise_backend.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,9 +75,9 @@ def __call__(self, **kwargs):
7575
ids_remove_padding: paddle.Tensor = kwargs["ids_remove_padding"]
7676
batch_size = ids_remove_padding.shape[0]
7777
padding_batch_size = self.batch_size_to_captured_size[batch_size]
78-
logger.debug(
78+
logger.debug((
7979
f"[CUDA GRAPH] The actual batch size obtained by CUDAGraph is :{batch_size}, ",
80-
f"The padded batch size is :{padding_batch_size}")
80+
f"The padded batch size is :{padding_batch_size}"))
8181

8282
entry = self.concrete_size_entries.get(padding_batch_size)
8383
assert entry is not None, f"Batch size:{padding_batch_size} is not in cuda graph capture list."
@@ -96,10 +96,10 @@ def __call__(self, **kwargs):
9696
for n in range(entry.num_finished_warmup, self.warm_up_size):
9797
entry.num_finished_warmup += 1
9898
entry.runnable(**kwargs)
99-
logger.debug(
99+
logger.debug((
100100
"[CUDA GRAPH] Warm up for batch size ",
101101
f"{padding_batch_size}, finished ({n+1}/{entry.num_finished_warmup}) times"
102-
)
102+
))
103103

104104
# Store input addresses for debug
105105
input_addresses = [

0 commit comments

Comments
 (0)