We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7feef67 commit f482fe7Copy full SHA for f482fe7
backends/arm/test/ops/test_floor_div.py
@@ -69,7 +69,7 @@ def forward(
69
return torch.floor_divide(input=input_, other=other_)
70
71
72
-input_t1 = Tuple[torch.Tensor, torch.Tensor]
+input_t1 = Tuple[torch.Tensor, Union[torch.Tensor, int]]
73
74
75
@common.parametrize("test_data", test_data_suite)
@@ -103,10 +103,12 @@ def test_floor_divide_u55_INT(test_data: input_t1):
103
FloorDivide(),
104
test_data(),
105
aten_ops=FloorDivide.aten_ops_int,
106
- exir_ops=FloorDivide.exir_ops_int,
+ exir_ops=[],
107
run_on_fvp=True,
108
use_to_edge_transform_and_lower=False,
109
)
110
+ pipeline.pop_stage("check_not.exir")
111
+ pipeline.pop_stage("check_count.exir")
112
pipeline.run()
113
114
0 commit comments