pytorch/test
Akifumi Imanishi aa1fd6b45a Add LazyBatchNormXd (#51548)
Summary:
This PR implements UninitializedBuffer and LazyBatchnormXd based on https://github.com/pytorch/pytorch/issues/44538. (cc. emcastillo and albanD)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/51548

Reviewed By: zhangguanheng66

Differential Revision: D26276903

Pulled By: albanD

fbshipit-source-id: 0ac706974178363f8af075e59b41d5989418922f
2021-02-05 10:27:04 -08:00
..
backward_compatibility make forward AD API private (#51693) 2021-02-04 19:02:29 -08:00
benchmark_utils Revert "Revert D25199264: Enable callgrind collection for C++ snippets" (#48720) 2020-12-02 11:10:11 -08:00
bottleneck_test
cpp [PyTorch Mobile] Support torchbind custom classes in lite interpreter (#51432) 2021-02-03 21:57:19 -08:00
cpp_api_parity Add PixelUnshuffle (#49334) 2020-12-22 20:14:55 -08:00
cpp_extensions Move generator state APIs to ATen (#49589) 2021-01-06 18:26:56 -08:00
custom_backend [BE] move more unittest.main() to run_tests() (#50923) 2021-01-25 07:23:09 -08:00
custom_operator [BE] move more unittest.main() to run_tests() (#50923) 2021-01-25 07:23:09 -08:00
distributed Revert D26237328: Add compare_set operation and test to TCPStore 2021-02-04 13:43:05 -08:00
distributions Fix Dirichlet.arg_constraints event_dim (#51369) 2021-02-02 10:26:45 -08:00
error_messages
expect Fix Module backward hooks for all Tensor inputs/outputs (#46163) 2020-12-18 09:04:36 -08:00
fx Extend subgraph_rewriter logic (#51532) 2021-02-03 18:14:37 -08:00
jit [JIT] Allow implicit boolean conversion of containers (#51683) 2021-02-05 00:34:35 -08:00
jit_hooks [WIP] JIT Static Hooks: cpp tests (#49547) 2021-01-20 09:12:57 -08:00
mobile [PyTorch Mobile] Support torchbind custom classes in lite interpreter (#51432) 2021-02-03 21:57:19 -08:00
namespace_b
onnx [ONNX] Modifications in remove inplace ops passes to better handle binary inplace ops (#51318) (#51572) 2021-02-04 12:44:49 -08:00
optim
package_a [package] mangle imported module names (#50049) 2021-01-13 16:32:36 -08:00
quantization Numeric Suite: Swap with shadow modules only for quantized part of model (#51052) 2021-02-04 11:40:30 -08:00
scripts
test_img
type_hint_tests Fix mypy type hint for AdaptiveAvgPool2,3d, AdaptiveMaxPool2,3d (#49963) 2021-01-06 09:47:15 -08:00
HowToWriteTestsUsingFileCheck.md
module_a.py
print_test_stats.py only upload s3 stats on master, nightly, and release branch (#49645) 2020-12-22 14:15:18 -08:00
run_test.py Introduced AliasInfo for OpInfo (#50368) 2021-02-02 00:10:09 -08:00
simulate_nccl_errors.py
test_autograd.py Linear autodiff revert revert (#51613) 2021-02-04 16:32:05 -08:00
test_binary_ufuncs.py Add division overload with rounding_mode selection (#51706) 2021-02-04 13:08:36 -08:00
test_bundled_images.py
test_bundled_inputs.py [Pytorch] Expanded Bundled Inputs To Any Public Function (#51153) 2021-02-02 10:33:59 -08:00
test_complex.py Add serialization logic for complex numbers (#51287) 2021-01-28 17:25:35 -08:00
test_cpp_api_parity.py
test_cpp_extensions_aot.py Test TORCH_LIBRARY in CUDA extension (#47524) 2020-11-16 13:12:22 -08:00
test_cpp_extensions_jit.py Internal gradcheck wrapper in testing._internal that sets certain flags to True (#51133) 2021-01-29 09:13:37 -08:00
test_cuda_primary_ctx.py
test_cuda.py [ROCm] disable tests for ROCm 4.0.1 (#51510) 2021-02-02 14:39:08 -08:00
test_dataloader.py Fix persistent_workers + pin_memory (#48543) 2021-01-08 07:04:10 -08:00
test_datapipe.py move ZipDataset to Zip DataPipe (#51599) 2021-02-03 15:42:59 -08:00
test_dataset.py [DataLoader] Rename Functional DataSet to DataPipe (#51488) 2021-02-03 07:01:09 -08:00
test_determination.py Add option for cpp_extensions to compile standalone executable (#47862) 2020-12-01 20:03:08 -08:00
test_dispatch.py Improve docs around Math/DefaultBackend & add PythonDispatcher class. (#50854) 2021-01-25 23:10:36 -08:00
test_expecttest.py [BE] replace unittest.main with run_tests (#50451) 2021-01-13 10:33:08 -08:00
test_foreach.py Revert D25074763: [WIP] Update foreach APIs to use scalar lists 2021-02-03 17:06:40 -08:00
test_function_schema.py
test_functional_autograd_benchmark.py Skip test_functional_autograd_benchmark during code coverage (#50183) 2021-01-07 11:17:21 -08:00
test_futures.py [JIT/Futures] support set_exception api (#50983) 2021-02-04 20:22:19 -08:00
test_fx_experimental.py [FX] Move some heavily used passes out of experimental (#51392) 2021-02-01 19:02:26 -08:00
test_fx.py [FX] Added partial concrete values for symbolic tracing (#51609) 2021-02-04 12:06:02 -08:00
test_indexing.py index_put_ for complex tensors on CUDA (#51148) 2021-01-27 09:11:37 -08:00
test_jit_cuda_fuser.py [ROCm] disable tests for ROCm 4.0.1 (#51510) 2021-02-02 14:39:08 -08:00
test_jit_disabled.py [BE] replace unittest.main with run_tests (#50451) 2021-01-13 10:33:08 -08:00
test_jit_fuser_legacy.py
test_jit_fuser_te.py [TensorExpr] Use wider type for scalars (#50774) 2021-01-20 15:12:27 -08:00
test_jit_fuser.py Clean up some type annotations in caffe2/test (#49943) 2021-01-13 10:01:55 -08:00
test_jit_legacy.py
test_jit_profiling.py Remove redundant code for unsupported Python versions (#49486) 2021-01-06 12:45:46 -08:00
test_jit_py3.py Workaround Python3.9 limitations in test_jit_py3 (#51088) 2021-01-26 12:49:35 -08:00
test_jit_simple.py Remove redundant code for unsupported Python versions (#49486) 2021-01-06 12:45:46 -08:00
test_jit_string.py [JIT] Allow implicit boolean conversion of containers (#51683) 2021-02-05 00:34:35 -08:00
test_jit.py Linear autodiff revert revert (#51613) 2021-02-04 16:32:05 -08:00
test_kernel_launch_checks.py Fix check_kernel_launches.py for macros and provide extended context (#49365) 2020-12-14 22:09:33 -08:00
test_license.py BLD: create a LICENSE_BUNDLED.txt file from third_party licenses (#50745) 2021-01-26 14:55:07 -08:00
test_linalg.py [ROCm] add hipMAGMA support (#51238) 2021-02-01 22:09:33 -08:00
test_logging.py
test_metal.py
test_mkldnn.py add OneDNN linear backward (#49453) 2021-02-02 12:18:59 -08:00
test_mobile_optimizer.py cleaned up ModuleAttributeError (#50298) 2021-01-14 06:58:01 -08:00
test_multiprocessing_spawn.py
test_multiprocessing.py [BE] unified run_process_no_exception code (#49774) 2021-01-04 13:43:09 -08:00
test_namedtensor.py
test_namedtuple_return_api.py make forward AD API private (#51693) 2021-02-04 19:02:29 -08:00
test_native_functions.py
test_nn.py Add LazyBatchNormXd (#51548) 2021-02-05 10:27:04 -08:00
test_numba_integration.py Drop unused imports from test (#49973) 2021-01-07 12:09:38 -08:00
test_numpy_interop.py Back out "Revert D25757721: [pytorch][PR] Run mypy on more test files" (#50142) 2021-01-06 11:27:36 -08:00
test_op_aliases.py Implementing NumPy-like function torch.broadcast_to (#48997) 2020-12-21 11:24:50 -08:00
test_openmp.py
test_ops.py Introduced AliasInfo for OpInfo (#50368) 2021-02-02 00:10:09 -08:00
test_optim.py Revert D25976245: [pytorch][PR] Enable Skipped ROCM Tests in common_nn.py 2021-01-21 13:28:37 -08:00
test_overrides.py Fix pickling for Tensor subclasses (redo) (#47732) 2021-02-01 07:32:52 -08:00
test_package.py [package] patch inspect.getfile to work with PackageImporter (#51568) 2021-02-02 11:29:29 -08:00
test_profiler.py Add 'repeat' argument to profiler.schedule (#51630) 2021-02-04 13:51:04 -08:00
test_pruning_op.py Rowwise Prune op (Add the test to OSS run_test), Make the op private. (#46131) 2021-01-29 06:08:18 -08:00
test_pytree.py
test_quantization.py update HistogramObserver to be scriptable (#51081) 2021-01-27 07:27:03 -08:00
test_reductions.py [numpy] torch.{all/any} : output dtype is always bool (#47878) 2021-01-08 11:05:39 -08:00
test_serialization.py Enable tests using named temp files on Windows (#49640) 2020-12-29 09:57:35 -08:00
test_shape_ops.py Removes nonzero method warning (#51618) 2021-02-04 17:43:43 -08:00
test_show_pickle.py [BE] replace unittest.main with run_tests (#50451) 2021-01-13 10:33:08 -08:00
test_sort_and_select.py Revert "Stable sort for CPU (#50052)" (#50752) 2021-01-19 18:21:25 -08:00
test_sparse.py Internal gradcheck wrapper in testing._internal that sets certain flags to True (#51133) 2021-01-29 09:13:37 -08:00
test_spectral_ops.py STFT: Improve error message when window is on wrong device (#51128) 2021-01-27 22:31:57 -08:00
test_static_runtime.py Clean up some type annotations in caffe2/test (#49943) 2021-01-13 10:01:55 -08:00
test_tensor_creation_ops.py [testing] Port torch.{repeat, tile} tests to use OpInfo machinery (#50199) 2021-01-19 06:02:27 -08:00
test_tensorboard.py
test_tensorexpr_pybind.py Make te a hidden package for now (#51690) 2021-02-04 07:58:38 -08:00
test_tensorexpr.py [BE] replace unittest.main with run_tests (#50451) 2021-01-13 10:33:08 -08:00
test_testing.py [reland] Refactor mypy configs list into editor-friendly wrapper (#50826) 2021-01-26 09:04:14 -08:00
test_throughput_benchmark.py Enable tests using named temp files on Windows (#49640) 2020-12-29 09:57:35 -08:00
test_torch.py Fix torch.cdist backward CUDA error due to illegal gridDim setting (#51569) 2021-02-02 20:41:24 -08:00
test_type_hints.py [reland] Refactor mypy configs list into editor-friendly wrapper (#50826) 2021-01-26 09:04:14 -08:00
test_type_info.py
test_type_promotion.py [numpy] torch.erf{c} : promote integer inputs to float (#48472) 2020-11-27 15:08:40 -08:00
test_unary_ufuncs.py fix hardsigmoid_backward for boundary case (#51454) 2021-02-04 14:37:58 -08:00
test_utils.py Re-enable test_standalone_load for Windows 11.1 (#51596) 2021-02-03 08:58:34 -08:00
test_view_ops.py Add tensor.view(dtype) (#47951) 2021-01-08 06:55:21 -08:00
test_vmap.py Add private mechanism to toggle vmap fallback warnings (#51218) 2021-01-28 13:05:00 -08:00
test_vulkan.py
test_xnnpack_integration.py