Skip to content

Commit 73e88e4

Browse files
committed
Remove redundant distributed amax unit test
Signed-off-by: realAsma <akuriparambi@nvidia.com>
1 parent daf150a commit 73e88e4

1 file changed

Lines changed: 0 additions & 17 deletions

File tree

tests/unit/torch/quantization/test_dist.py

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -13,9 +13,6 @@
1313
# See the License for the specific language governing permissions and
1414
# limitations under the License.
1515

16-
from unittest.mock import Mock, patch
17-
18-
import pytest
1916
import torch
2017
import torch.distributed as dist
2118
from _test_utils.torch.distributed.utils import spawn_multiprocess_job
@@ -48,17 +45,3 @@ def forward_loop(model):
4845

4946
def test_data_parallel(skip_on_windows):
5047
spawn_multiprocess_job(2, _test_data_parallel_helper, backend="gloo")
51-
52-
53-
def test_sync_amax_propagates_collective_errors():
54-
quantizer = TensorQuantizer()
55-
quantizer._amax = torch.tensor(1.0)
56-
parallel_group = Mock()
57-
parallel_group.is_initialized.return_value = True
58-
59-
with (
60-
patch.object(dist, "get_backend", return_value=dist.Backend.GLOO),
61-
patch.object(dist, "all_reduce", side_effect=RuntimeError("out of memory")),
62-
pytest.raises(RuntimeError, match="out of memory"),
63-
):
64-
quantizer.sync_amax_across_distributed_group(parallel_group)

0 commit comments

Comments
 (0)