frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Move pytorch distributed tests to separate folder for contbuild. (#30445)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/30445

Create distributed and rpc directories under caffe/test for better management
of unit tests.

Differential Revision: D18702786

fbshipit-source-id: e9daeed0cfb846ef68806f6decfcb57c0e0e3606
This commit is contained in:
Pritam Damania
2020-01-22 21:05:28 -08:00
committed by Facebook Github Bot
parent e735395fc6
commit f050b16dd9
96 changed files with 362 additions and 349 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_quantization.py
View File

@ -22,8 +22,8 @@ from torch.quantization import default_per_channel_weight_observer
from torch.quantization import default_per_channel_qconfig
from torch.quantization._quantize_script import quantize_script
from common_utils import run_tests
from common_quantization import QuantizationTestCase, \
from torch.testing._internal.common_utils import run_tests
from torch.testing._internal.common_quantization import QuantizationTestCase, \
AnnotatedSingleLayerLinearModel, SingleLayerLinearModel, \
AnnotatedConvModel, ConvModel, \
AnnotatedConvBnModel, ConvBnModel, \
@ -35,12 +35,12 @@ from common_quantization import QuantizationTestCase, \
TwoLayerLinearModel, NestedModel, ResNetBase, LSTMDynamicModel, \
ModelWithNoQconfigPropagation
from common_quantization import AnnotatedTwoLayerLinearModel, AnnotatedNestedModel, \
from torch.testing._internal.common_quantization import AnnotatedTwoLayerLinearModel, AnnotatedNestedModel, \
AnnotatedSubNestedModel, AnnotatedCustomConfigNestedModel
from hypothesis import given
from hypothesis import strategies as st
import hypothesis_utils as hu
import torch.testing._internal.hypothesis_utils as hu
hu.assert_deadline_disabled()
import io
import copy