Skip to content

Commit f482fe7

Browse files
committed
Arm backend: Add support for floor_divide.default
Signed-off-by: Agrima Khare <[email protected]> Change-Id: I85153b1b245862a107e6469600e69c059477f417
1 parent 7feef67 commit f482fe7

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

backends/arm/test/ops/test_floor_div.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ def forward(
6969
return torch.floor_divide(input=input_, other=other_)
7070

7171

72-
input_t1 = Tuple[torch.Tensor, torch.Tensor]
72+
input_t1 = Tuple[torch.Tensor, Union[torch.Tensor, int]]
7373

7474

7575
@common.parametrize("test_data", test_data_suite)
@@ -103,10 +103,12 @@ def test_floor_divide_u55_INT(test_data: input_t1):
103103
FloorDivide(),
104104
test_data(),
105105
aten_ops=FloorDivide.aten_ops_int,
106-
exir_ops=FloorDivide.exir_ops_int,
106+
exir_ops=[],
107107
run_on_fvp=True,
108108
use_to_edge_transform_and_lower=False,
109109
)
110+
pipeline.pop_stage("check_not.exir")
111+
pipeline.pop_stage("check_count.exir")
110112
pipeline.run()
111113

112114

0 commit comments

Comments
 (0)