[easy] Fix logspam on PiecewiseBackend errors (#17138)

Signed-off-by: rzou <zou3519@gmail.com>
This commit is contained in:
Richard Zou 2025-05-06 02:46:11 -04:00 committed by GitHub
parent 999328be0d
commit edbf2d609e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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 = [