Skip to content

Commit 092d86e

Browse files
authored
add llm/export to pytest.ini (#8502)
1 parent adf3956 commit 092d86e

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

extension/llm/export/builder.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@
3434

3535
from executorch.extension.export_util.utils import export_to_edge, save_pte_program
3636

37-
from executorch.extension.llm.export.export_passes import RemoveRedundantPermutes
37+
from executorch.extension.llm.export.export_passes import RemoveRedundantTransposes
3838
from executorch.extension.llm.tokenizer.utils import get_tokenizer
3939
from torch.ao.quantization.quantize_pt2e import convert_pt2e, prepare_pt2e
4040
from torch.ao.quantization.quantizer import Quantizer
@@ -113,7 +113,7 @@ def __init__(
113113
self.calibration_seq_length = calibration_seq_length
114114
self.calibration_data = calibration_data
115115
self.tokenizer_path = tokenizer_path
116-
self.canonical_passes = [RemoveRedundantPermutes()]
116+
self.canonical_passes = [RemoveRedundantTransposes()]
117117

118118
def set_output_dir(self, output_dir: str) -> "LLMEdgeManager":
119119
"""

extension/llm/export/export_passes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def _normalize_dims(tensor: FakeTensor, dim_0: int, dim_1: int):
1919
return dim_0, dim_1
2020

2121

22-
class RemoveRedundantPermutes(ExportPass):
22+
class RemoveRedundantTransposes(ExportPass):
2323
"""
2424
This pass removes redundant transpose nodes in the graph.
2525
It checks if the next node is also a transpose node and if the two transpose nodes undo each other.

pytest.ini

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,7 @@ addopts =
4949
backends/xnnpack/test/serialization
5050
# extension/
5151
extension/llm/modules/test
52+
extension/llm/export
5253
extension/pybindings/test
5354
# Runtime
5455
runtime

0 commit comments

Comments
 (0)