Arm backend: avg_pool2d test not assume availibility of tosa_ref_model

Differential Revision: D77278468

Pull Request resolved: pytorch#11951

Author: digantdesai

backends/arm/test/ops/test_avg_pool2d.py

@@ -8,6 +8,8 @@
 
 from typing import Tuple
 
+import conftest
+
 import torch
 
 from executorch.backends.arm.test import common
@@ -119,6 +121,7 @@ def test_avg_pool2d_tosa_MI(test_module):
         input_tensor,
         aten_op,
         exir_op,
+        run_on_tosa_ref_model=conftest.is_option_enabled("tosa_ref_model"),
     )
     pipeline.run()
 
@@ -132,6 +135,7 @@ def test_avg_pool2d_tosa_BI(test_module):
         input_tensor,
         aten_op,
         exir_op,
+        run_on_tosa_ref_model=conftest.is_option_enabled("tosa_ref_model"),
     )
     pipeline.run()