Skip to content

Commit 026f75b

Browse files
lucylqfacebook-github-bot
authored andcommitted
linter
Summary: ^ Differential Revision: D67551054
1 parent c09d39b commit 026f75b

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

exir/emit/test/test_emit.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -335,7 +335,8 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
335335

336336
program = (
337337
to_edge(export(f, (torch.randn(2, 3, 5),), strict=True))
338-
.to_executorch().executorch_program
338+
.to_executorch()
339+
.executorch_program
339340
)
340341
exir.print_program.pretty_print(program)
341342

@@ -522,8 +523,7 @@ def forward(self, x: torch.Tensor, n: torch.Tensor) -> torch.Tensor:
522523
program = (
523524
to_edge(export(f, (x, x), strict=True))
524525
# .to_edge(self.compile_config) # TODO(larryliu): fix cat
525-
.to_executorch()
526-
.executorch_program
526+
.to_executorch().executorch_program
527527
)
528528

529529
self.assertEqual(len(program.execution_plan[0].chains[0].instructions), 1)

exir/tests/test_quant_fusion_pass.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,9 @@ def forward(self, x, y):
163163
exactly=True,
164164
).check("executorch_exir_dialects_edge__ops_aten_slice_copy_Tensor").check(
165165
"executorch_exir_dialects_edge__ops_quantized_decomposed_quantize_per_tensor_default"
166-
).check("executorch_exir_dialects_edge__ops_aten_slice_copy_Tensor").check(
166+
).check(
167+
"executorch_exir_dialects_edge__ops_aten_slice_copy_Tensor"
168+
).check(
167169
"executorch_exir_dialects_edge__ops_quantized_decomposed_add_default"
168170
).check(
169171
"executorch_exir_dialects_edge__ops_quantized_decomposed_dequantize_per_tensor_default"
@@ -175,9 +177,7 @@ def forward(self, x, y):
175177
# check that we are using out variant of add and slice_copy
176178
FileCheck().check("torch.ops.quantized_decomposed.add.out").check(
177179
"torch.ops.aten.slice_copy.Tensor_out"
178-
).run(
179-
m.exported_program().graph_module.code
180-
)
180+
).run(m.exported_program().graph_module.code)
181181

182182
def test_cat(self) -> None:
183183
class M(torch.nn.Module):

0 commit comments

Comments
 (0)