Skip to content

Commit

Permalink
remove a backend
Browse files Browse the repository at this point in the history
  • Loading branch information
heheda12345 committed May 11, 2024
1 parent b021011 commit a943b90
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions frontend/fx_graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}")

Expand Down

0 comments on commit a943b90

Please sign in to comment.