remove torchao dependency (#2599)

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
pytorch · Mar 23, 2024 · a9dc341 · a9dc341
1 parent 4c99948
commit a9dc341
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 5 deletions.
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",
 ]