Skip to content

Commit 512b4e0

Browse files
Arm backend: Make get_pass_pipeline_config private
Change-Id: Ifaa0c004b6d1ebeb807902b74ed07bde33b3f5b5 Signed-off-by: Sebastian Larsson <sebastian.larsson@arm.com>
1 parent 0479698 commit 512b4e0

File tree

6 files changed

+5
-21
lines changed

6 files changed

+5
-21
lines changed

backends/arm/_passes/arm_pass_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def configure_skip_passes(
188188
"""
189189
skip_set: set[type] = set()
190190

191-
config = override_config or self.compile_spec.get_pass_pipeline_config()
191+
config = override_config or self.compile_spec._get_pass_pipeline_config()
192192
logger.debug(f"Skip Config: {config}")
193193

194194
match config.softmax:

backends/arm/common/arm_compile_spec.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,7 @@ def _to_list(self):
211211
)
212212
return compile_spec
213213

214-
def get_pass_pipeline_config(self) -> ArmPassPipelineConfig:
214+
def _get_pass_pipeline_config(self) -> ArmPassPipelineConfig:
215215
"""Returns configuration that controls how the Arm pass pipeline should
216216
behave.
217217

backends/arm/test/misc/test_compile_spec.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,14 +32,14 @@ def test_compile_spec_u55_INT():
3232
def test_ethos_u55_defaults_to_stable_softmax_u55_INT():
3333
"""Test that EthosUCompileSpec for U55 defaults to STABLE softmax config."""
3434
compile_spec = EthosUCompileSpec("ethos-u55-128")
35-
pipeline_config = compile_spec.get_pass_pipeline_config()
35+
pipeline_config = compile_spec._get_pass_pipeline_config()
3636
assert pipeline_config.softmax == SoftmaxDecompositionConfig.STABLE
3737

3838

3939
def test_ethos_u85_defaults_to_masked_softmax_u85_INT():
4040
"""Test that EthosUCompileSpec for U85 defaults to MASKED softmax config."""
4141
compile_spec = EthosUCompileSpec("ethos-u85-256")
42-
pipeline_config = compile_spec.get_pass_pipeline_config()
42+
pipeline_config = compile_spec._get_pass_pipeline_config()
4343
assert pipeline_config.softmax == SoftmaxDecompositionConfig.MASKED
4444

4545

backends/arm/tosa/backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -431,7 +431,7 @@ def filter_tosa_compile_specs(
431431
432432
"""
433433

434-
pipeline_config = compile_spec.get_pass_pipeline_config()
434+
pipeline_config = compile_spec._get_pass_pipeline_config()
435435
tosa_compile_spec = TosaCompileSpec(compile_spec.tosa_spec)
436436
tosa_compile_spec.set_pass_pipeline_config(pipeline_config)
437437
return (

docs/source/backends/arm-ethos-u/arm-ethos-u-overview.md

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -78,14 +78,6 @@ def EthosUCompileSpec.get_output_order_workaround(self) -> bool:
7878
```
7979
Gets whether the output order workaround is being applied.
8080

81-
```python
82-
def EthosUCompileSpec.get_pass_pipeline_config(self) -> executorch.backends.arm.common.pipeline_config.ArmPassPipelineConfig:
83-
```
84-
Returns configuration that controls how the Arm pass pipeline should
85-
behave.
86-
87-
Subclasses may override to tweak defaults for specific targets.
88-
8981
```python
9082
def EthosUCompileSpec.set_output_order_workaround(self, output_order_workaround: bool):
9183
```

docs/source/backends/arm-vgf/arm-vgf-overview.md

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -69,14 +69,6 @@ def VgfCompileSpec.get_output_order_workaround(self) -> bool:
6969
```
7070
Gets whether the output order workaround is being applied.
7171

72-
```python
73-
def VgfCompileSpec.get_pass_pipeline_config(self) -> executorch.backends.arm.common.pipeline_config.ArmPassPipelineConfig:
74-
```
75-
Returns configuration that controls how the Arm pass pipeline should
76-
behave.
77-
78-
Subclasses may override to tweak defaults for specific targets.
79-
8072
```python
8173
def VgfCompileSpec.set_output_order_workaround(self, output_order_workaround: bool):
8274
```

0 commit comments

Comments
 (0)