Skip to content

Commit edbf2d6

Browse files
authored
[easy] Fix logspam on PiecewiseBackend errors (#17138)
Signed-off-by: rzou <[email protected]>
1 parent 999328b commit edbf2d6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

vllm/compilation/backends.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -241,6 +241,8 @@ def __init__(self, module: torch.fx.GraphModule,
241241
self.graph_pool = graph_pool
242242
self.vllm_config = vllm_config
243243
self.vllm_backend = vllm_backend
244+
# When True, it annoyingly dumps the torch.fx.Graph on errors.
245+
self.extra_traceback = False
244246

245247
def run(self, *args):
246248
fake_args = [

0 commit comments

Comments
 (0)