Skip to content

Commit

Permalink
Update on "[quant][graphmode][fx] Add support for additional_{fusion/…
Browse files Browse the repository at this point in the history
…quant}_pattern"

Summary:
Allow user to provide additional fusion/quant patterns for fx graph mode

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

Differential Revision: [D24317437](https://our.internmc.facebook.com/intern/diff/D24317437)

[ghstack-poisoned]
  • Loading branch information
jerryzh168 committed Oct 14, 2020
2 parents 0f78cca + 6ea5e9e commit 2c1f44f
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion torch/quantization/quantize_fx.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import torch
from torch.fx import GraphModule # type: ignore
from torch.fx import symbolic_trace # type: ignore
from torch.fx.symbolic_trace import Tracer # type: ignore
from .fx import Fuser # noqa: F401
from .fx import Quantizer # noqa: F401
Expand Down

0 comments on commit 2c1f44f

Please sign in to comment.