pytorch/test
Alban Desmaison b858993c97 Fix engine check for case where grad is a subclass (#65568)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65568

Test Plan: Imported from OSS

Reviewed By: soulitzer

Differential Revision: D31158089

Pulled By: albanD

fbshipit-source-id: 2a77df9b6340107de02a043b57a36cb7ae68df34
2021-09-24 08:41:19 -07:00
..
ao/sparsity [pruner] add getter for pruned outputs in base pruner (#63520) 2021-08-25 09:57:29 -07:00
autograd extract TestAutogradComplex into its own test file (#63400) 2021-09-02 04:34:35 -07:00
backward_compatibility Make {select,slice,diagonal}_backward primitives wrt autograd (#64933) 2021-09-14 08:10:59 -07:00
benchmark_utils
bottleneck_test
cpp [nn] Fold : no batch dim (#64909) 2021-09-23 08:37:32 -07:00
cpp_api_parity ENH Adds test and docs for modules that already support no batch dims (#62729) 2021-09-02 12:36:54 -07:00
cpp_extensions Add support for the ONNX Runtime Eager Mode backend (#58248) 2021-08-20 11:17:13 -07:00
custom_backend
custom_operator [jit] Make operation call accept Stack& instead Stack* (#63414) 2021-08-30 11:49:20 -07:00
distributed Explicitly destory ProcessGroup in allgather_coalesced_async test (#65513) 2021-09-23 18:35:08 -07:00
distributions Pin SciPy to 1.6.2 on Windows (#65017) 2021-09-16 18:25:43 -07:00
error_messages
expect Support Union in TorchScript (#64234) 2021-09-03 06:12:24 -07:00
fx [fx const fold] fix a case when some inputs are unused (#65223) 2021-09-17 17:29:55 -07:00
jit [ONNX] Deprecated the example_outputs param from torch.onnx.export() function. (#62815) (#64380) 2021-09-23 22:20:46 -07:00
jit_hooks
mobile Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00
onnx [ONNX] Deprecate use_external_data_format param from torch.onnx.export() function. (#62257) (#64382) 2021-09-23 22:20:48 -07:00
optim
package [Codemod][FBSourceBlackLinter] Daily arc lint --take BLACK 2021-09-17 20:33:12 -07:00
quantization [ONNX] Deprecated the example_outputs param from torch.onnx.export() function. (#62815) (#64380) 2021-09-23 22:20:46 -07:00
scripts
test_img
typing
delete.py
HowToWriteTestsUsingFileCheck.md
linear.py
run_test.py Basic implementation of ShardedLinear using ShardedTensor. (#64128) 2021-09-20 18:31:11 -07:00
simulate_nccl_errors.py
test_ao_sparsity.py
test_autocast.py
test_autograd.py Remove run_functional_checks from test_autograd and create necessary OpInfos (#64993) 2021-09-15 12:45:38 -07:00
test_binary_ufuncs.py restore test_inplace_comparison_ops_require_inputs_have_same_dtype Expected behavior (#64267) 2021-09-08 06:42:12 -07:00
test_buffer_protocol.py [Doc] make_tensor to torch.testing module (#63925) 2021-08-30 12:25:40 -07:00
test_bundled_images.py replace self.assertTrue(torch.allclose(..)) with self.assertEqual(…) (#63637) 2021-08-25 16:47:40 -07:00
test_bundled_inputs.py [PyTorch][Edge] Improve InflatableArgs for Bundled Inputs (#62368) 2021-08-20 09:36:08 -07:00
test_complex.py deprecate dtype getters from torch.testing namespace (#63554) 2021-09-07 08:58:51 -07:00
test_cpp_api_parity.py
test_cpp_extensions_aot.py Add support for the ONNX Runtime Eager Mode backend (#58248) 2021-08-20 11:17:13 -07:00
test_cpp_extensions_jit.py Add force_on_cpu test to win cuda10.2 on GHA (#65094) 2021-09-21 11:14:15 -07:00
test_cuda.py simplify op name determination into a single forward pass (#64261) 2021-09-02 07:32:11 -07:00
test_cuda_primary_ctx.py
test_dataloader.py When test set_affinity, don't hardcode the CPU ID (#65042) 2021-09-15 08:10:59 -07:00
test_datapipe.py [DataPipe] Fix deepcopy filehandle for Mapper and in-place modification for IterableWrapper (#65220) 2021-09-21 14:29:40 -07:00
test_determination.py [Dist/CI] Remove dist from target determinator (#64721) 2021-09-09 12:07:43 -07:00
test_dispatch.py add python API to print all operators that have kernels registered to a particular DispatchKey (#63575) 2021-09-22 09:15:55 -07:00
test_foreach.py deprecate dtype getters from torch.testing namespace (#63554) 2021-09-07 08:58:51 -07:00
test_function_schema.py Detect out argument in the schema (#62755) 2021-08-27 11:20:33 -07:00
test_functional_autograd_benchmark.py
test_functional_optim.py [BE] Enable functional optim tests for windows (#63462) 2021-08-23 17:49:01 -07:00
test_futures.py
test_fx.py [FX] Fix tracing of bitwise and/or (#65196) 2021-09-17 14:33:02 -07:00
test_fx_experimental.py OpInfo: nn.functional.conv2d (#65233) 2021-09-21 09:26:23 -07:00
test_gen_backend_stubs.py Add support for the ONNX Runtime Eager Mode backend (#58248) 2021-08-20 11:17:13 -07:00
test_import_stats.py Track peak memory usage (#65157) 2021-09-20 17:25:16 -07:00
test_indexing.py [Doc] make_tensor to torch.testing module (#63925) 2021-08-30 12:25:40 -07:00
test_jit.py [JIT] Improve BatchMM mutability handling (#65097) 2021-09-16 10:46:14 -07:00
test_jit_cuda_fuser.py Revert "Revert D30752939: [pytorch][PR] nvfuser update" (#65137) 2021-09-22 04:54:51 -07:00
test_jit_disabled.py
test_jit_fuser.py
test_jit_fuser_legacy.py
test_jit_fuser_te.py [JIT] canonicalize aten::rsub (#65014) 2021-09-22 17:20:46 -07:00
test_jit_legacy.py
test_jit_profiling.py
test_jit_simple.py
test_jit_string.py
test_kernel_launch_checks.py
test_license.py
test_linalg.py Bf16 matmul (#64619) 2021-09-17 10:31:56 -07:00
test_logging.py
test_metal.py
test_mkldnn.py
test_mobile_optimizer.py Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00
test_model_dump.py
test_module_init.py
test_modules.py Revert D30867266: [pytorch][PR] TST Adds gradcheck and gradgradcheck to module info 2021-09-12 10:30:28 -07:00
test_multiprocessing.py
test_multiprocessing_spawn.py
test_namedtensor.py disallow empty named dims list to flatten(names, name) (#61953) 2021-08-31 19:32:30 -07:00
test_namedtuple_return_api.py
test_native_functions.py
test_nn.py [nn] Fold : no batch dim (#64909) 2021-09-23 08:37:32 -07:00
test_nnapi.py Fix typo in NNAPI tests (#63797) 2021-08-23 16:34:24 -07:00
test_numba_integration.py
test_numpy_interop.py deprecate dtype getters from torch.testing namespace (#63554) 2021-09-07 08:58:51 -07:00
test_openmp.py
test_ops.py [JIT] canonicalize aten::rsub (#65014) 2021-09-22 17:20:46 -07:00
test_optim.py To add SequentialLR to PyTorch Core Schedulers (#64037) 2021-09-09 09:36:32 -07:00
test_overrides.py Actually deprecate __torch_function__ as plain methods (#64843) 2021-09-17 08:32:53 -07:00
test_package.py
test_profiler.py Revert D30721329: [pytorch][PR] Enable CUPTI for kineto by default on windows. 2021-09-23 22:14:32 -07:00
test_pruning_op.py Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00
test_public_bindings.py Allow disabling cache in autocast (automatic mixed precision) (#63552) 2021-09-08 07:47:18 -07:00
test_python_dispatch.py Fix engine check for case where grad is a subclass (#65568) 2021-09-24 08:41:19 -07:00
test_pytree.py
test_quantization.py [quant] AO migration of the torch/quantization/quantize_fx.py and torch/quantization/fx/* (#65033) 2021-09-22 09:29:15 -07:00
test_reductions.py [Reland] Added reference tests to ReductionOpInfo (#64273) 2021-09-12 20:05:43 -07:00
test_segment_reductions.py
test_serialization.py
test_set_default_mobile_cpu_allocator.py
test_shape_ops.py deprecate dtype getters from torch.testing namespace (#63554) 2021-09-07 08:58:51 -07:00
test_show_pickle.py
test_sort_and_select.py deprecate dtype getters from torch.testing namespace (#63554) 2021-09-07 08:58:51 -07:00
test_sparse.py [Fixed] Enable Half, BFloat16, and Complex dtypes for coo-coo sparse matmul [CUDA] (#59980) 2021-09-21 13:03:40 -07:00
test_sparse_csr.py deprecate dtype getters from torch.testing namespace (#63554) 2021-09-07 08:58:51 -07:00
test_spectral_ops.py Fix torch.istft length mismatch and window runtime error (#63469) 2021-09-02 09:31:47 -07:00
test_stateless.py Add functional api for nn.Module (#61447) 2021-09-21 12:39:43 -07:00
test_static_runtime.py Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00
test_tensor_creation_ops.py implement "xy" indexing for torch.meshgrid (#62724) 2021-09-17 08:31:17 -07:00
test_tensorboard.py
test_tensorexpr.py [nnc] Updated indices during broadcast to use int64_t (#64627) 2021-09-09 08:29:37 -07:00
test_tensorexpr_pybind.py [TensorExpr] PyBinds: improve QoL of pybind users. (#64886) 2021-09-14 00:21:28 -07:00
test_testing.py Generic test parametrization functionality (#60753) 2021-09-14 19:52:59 -07:00
test_throughput_benchmark.py Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00
test_torch.py [fix] don't expose unique_dim in torch (#63080) 2021-09-14 18:19:17 -07:00
test_type_hints.py
test_type_info.py
test_type_promotion.py deprecate dtype getters from torch.testing namespace (#63554) 2021-09-07 08:58:51 -07:00
test_typing.py
test_unary_ufuncs.py deprecate dtype getters from torch.testing namespace (#63554) 2021-09-07 08:58:51 -07:00
test_utils.py Torchhub: More robust assumption regarding main or master branch (#64364) 2021-09-20 10:36:13 -07:00
test_view_ops.py Add crow_/col_indices to view types (#63176) 2021-09-20 14:35:58 -07:00
test_vmap.py
test_vulkan.py
test_xnnpack_integration.py Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00