OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ac26f8237c
pytorch/test
History
Pearu Peterson 214f4bf2ff Support sparse.sum on empty sparse tensor (#71091) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/71091

Fixes https://github.com/pytorch/pytorch/issues/65394

The masked sum on a full input tensor (of any layout) with an all-true mask is the same as the sum on the strided input tensor (after applying `to_dense` to sparse inputs).
Since masked sum uses `torch.sparse.sum` then, for the simplicity of masked reductions implementations, its reduction behavior ought to be defined by the behavior of the `torch.sum`. This PR implements the behavioral connection with respect to the directional summation of empty sparse tensors that correspond to all-zero strided tensors.

cc nikitaved pearu cpuhrsch

Test Plan: Imported from OSS

Reviewed By: davidberard98

Differential Revision: D33651750

Pulled By: cpuhrsch

fbshipit-source-id: 703891bff88c8da6270b4272f5d2da81688db67d
(cherry picked from commit 53f97e80f7)
2022-01-19 18:58:08 +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 empty_cpu: Add functions that don't depend on Tensor (#70613) 2022-01-19 00:01:58 +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 [DDP Checkpointing] non-reentrant checkpoint tests (#69060) 2022-01-19 18:09:41 +00:00
distributions [Reinstate] Wishart distribution (#70377) 2021-12-30 11:41:46 -08: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 [fx2trt] break down div (#71172) 2022-01-12 09:46:46 -08:00
fx2trt [quant] Remove some redundant entries in backend_config_dict for TensorRT (#70971) 2022-01-14 12:43:25 -08:00
jit backout D33469839 (#71443) 2022-01-18 23:51:51 +00:00
jit_hooks Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
mobile Reland D33284352: [jit][edge] Do not reuse mobile type parser for all unpicklers. (#71048) 2022-01-10 12:42:23 -08:00
onnx backout D33469839 (#71443) 2022-01-18 23:51:51 +00:00
package detect mocked module on saving pass (#70641) 2022-01-10 11:11:55 -08:00
quantization [quant][fix][fx][graphmode] Fix qconfig setting for fused modules (#71254) 2022-01-14 23:31:11 -08:00
scripts Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
test_img
typing Deprecate QR (#57745) 2021-05-10 22:56:37 -07:00
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 Document torch.cuda.ExternalStream, torch.cuda.caching_allocator_alloc and torch.cuda.caching_allocator_delete (#70126) 2022-01-12 15:44:40 -08:00
test_dataloader.py [DataPipe] Control shuffle settings from DataLoader2 (#65756) 2021-12-14 07:35:26 -08:00
test_datapipe.py Make Demux serializable with lambda function (#71311) 2022-01-18 06:47:54 -08: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 [skip ci] Add test owners for a special hi-pri class of tests (#67553) 2021-10-29 12:17:21 -07: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 functionalization: update is_aliased() logic (#68881) 2021-12-02 09:19:12 -08: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 Add support for permutting dynamic fusion group outputs to channels last format (#70656) 2022-01-12 09:11:34 -08: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 Testing for Default Inference for Device Type (#69052) 2022-01-13 13:59:12 -08: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 Remove random_fullrank_matrix_distinc_singular_value (#68183) 2022-01-05 20:33:37 -08:00
test_logging.py Set test owners for tests with unknown owners (#67552) 2021-10-29 12:42:01 -07:00
test_masked.py Strided masked var. (#68738) 2021-12-01 19:19:37 -08: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 No-batch-dim support for ConvNd (#70506) 2022-01-06 16:53:50 -08:00
test_monitor.py torch/monitor: add pybind (#69567) 2022-01-12 13:35:11 -08: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 add BFloat16 support for AvgPool2d on CPU (#66927) 2022-01-14 07:59:10 -08: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 add python dispatch test back to CI and fix typo in test (#69565) 2021-12-08 08:44:49 -08: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 Updated searchsorted functionality (#66818) 2021-11-05 12:13:47 -07: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 Implement topk with sort for some cases (#68632) 2021-11-19 17:18:20 -08: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 Add meta support to tensor range factories (#67032) 2021-11-05 15:36:29 -07: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 Copy: Fix CPU transpose path ignoring neg and conj bits (#69026) 2022-01-14 10:13:33 -08: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