[export] Remove CallSpec (#117671)

angelayi · facebook-github-bot · commit c54cd7191cad · 2024-01-17T13:34:24.000-08:00
Summary: X-link: pytorch/executorch#1618 This is not really being used anywhere Test Plan: CI Differential Revision: D52842563
diff --git a/torch/_export/__init__.py b/torch/_export/__init__.py
@@ -72,9 +72,6 @@
 from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo
 from torch.utils._sympy.value_ranges import ValueRangeError, ValueRanges
 
-from .exported_program import (
-    CallSpec,
-)
 from .passes.add_runtime_assertions_for_constraints_pass import (
     _AddRuntimeAssertionsForInlineConstraintsPass,
 )
diff --git a/torch/_export/exported_program.py b/torch/_export/exported_program.py
@@ -1,11 +1,8 @@
-import dataclasses
-from typing import Optional
 import warnings
 
 
 import torch
 import torch.fx
-import torch.utils._pytree as pytree
 
 
 # TODO(ycao): This is added to avoid breaking existing code temporarily.
@@ -32,13 +29,6 @@
 ]
 
 
-# Information to maintain user calling/returning specs
-@dataclasses.dataclass
-class CallSpec:
-    in_spec: Optional[pytree.TreeSpec]
-    out_spec: Optional[pytree.TreeSpec]
-
-
 def _create_graph_module_for_export(root, graph):
     try:
         gm = torch.fx.GraphModule(root, graph)

Original file line number	Diff line number	Diff line change
`@@ -72,9 +72,6 @@`
`72`	`72`	`from torch.fx.graph import _PyTreeCodeGen, _PyTreeInfo`
`73`	`73`	`from torch.utils._sympy.value_ranges import ValueRangeError, ValueRanges`
`74`	`74`
`75`		`-from .exported_program import (`
`76`		`- CallSpec,`
`77`		`-)`
`78`	`75`	`from .passes.add_runtime_assertions_for_constraints_pass import (`
`79`	`76`	`_AddRuntimeAssertionsForInlineConstraintsPass,`
`80`	`77`	`)`