From a943b9015ec86a41ad39bfddb7081e0b6893a4ac Mon Sep 17 00:00:00 2001 From: heheda Date: Sat, 11 May 2024 22:58:10 +0800 Subject: [PATCH] remove a backend --- frontend/fx_graph.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/frontend/fx_graph.py b/frontend/fx_graph.py index fcdbab2e07dc..246eb3b730d4 100644 --- a/frontend/fx_graph.py +++ b/frontend/fx_graph.py @@ -106,11 +106,6 @@ def eager_due_to_inductor_bug(node: torch.fx.Node) -> bool: with torch.no_grad(): script_model = torch.jit.script(model, real_inputs) return script_model - elif backend == 'nnf': - model_name = config.get_config('model_name') - from fx2onnx import compile_with_nnf # type: ignore[import] - real_inputs = generate_real_tensors(example_inputs) - return compile_with_nnf(model_name, gm, real_inputs) else: raise RuntimeError(f"Unknown backend: {backend}")