From edbf2d609e6a23a105cc109844a187d72727b014 Mon Sep 17 00:00:00 2001
From: Richard Zou <zou3519@users.noreply.github.com>
Date: Tue, 6 May 2025 02:46:11 -0400
Subject: [PATCH] [easy] Fix logspam on PiecewiseBackend errors (#17138)

Signed-off-by: rzou <zou3519@gmail.com>
---
 vllm/compilation/backends.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/vllm/compilation/backends.py b/vllm/compilation/backends.py
index fcaf4a0f987ab..a1ff5fb1196b8 100644
--- a/vllm/compilation/backends.py
+++ b/vllm/compilation/backends.py
@@ -241,6 +241,8 @@ class PiecewiseCompileInterpreter(torch.fx.Interpreter):
         self.graph_pool = graph_pool
         self.vllm_config = vllm_config
         self.vllm_backend = vllm_backend
+        # When True, it annoyingly dumps the torch.fx.Graph on errors.
+        self.extra_traceback = False
 
     def run(self, *args):
         fake_args = [