From ca55689b43e0277bc1c58d69b0b631204e897324 Mon Sep 17 00:00:00 2001 From: "Tugsbayasgalan (Tugsuu) Manlaibaatar" Date: Tue, 5 Nov 2024 14:26:07 -0800 Subject: [PATCH] Remove IR check after aten in arm Summary: Export is switching to non-functional training IR, as a result, some checks after second export can fail due to difference in the IR. We remove those checks as this IR doesn't really matter to backends Differential Revision: D65504497 --- backends/arm/test/ops/test_batch_norm.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/backends/arm/test/ops/test_batch_norm.py b/backends/arm/test/ops/test_batch_norm.py index 4935e910d6d..bfe1146a904 100644 --- a/backends/arm/test/ops/test_batch_norm.py +++ b/backends/arm/test/ops/test_batch_norm.py @@ -536,9 +536,6 @@ def _test_batchnorm2d_tosa_MI_pipeline( compile_spec=common.get_tosa_compile_spec(), ) .export() - .check_count( - {"torch.ops.aten._native_batch_norm_legit_no_training.default": 1} - ) .check_not(["torch.ops.quantized_decomposed"]) .to_edge() .check_count(