remove torchao dependency (#2599)

guangy10 · facebook-github-bot · commit a9dc341307dc · 2024-03-22T20:06:40.000-07:00
Summary: X-link: pytorch/pytorch#122524 Pull Request resolved: #2599 bypass-github-export-checks bypass-github-pytorch-ci-checks bypass-github-executorch-ci-checks Reviewed By: jerryzh168 Differential Revision: D55263008 fbshipit-source-id: e62d8df96fb64b811cd3ecf4e547bbb9e9fefcc7
diff --git a/.ci/docker/ci_commit_pins/pytorch.txt b/.ci/docker/ci_commit_pins/pytorch.txt
@@ -1 +1 @@
-ef0d470eb372a5e68f6332bea9b1df26fca28b59
+export-D55263008
diff --git a/examples/models/llama2/quantize.py b/examples/models/llama2/quantize.py
@@ -13,7 +13,6 @@
 import torch.nn.functional as F
 from .ops.quantized_ops import *  # noqa
 
-# TODO: move to correct place
 from torchao.quantization.quant_primitives import (
     get_group_qparams_symmetric,
     group_quantize_tensor_symmetric,
diff --git a/exir/passes/TARGETS b/exir/passes/TARGETS
@@ -127,7 +127,6 @@ 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
@@ -14,7 +14,7 @@
     should_lower_to_edge,
 )
 from torch import fx
-from torchao.quantization.quant_primitives import quantized_decomposed_lib
+from torch.ao.quantization.fx._decomposed import quantized_decomposed_lib
 
 
 __all__ = [
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,7 +21,6 @@ dependencies=[
   "sympy",
   "tabulate",
   "tomli",
-  "torchao-nightly",
   "zstd",
 ]
 

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-ef0d470eb372a5e68f6332bea9b1df26fca28b59`
	`1`	`+export-D55263008`
Original file line number	Diff line number	Diff line change
`@@ -127,7 +127,6 @@ python_library(`
`127`	`127`	`"//caffe2:torch",`
`128`	`128`	`"//executorch/exir:pass_base",`
`129`	`129`	`"//executorch/exir/dialects:lib",`
`130`		`- "//pytorch/ao:torchao",`
`131`	`130`	`],`
`132`	`131`	`)`
`133`	`132`
Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@`
`14`	`14`	`should_lower_to_edge,`
`15`	`15`	`)`
`16`	`16`	`from torch import fx`
`17`		`-from torchao.quantization.quant_primitives import quantized_decomposed_lib`
	`17`	`+from torch.ao.quantization.fx._decomposed import quantized_decomposed_lib`
`18`	`18`
`19`	`19`
`20`	`20`	`__all__ = [`
Original file line number	Diff line number	Diff line change
`@@ -21,7 +21,6 @@ dependencies=[`
`21`	`21`	`"sympy",`
`22`	`22`	`"tabulate",`
`23`	`23`	`"tomli",`
`24`		`- "torchao-nightly",`
`25`	`24`	`"zstd",`
`26`	`25`	`]`
`27`	`26`