Skip to content

Commit 3c68b67

Browse files
bdemirbBaris Demir
andauthored
Arm backend: Stabilize MobileNetV3 fp16 TOSA test (#19590)
cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell @rascani Signed-off-by: Baris Demir <baris.demir@arm.com> Co-authored-by: Baris Demir <baris.demir@arm.com>
1 parent c531386 commit 3c68b67

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

backends/arm/test/models/test_mobilenet_v3_arm.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -47,14 +47,17 @@ def test_mv3_tosa_FP():
4747

4848
@pytest.mark.slow
4949
def test_mv3_tosa_FP_fp16():
50-
inputs_fp16 = tuple(t.to(torch.float16) for t in model_inputs)
50+
input_tensor_fp16 = torch.rand(
51+
1, 3, 232, 232, generator=torch.Generator().manual_seed(0)
52+
)
53+
inputs_fp16 = (normalize(input_tensor_fp16).to(torch.float16),)
5154
pipeline = TosaPipelineFP[input_t](
5255
mv3_fp16,
5356
inputs_fp16,
5457
aten_op=[],
5558
exir_op=[],
5659
use_to_edge_transform_and_lower=True,
57-
atol=5e-2,
60+
atol=6e-2,
5861
)
5962
pipeline.run()
6063

0 commit comments

Comments
 (0)