@@ -101,7 +101,7 @@ def _assert_disallow_flags(
101101 ), f"Node '{ node_name } ': expected DISALLOW_TFA_META_KEY={ expected_flag } , got { actual_flag } "
102102
103103
104- def test_disallow_tfa_for_skipped_module ():
104+ def test_disallow_tfa_for_skipped_module_no_target ():
105105 """Ensure a softmax skipped for quantization is not decomposed and that its
106106 node has `disallow_tfa` set.
107107 """
@@ -133,7 +133,7 @@ def example_inputs(self) -> tuple[torch.Tensor, ...]:
133133 )
134134
135135
136- def test_disallow_tfa_for_two_skipped_modules ():
136+ def test_disallow_tfa_for_two_skipped_modules_no_target ():
137137 """Ensure a softmax and linear skipped for quantization are not decomposed
138138 and have their `disallow_tfa` set.
139139 """
@@ -156,7 +156,7 @@ def test_disallow_tfa_for_two_skipped_modules():
156156 )
157157
158158
159- def test_disallow_tfa_with_global_none_and_one_quantized_module ():
159+ def test_disallow_tfa_with_global_none_and_one_quantized_module_no_target ():
160160 """Ensure that with a global None quantization config, only the linear
161161 module (with its own quantization config) is quantized, and that the
162162 other nodes have `disallow_tfa` set.
@@ -180,7 +180,7 @@ def test_disallow_tfa_with_global_none_and_one_quantized_module():
180180 )
181181
182182
183- def test_disallow_tfa_for_submodule_by_name ():
183+ def test_disallow_tfa_for_submodule_by_name_no_target ():
184184 """Ensure submodules can be skipped for quantization by name and have their
185185 nodes marked as disallowed for TFA.
186186 """
@@ -203,7 +203,7 @@ def test_disallow_tfa_for_submodule_by_name():
203203 )
204204
205205
206- def test_disallow_tfa_name_config_contradicts_type_config ():
206+ def test_disallow_tfa_name_config_contradicts_type_config_no_target ():
207207 """Ensure that module name configs take precedence over module type configs
208208 when they contradict each other.
209209 """
0 commit comments