diff --git a/backends/arm/test/passes/test_replace_inf_values_pass.py b/backends/arm/test/passes/test_replace_inf_values_pass.py index d0790584b1e..c951ed1ec1c 100644 --- a/backends/arm/test/passes/test_replace_inf_values_pass.py +++ b/backends/arm/test/passes/test_replace_inf_values_pass.py @@ -1,4 +1,4 @@ -# Copyright 2025 Arm Limited and/or its affiliates. +# Copyright 2025-2026 Arm Limited and/or its affiliates. # # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. @@ -49,7 +49,7 @@ def _get_mask_buffer(graph_module: fx.GraphModule) -> torch.Tensor: return buffers["mask"] -def test_replace_inf_values_pass_clamps_inf_constants(): +def test_replace_inf_and_limit_values_no_target_clamps_inf_constants(): """ Trace a module with infinities, run ReplaceInfAndLimitValuesPass, and expect the buffer and scalar literals to be clamped to ±255 with no infinities left. @@ -66,7 +66,7 @@ def test_replace_inf_values_pass_clamps_inf_constants(): assert sorted(_get_add_constants(result.graph_module)) == [-255, 255] -def test_replace_inf_values_respects_disallowed_nodes(): +def test_replace_inf_and_limit_values_no_target_respects_disallowed_nodes(): """ When nodes opt out of transforms, running the pass in TFA mode should leave the mask buffer untouched while still clamping scalar literals to ±255.