import more quantized decomposed ops into edge dialect (#2454)

larryliu0820 · facebook-github-bot · commit 5b8fc4c72203 · 2024-03-15T09:35:29.000-07:00
Summary:

As titled

Reviewed By: digantdesai

Differential Revision: D54930682
diff --git a/exir/passes/TARGETS b/exir/passes/TARGETS
@@ -127,6 +127,7 @@ python_library(
         "//caffe2:torch",
         "//executorch/exir:pass_base",
         "//executorch/exir/dialects:lib",
+        "//pytorch/ao:torchao",
     ],
 )
 
diff --git a/exir/passes/_quant_patterns_and_replacements.py b/exir/passes/_quant_patterns_and_replacements.py
@@ -15,7 +15,7 @@
     should_lower_to_edge,
 )
 from torch import fx
-from torch.ao.quantization.fx._decomposed import quantized_decomposed_lib
+from torchao.quantization.quant_primitives import quantized_decomposed_lib
 
 
 __all__ = [
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,6 +21,7 @@ dependencies=[
   "sympy",
   "tabulate",
   "tomli",
+  "torchao-nightly",
   "zstd",
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -127,6 +127,7 @@ python_library(`
`127`	`127`	`"//caffe2:torch",`
`128`	`128`	`"//executorch/exir:pass_base",`
`129`	`129`	`"//executorch/exir/dialects:lib",`
	`130`	`+ "//pytorch/ao:torchao",`
`130`	`131`	`],`
`131`	`132`	`)`
`132`	`133`
Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@`
`15`	`15`	`should_lower_to_edge,`
`16`	`16`	`)`
`17`	`17`	`from torch import fx`
`18`		`-from torch.ao.quantization.fx._decomposed import quantized_decomposed_lib`
	`18`	`+from torchao.quantization.quant_primitives import quantized_decomposed_lib`
`19`	`19`
`20`	`20`
`21`	`21`	`__all__ = [`