Skip to content

Commit c228578

Browse files
lucylqdigantdesai
andauthored
Arm backend: Run only when tosa_reference_model is availalbe (#12312)
Summary: as title Differential Revision: D78031934 Co-authored-by: Digant Desai <[email protected]>
1 parent defa089 commit c228578

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

backends/arm/test/passes/test_decorate_fp32_to_int32_casting_pass.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from typing import Tuple
77

88
import torch
9-
from executorch.backends.arm.test import common
9+
from executorch.backends.arm.test import common, conftest
1010

1111
from executorch.backends.arm.test.tester.test_pipeline import (
1212
OpNotSupportedPipeline,
@@ -55,6 +55,7 @@ def test_decorate_fp32_to_int32_casting_tosa_MI(test_data: Tuple):
5555
(test_tensor,),
5656
aten_op=[],
5757
exir_op=[],
58+
run_on_tosa_ref_model=conftest.is_option_enabled("tosa_ref_model"),
5859
)
5960
pipeline.run()
6061

0 commit comments

Comments
 (0)