pytorch/torch/testing/_internal
Philip Meier f4967d68f5 make torch.testing asserts importable (#54769)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54769

Follow-up to #53820. This

- makes the `asserts.py` module private as per suggestion from rgommers in https://github.com/pytorch/pytorch/pull/53820#issuecomment-802661387. With this the functions should only be accessible through `torch.testing`, giving us the option the change the underlying structure later.
- moves the code from `torch/testing/__init__.py` to `torch/testing/_core.py` (happy to accept other name suggestions). Otherwise we can't import the new `_asserts.py` in `torch/testing/__init__.py` due to circular imports.

Test Plan: Imported from OSS

Reviewed By: mrshenli

Differential Revision: D27438451

Pulled By: mruberry

fbshipit-source-id: c7292b4d5709185b42b4aac8016648562688040e
2021-04-07 23:53:02 -07:00
..
codegen Enable CUDA Fuser for ROCm (#45965) 2020-10-08 10:41:56 -07:00
data
distributed [Gradient Compression] Update the default value of start_powerSGD_iter and update the docstring (#55272) 2021-04-06 01:27:29 -07:00
generated
test_module Drop unused imports (#49972) 2021-01-13 12:26:17 -08:00
__init__.py
autocast_test_lists.py [Relanding] Implemented torch.linalg.multi_dot (#52859) 2021-04-01 04:49:05 -07:00
common_cuda.py Fixes new tf32 failures in test_nn.py (#52871) 2021-03-24 21:53:33 -07:00
common_device_type.py [ROCm] if TEST_WITH_ROCM, only instantiate GPU device tests (#55069) 2021-04-07 09:12:09 -07:00
common_distributed.py make torch.testing asserts importable (#54769) 2021-04-07 23:53:02 -07:00
common_jit.py Fixed OpInfo jit tests failing for TensorList inputs (#54954) 2021-04-01 07:09:07 -07:00
common_methods_invocations.py make torch.testing asserts importable (#54769) 2021-04-07 23:53:02 -07:00
common_nn.py Lowering NLLLoss/CrossEntropyLoss to ATen code (#53789) 2021-03-26 07:31:08 -07:00
common_quantization.py ns for fx: create subgraph type (#54253) 2021-03-25 22:35:34 -07:00
common_quantized.py [Pytorch, Sparsity] Integrate sparse qnnpack operator in framework (#52377) 2021-02-17 18:25:13 -08:00
common_utils.py make torch.testing asserts importable (#54769) 2021-04-07 23:53:02 -07:00
dist_utils.py Add some missing types to torch (#55184) 2021-04-05 21:44:47 -07:00
expecttest.py Fix bug in self.assertExpectedInline (#55149) 2021-04-05 06:37:36 -07:00
framework_utils.py Use shard file if present, improve functions used for sharding (#54210) 2021-03-18 13:25:51 -07:00
hypothesis_utils.py [quant] conv_transpose1d_prepack / conv_transpose1d_unpack (#40360) 2020-09-11 04:12:28 -07:00
jit_metaprogramming_utils.py Fixed OpInfo jit tests failing for TensorList inputs (#54954) 2021-04-01 07:09:07 -07:00
jit_utils.py [package] fix mangling issues with TorchScript (#54915) 2021-03-31 00:58:05 -07:00
te_utils.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00