mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
b3e4f5029b
This PR is a copy of https://github.com/pytorch/pytorch/pull/90849 that merge was reverted. The PR adds "check sparse tensor invariants" flag to Context that when enabled will trigger sparse tensor data invariants checks in unsafe methods of constructing sparse COO/CSR/CSC/BSR/BSC tensors. The feature includes the following changes to UI: `torch.sparse.check_sparse_tensor_invariants` class provides different ways to enable/disable the invariant checking. `torch.sparse_coo/csr/csc/bsr/bsc/compressed_tensor` functions have a new optional argument `check_invariants` to enable/disable the invariant checks explicitly. When the `check_invariants` argument is specified, the global state of the feature is temporarily overridden. The PR fixes https://github.com/pytorch/pytorch/issues/90833 Pull Request resolved: https://github.com/pytorch/pytorch/pull/92094 Approved by: https://github.com/cpuhrsch |
||
|---|---|---|
| .. | ||
| auto_gil.h | ||
| byte_order.cpp | ||
| byte_order.h | ||
| cpp_stacktraces.cpp | ||
| cpp_stacktraces.h | ||
| cuda_enabled.h | ||
| cuda_lazy_init.cpp | ||
| cuda_lazy_init.h | ||
| disable_torch_function.cpp | ||
| disable_torch_function.h | ||
| init.cpp | ||
| init.h | ||
| invalid_arguments.cpp | ||
| invalid_arguments.h | ||
| memory.h | ||
| nested.cpp | ||
| nested.h | ||
| numpy_stub.h | ||
| object_ptr.cpp | ||
| object_ptr.h | ||
| out_types.cpp | ||
| out_types.h | ||
| pybind.cpp | ||
| pybind.h | ||
| pycfunction_helpers.h | ||
| python_arg_parser.cpp | ||
| python_arg_parser.h | ||
| python_compat.h | ||
| python_dispatch.cpp | ||
| python_dispatch.h | ||
| python_numbers.h | ||
| python_scalars.h | ||
| python_strings.h | ||
| python_stub.h | ||
| python_symnode.cpp | ||
| python_symnode.h | ||
| python_torch_function_mode.h | ||
| python_tuples.h | ||
| pythoncapi_compat.h | ||
| schema_info.cpp | ||
| schema_info.h | ||
| six.h | ||
| structseq.cpp | ||
| structseq.h | ||
| tensor_apply.cpp | ||
| tensor_apply.h | ||
| tensor_dtypes.cpp | ||
| tensor_dtypes.h | ||
| tensor_flatten.cpp | ||
| tensor_flatten.h | ||
| tensor_layouts.cpp | ||
| tensor_layouts.h | ||
| tensor_list.cpp | ||
| tensor_list.h | ||
| tensor_memoryformats.cpp | ||
| tensor_memoryformats.h | ||
| tensor_new.cpp | ||
| tensor_new.h | ||
| tensor_numpy.cpp | ||
| tensor_numpy.h | ||
| tensor_qschemes.cpp | ||
| tensor_qschemes.h | ||
| tensor_types.cpp | ||
| tensor_types.h | ||
| throughput_benchmark-inl.h | ||
| throughput_benchmark.cpp | ||
| throughput_benchmark.h | ||
| torch_dispatch_mode.h | ||
| variadic.cpp | ||
| variadic.h | ||
| verbose.cpp | ||