pytorch/test
kshitij12345 b372be4211 [nn] lstm : no batch dim support (#71056)
Summary:
Reference: https://github.com/pytorch/pytorch/issues/60585

TODO:
* [x] Update docs

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

Reviewed By: samdow

Differential Revision: D33638643

Pulled By: jbschlosser

fbshipit-source-id: c0949829de8a8e6e7b2873f459a8d7da597a3be3
(cherry picked from commit f94d5849f6)
2022-01-24 15:13:40 +00:00
..
ao/sparsity [sparsity] Convert function for sparse kernels without a context manager (#66778) 2021-12-09 02:58:57 -08:00
autograd [skip ci] Set test owners for autograd tests (#66834) 2021-10-19 08:35:02 -07:00
benchmark_utils Add lint to ensure all test files have headers with ownership info (#66826) 2021-11-03 18:21:49 -07:00
bottleneck_test Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
cpp Codegen: python_torch_functions only include relevant operators (#68693) 2022-01-21 15:37:06 +00: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 Resubmit #67161 (#67735) 2021-11-04 09:59:30 -07:00
custom_backend [PyTorch] Adopt IValue::toTupleRef() where obvious (#65505) 2021-11-02 10:22:18 -07:00
custom_operator Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
distributed [Opt Overlap] Remove redundant tests (#71600) 2022-01-23 00:04:32 +00:00
distributions [torch.distributions] Implement positive-semidefinite constraint (#71375) 2022-01-20 17:33:08 +00:00
error_messages Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
expect [FX] Add a default_value arg to Graph.placeholder and fix split_module (#71016) 2022-01-12 14:03:17 -08:00
forward_backward_compatibility Remove deprecated cuDNN convolution ops (#71128) 2022-01-12 06:34:42 -08:00
fx [const_fold] Fix call_module const folding (#68614) 2021-11-18 20:56:06 -08:00
fx_acc stochastic_depth support (#71536) 2022-01-21 17:44:19 +00:00
fx2trt Fix test ownership lint (#71554) 2022-01-21 18:24:42 +00:00
jit Fix test ownership lint (#71554) 2022-01-21 18:24:42 +00:00
jit_hooks Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
mobile Move upgraders from python to cpp (#70593) 2022-01-22 00:24:24 +00:00
onnx Export ONNX models with readable input/output names (#68976) 2022-01-21 00:34:56 +00:00
package detect mocked module on saving pass (#70641) 2022-01-10 11:11:55 -08:00
quantization dbr quant: do not crash on unsupported qconfig_dict keys if they are empty (#71233) 2022-01-24 14:24:48 +00:00
scripts Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
test_img
typing
delete.py Introducing DataChunk for DataPipes batching (#62768) 2021-08-06 08:38:33 -07:00
HowToWriteTestsUsingFileCheck.md Allow for heterogenous List and Dict values + Improve container typing algorithm (#57137) 2021-07-10 14:29:05 -07:00
linear.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
run_test.py Fix fx2trt CI test trigger condition (#71014) 2022-01-10 13:37:24 -08:00
simulate_nccl_errors.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
test_ao_sparsity.py [sparsity] Fix and enable the pruning tests (#66411) 2021-11-22 15:28:12 -08:00
test_autocast.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_autograd.py Enable nested default hooks (#70932) 2022-01-11 15:03:49 -08:00
test_binary_ufuncs.py Low precision support for jiterator (#70157) 2021-12-19 11:56:57 -08:00
test_bundled_images.py Add ownership to more edge tests (#67859) 2021-11-05 11:01:16 -07:00
test_bundled_inputs.py Add lint to ensure all test files have headers with ownership info (#66826) 2021-11-03 18:21:49 -07:00
test_complex.py [skip ci] Set test owner for test_complex.py (#66835) 2021-10-19 08:36:27 -07:00
test_cpp_api_parity.py [skip ci] Set test owners for cpp test (#66836) 2021-10-21 13:17:46 -07:00
test_cpp_extensions_aot.py No-batch-dim support for ConvNd (#70506) 2022-01-06 16:53:50 -08:00
test_cpp_extensions_jit.py Remove remaining THC code (#69039) 2021-12-08 12:18:08 -08:00
test_cuda_primary_ctx.py [skip ci] Set test owner for test_cuda* tests (#66838) 2021-10-21 17:36:25 -07:00
test_cuda.py Kill the test_torch.py mixin and creates test_scatter_gather_ops (#71691) 2022-01-24 09:32:32 +00:00
test_dataloader.py Revert D33694867: Fix persistent worker exits before pin_memory thread 2022-01-22 00:09:28 +00:00
test_datapipe.py [DataPipe] Make GroupBy serializable with lambda function (#71497) 2022-01-21 16:04:45 +00:00
test_deploy.py [skip ci] Set test owner for deploy/package tests (#66830) 2021-10-26 07:49:33 -07:00
test_determination.py Have test classes extend from common_utils.TestCase, not unittest.TestCase (#66900) 2021-10-19 16:54:05 -07:00
test_dispatch.py [skip ci] Set test owner for test_dispatch.py (#66840) 2021-10-21 08:48:37 -07:00
test_foreach.py Fix device type / dtype handling for parametrized test names (#65217) 2021-11-29 19:02:23 -08:00
test_function_schema.py Add forward compatability tests in CI (#64139) 2022-01-05 23:40:06 -08:00
test_functional_autograd_benchmark.py fix python version error (#71021) 2022-01-21 01:05:53 +00:00
test_functional_optim.py Add test owner to distributed files starting with test_ (#66797) 2021-10-19 10:55:20 -07:00
test_functionalization.py Fix test ownership lint (#71554) 2022-01-21 18:24:42 +00:00
test_futures.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_fx_experimental.py fix conv+bn folding issue when bn hasn't running states (#71259) 2022-01-18 22:12:41 +00:00
test_fx.py OpInfos for: normal, bernoulli, multinomial (#66358) 2021-12-14 06:59:38 -08:00
test_import_stats.py Set test owners for module: ci (#66796) 2021-10-18 10:29:50 -07:00
test_indexing.py Fixes CUDA vs CPU consistency for index_put_ when accumulating (part 2) (#67189) 2021-11-08 17:56:43 -08:00
test_jit_autocast.py [JIT] additional support for CallMethod with autocasting (#67925) 2021-11-08 14:37:09 -08:00
test_jit_cuda_fuser.py Allow single node fusion for nvfuser (#70000) 2021-12-23 17:07:57 -08:00
test_jit_disabled.py Set test owners for jit tests (#66808) 2021-10-25 07:51:10 -07:00
test_jit_fuser_legacy.py Set test owners for jit tests (#66808) 2021-10-25 07:51:10 -07:00
test_jit_fuser_te.py [nnc] Update block and thread extents in cuda_codegen to use int64_t (#71428) 2022-01-19 23:21:24 +00:00
test_jit_fuser.py skip test_jit_legacy (#68129) 2021-11-10 23:08:32 -08:00
test_jit_legacy.py Set test owners for jit tests (#66808) 2021-10-25 07:51:10 -07:00
test_jit_profiling.py Set test owners for jit tests (#66808) 2021-10-25 07:51:10 -07:00
test_jit_simple.py Set test owners for jit tests (#66808) 2021-10-25 07:51:10 -07:00
test_jit_string.py [PyTorch] Move prim string ops to JIT op registry (#70501) 2022-01-06 18:26:28 -08:00
test_jit.py Update min python version to 3.7 in setup.py and mypy configs (#71494) 2022-01-20 00:03:57 +00:00
test_kernel_launch_checks.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_license.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_linalg.py Skip broken svd tests (#71646) 2022-01-21 20:47:21 +00:00
test_logging.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_masked.py Fix test ownership lint (#71554) 2022-01-21 18:24:42 +00:00
test_metal.py [skip ci] Set test owner for mobile tests (#66829) 2021-10-26 10:20:01 -07:00
test_mkldnn.py Avoid the view for mkldnn case in 1D convolution (#68166) 2021-11-15 11:56:45 -08:00
test_mobile_optimizer.py [skip ci] Set test owner for mobile tests (#66829) 2021-10-26 10:20:01 -07:00
test_model_dump.py Add ownership to more edge tests (#67859) 2021-11-05 11:01:16 -07:00
test_module_init.py [quant] fix dropout in FX graph mode quantization (#71043) 2022-01-13 15:59:59 -08:00
test_modules.py [nn] lstm : no batch dim support (#71056) 2022-01-24 15:13:40 +00:00
test_monitor.py monitor: add docstrings to pybind interface (#71481) 2022-01-21 23:04:33 +00:00
test_multiprocessing_spawn.py make ProcessException pickleable (#70118) 2021-12-30 09:09:55 -08:00
test_multiprocessing.py [skip ci] Set test owner for multiprocessing tests (#66848) 2021-10-21 13:13:53 -07:00
test_namedtensor.py [skip ci] Set test owner for named tensor tests (#66849) 2021-10-21 08:22:26 -07:00
test_namedtuple_return_api.py Add linalg.lu_factor (#66933) 2022-01-05 20:32:12 -08:00
test_native_functions.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_nn.py nice error msg from load_state_dict for non-tensor value (#70596) 2022-01-21 22:02:13 +00:00
test_nnapi.py NNAPI: quant logistic fix (#70847) 2022-01-07 13:36:33 -08:00
test_numba_integration.py Remove unwanted comma (#71193) 2022-01-13 19:09:05 -08:00
test_numpy_interop.py [skip ci] Set test owner for test_numpy_interop.py (#66851) 2021-10-19 10:50:54 -07:00
test_openmp.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_ops.py Do not use ZeroTensor for inplace ops (#69998) 2021-12-23 15:52:34 -08:00
test_optim.py Remove unused optimizers variable in test (#70668) 2022-01-14 11:59:49 -08:00
test_overrides.py Add linalg.solve_triangular (#63568) 2021-11-22 12:41:06 -08:00
test_package.py [skip ci] Set test owner for deploy/package tests (#66830) 2021-10-26 07:49:33 -07:00
test_profiler.py [Profiler] Fix memory profile type from recent refactor (#71417) 2022-01-18 22:18:11 +00:00
test_pruning_op.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_public_bindings.py Have test classes extend from common_utils.TestCase, not unittest.TestCase (#66900) 2021-10-19 16:54:05 -07:00
test_python_dispatch.py Enable Python bindings for UntypedStorage (#68945) 2022-01-20 02:11:34 +00:00
test_pytree.py [skip ci] Add test owners for a special hi-pri class of tests (#67553) 2021-10-29 12:17:21 -07:00
test_quantization.py dbr quant: break up test class into multiple classes (#70246) 2022-01-05 06:36:41 -08:00
test_reductions.py Deprecates _aminmax (#71576) 2022-01-21 09:23:29 +00:00
test_scatter_gather_ops.py Kill the test_torch.py mixin and creates test_scatter_gather_ops (#71691) 2022-01-24 09:32:32 +00:00
test_segment_reductions.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_serialization.py Avoid dtype mismatch error in torch.save if storages are unallocated (#68787) 2021-11-24 09:51:29 -08:00
test_set_default_mobile_cpu_allocator.py Add ownership to more edge tests (#67859) 2021-11-05 11:01:16 -07:00
test_shape_ops.py replace onlyOnCPUAndCUDA with onlyNativeDeviceTypes (#65201) 2021-11-01 09:22:34 -07:00
test_show_pickle.py Add ownership to more edge tests (#67859) 2021-11-05 11:01:16 -07:00
test_sort_and_select.py Fix torch.{unique, unique_consecutive} out of bound (#71540) 2022-01-21 02:36:49 +00:00
test_sparse_csr.py Sparse CSR: Handle zero matrix consistently for triangular_solve (#71304) 2022-01-17 13:47:49 -08:00
test_sparse.py Support sparse.sum on empty sparse tensor (#71091) 2022-01-19 18:58:08 +00:00
test_spectral_ops.py replace onlyOnCPUAndCUDA with onlyNativeDeviceTypes (#65201) 2021-11-01 09:22:34 -07:00
test_stateless.py [skip ci] Add test owners for a special hi-pri class of tests (#67553) 2021-10-29 12:17:21 -07:00
test_static_runtime.py [static runtime][dper] multi_env tests for static runtime: selective enable (#67467) 2021-11-18 01:04:12 -08:00
test_tensor_creation_ops.py Kill the test_torch.py mixin and creates test_scatter_gather_ops (#71691) 2022-01-24 09:32:32 +00:00
test_tensorboard.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_tensorexpr_pybind.py [tensorexpr] check for index out of bounds in ir_eval (#68858) 2021-12-16 09:27:45 -08:00
test_tensorexpr.py [skip ci] Set test owners for NNC tests (#66833) 2021-10-26 07:46:18 -07:00
test_testing.py fix TensorLikePair origination (#70304) 2022-01-12 06:44:00 -08:00
test_throughput_benchmark.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_torch.py Kill the test_torch.py mixin and creates test_scatter_gather_ops (#71691) 2022-01-24 09:32:32 +00:00
test_type_hints.py [skip ci] Set test owners for test_typing.py (#66869) 2021-10-20 09:41:13 -07:00
test_type_info.py [skip ci] Set test owners for test_typing.py (#66869) 2021-10-20 09:41:13 -07:00
test_type_promotion.py replace onlyOnCPUAndCUDA with onlyNativeDeviceTypes (#65201) 2021-11-01 09:22:34 -07:00
test_typing.py [skip ci] Set test owners for test_typing.py (#66869) 2021-10-20 09:41:13 -07:00
test_unary_ufuncs.py The Jiterator (#69439) 2021-12-06 07:32:48 -08:00
test_utils.py Add support for C++ frontend wrapper on Linux (#69094) 2021-12-02 16:47:00 -08:00
test_view_ops.py Add forward AD support for as_strided (#68629) 2021-12-14 04:33:05 -08:00
test_vmap.py update example in classification losses (#69816) 2021-12-21 02:46:48 -08:00
test_vulkan.py [skip ci] Set test owner for mobile tests (#66829) 2021-10-26 10:20:01 -07:00
test_xnnpack_integration.py Add ownership to more edge tests (#67859) 2021-11-05 11:01:16 -07:00