From 2681b0694f68f63c621e611f1e6e39114b4a6442 Mon Sep 17 00:00:00 2001 From: Zhengxu Chen Date: Thu, 6 Feb 2025 09:52:59 -0800 Subject: [PATCH] Fix unittests failures introduced from D69212362 (#8278) Summary: There'll be some minor test failures caused by D69212362, fixing these before the source diff get landed. Reviewed By: larryliu0820 Differential Revision: D69251115 --- exir/emit/test/test_emit.py | 7 +++++-- exir/program/test/test_program.py | 3 --- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/exir/emit/test/test_emit.py b/exir/emit/test/test_emit.py index c2561d06487..a6ddf1c82d2 100644 --- a/exir/emit/test/test_emit.py +++ b/exir/emit/test/test_emit.py @@ -856,7 +856,7 @@ def test_non_const_buffer_sizes(self) -> None: class Add(torch.nn.Module): def forward(self, x: torch.Tensor) -> torch.Tensor: b = 3 + 1 - return x + b + return x + torch.tensor(b) f = Add() @@ -1325,7 +1325,10 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: # Find the multiplication node in the graph that was emitted. for node in program_mul.exported_program().graph.nodes: - if node.target == torch.ops.aten.mul.out: + if ( + node.target == torch.ops.aten.mul.out + or node.target == torch.ops.aten.mul.Scalar_out + ): break self.assertIsNotNone(node) diff --git a/exir/program/test/test_program.py b/exir/program/test/test_program.py index d96e8a24143..fca8bd2212f 100644 --- a/exir/program/test/test_program.py +++ b/exir/program/test/test_program.py @@ -292,9 +292,6 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: ).exported_program() for node in ep.graph.nodes: self.assertNotEqual(node.op, "get_attr") - self.assertEqual( - len([node for node in ep.graph.nodes if node.op == "placeholder"]), 2 - ) def test_constraint_present_after_dce(self): import executorch.exir as exir