Ivan Yashchuk 75a2a92b02 Add torch.linalg.cholesky_ex without checking for errors by default (#56724) ... Summary: The new function has the following signature `cholesky_ex(Tensor input, *, bool check_errors=False) -> (Tensor L, Tensor infos)`. When `check_errors=True`, an error is thrown if the decomposition fails; `check_errors=False` - responsibility for checking the decomposition is on the user. When `check_errors=False`, we don't have host-device memory transfers for checking the values of the `info` tensor. Rewrote the internal code for `torch.linalg.cholesky`. Added `cholesky_stub` dispatch. `linalg_cholesky` is implemented using calls to `linalg_cholesky_ex` now. Resolves https://github.com/pytorch/pytorch/issues/57032. Ref. https://github.com/pytorch/pytorch/issues/34272, https://github.com/pytorch/pytorch/issues/47608, https://github.com/pytorch/pytorch/issues/47953 Pull Request resolved: https://github.com/pytorch/pytorch/pull/56724 Reviewed By: ngimel Differential Revision: D27960176 Pulled By: mruberry fbshipit-source-id: f05f3d5d9b4aa444e41c4eec48ad9a9b6fd5dfa5		2021-05-01 18:48:27 -07:00
..
_static	Add documentation page for pipeline parallelism. (#50791)	2021-01-25 13:47:13 -08:00
_templates	various doc building cleanups (#53851)	2021-03-16 15:01:59 -07:00
community	Lint trailing newlines (#54737)	2021-03-30 13:09:52 -07:00
notes	Add first draft of gradcheck note (#55966)	2021-04-27 14:33:42 -07:00
rpc	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
scripts	Optimize SiLU (Swish) op in PyTorch (#42976)	2020-08-16 13:21:57 -07:00
__config__.rst	Fix __config__ docs (#48557)	2020-11-29 23:57:06 -08:00
amp.rst	[Relanding] Implemented torch.linalg.multi_dot (#52859)	2021-04-01 04:49:05 -07:00
autograd.rst	breakup autograd documentation (#55672)	2021-04-14 12:40:00 -07:00
backends.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
benchmark_utils.rst	Expand benchmark utils docs (#51664)	2021-02-04 00:22:41 -08:00
bottleneck.rst
checkpoint.rst
complex_numbers.rst	various doc building cleanups (#53851)	2021-03-16 15:01:59 -07:00
conf.py	Hide module paths leaking in the documentation. (#54585)	2021-04-27 10:58:01 -07:00
cpp_extension.rst	correct some cpp extension code usages and documents (#39766)	2020-06-10 08:31:22 -07:00
cpp_index.rst
cuda.rst	breakup optim, cuda documentation (#55673)	2021-04-14 12:44:00 -07:00
cudnn_persistent_rnn.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
cudnn_rnn_determinism.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
data.rst	[DataLoader][doc] Randomness for base_seed generator and NumPy seed (#56528)	2021-04-22 09:40:45 -07:00
ddp_comm_hooks.rst	[SPMD] Restrict DDP communication hooks to SPSD mode (#55253)	2021-04-05 16:46:47 -07:00
distributed.optim.rst	[Reland] Update and expose ZeroRedundancyOptimizer docs (#53112)	2021-03-02 14:16:12 -08:00
distributed.rst	update distributed doc table for alltoall nccl (#54277)	2021-03-19 15:35:10 -07:00
distributions.rst	Add sample validation for LKJCholesky.log_prob (#52763)	2021-02-25 16:12:29 -08:00
dlpack.rst	Lint trailing newlines (#54737)	2021-03-30 13:09:52 -07:00
docutils.conf
fft.rst	Use autosummary on torch.fft, torch.linalg (#55748)	2021-04-13 12:02:36 -07:00
futures.rst	[JIT x RPC] Consolidate Future type class and Future impl class (#40406)	2020-06-24 01:44:49 -07:00
fx.rst	[FX][docs] Render inherited methods in fx.Tracer API reference (#53630)	2021-03-09 14:30:41 -08:00
hub.rst	Add a torch.hub.load_local() function that can load models from any local directory with a hubconf.py (#44204)	2020-09-21 14:17:21 -07:00
index.rst	Updated the tech docs to be consistent with other two descriptions (#56338)	2021-04-20 09:00:42 -07:00
jit_builtin_functions.rst	Lint trailing newlines (#54737)	2021-03-30 13:09:52 -07:00
jit_language_reference_v2.rst	Remove duplicate entry for filter in language ref v2 (#57154)	2021-04-29 04:52:50 -07:00
jit_language_reference.rst	add type annotations to torch.nn.modules.conv (#49564)	2021-01-15 11:16:11 -08:00
jit_python_reference.rst	[JIT] Add support for with statements (#34705)	2020-06-18 16:57:18 -07:00
jit_unsupported.rst	[JIT] Update docs for recently added features (#45232)	2020-09-28 18:17:42 -07:00
jit.rst	Remove caption for Lang Reference (#56526)	2021-04-20 14:33:42 -07:00
linalg.rst	Add torch.linalg.cholesky_ex without checking for errors by default (#56724)	2021-05-01 18:48:27 -07:00
math-quantizer-equation.png
mobile_optimizer.rst	Mod lists to neutral+descriptive terms in caffe2/docs (#49803)	2020-12-23 11:37:11 -08:00
model_zoo.rst
multiprocessing.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
name_inference.rst	Add CSR (compressed sparse row) layout for sparse tensors (#50937)	2021-04-12 10:09:12 -07:00
named_tensor.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
nn.functional.rst	Fix bug in gaussian_nll_loss (#56469)	2021-04-22 07:43:48 -07:00
nn.init.rst
nn.rst	docs: separate autosummary for flatten layers (#54663)	2021-03-29 10:23:34 -07:00
onnx.rst	[ONNX] Add hardsigmoid symbolic in opset 9 #49649 (#54193)	2021-04-07 14:28:31 -07:00
optim.rst	To add single and chained learning schedulers to docs (#56705)	2021-04-23 09:36:00 -07:00
package.rst	torch.package change Folder to Directory and add doc strings (#56925)	2021-04-28 13:03:12 -07:00
pipeline.rst	Add tutorials to pipeline docs. (#55209)	2021-04-05 20:01:00 -07:00
profiler.rst	docs: fix profiler docstring (#55750)	2021-04-13 00:23:14 -07:00
quantization-support.rst	[docs][quant] Add fx graph mode quant api doc (#55306)	2021-04-05 13:56:23 -07:00
quantization.rst	Fix broken link to fx graph quant guide in quantization.rst (#56776)	2021-04-26 08:22:28 -07:00
random.rst	Remove duplicated entries in random.rst (#39725)	2020-06-10 16:51:15 -07:00
rpc.rst	Add 'remote_parameters' and 'get_module_rref' to RemoteModule docs. (#54645)	2021-03-26 21:41:28 -07:00
sparse.rst	Add CSR (compressed sparse row) layout for sparse tensors (#50937)	2021-04-12 10:09:12 -07:00
special.rst	[special] Add xlog1py (#55138)	2021-04-30 05:51:13 -07:00
storage.rst	Lint trailing newlines (#54737)	2021-03-30 13:09:52 -07:00
tensor_attributes.rst	Remove legacy constructor calls from pytorch codebase. (#54142)	2021-04-11 15:45:17 -07:00
tensor_view.rst	Add torch.swapdims and torch.swapaxes (#46041)	2020-11-18 11:35:53 -08:00
tensorboard.rst
tensors.rst	Support __pos__ and positive (#55891)	2021-04-27 13:23:59 -07:00
torch.nn.intrinsic.qat.rst	[quantization] Add some support for 3d operations (#50003)	2021-03-10 16:40:35 -08:00
torch.nn.intrinsic.quantized.rst	Lint trailing newlines (#54737)	2021-03-30 13:09:52 -07:00
torch.nn.intrinsic.rst	[quantization] Add some support for 3d operations (#50003)	2021-03-10 16:40:35 -08:00
torch.nn.qat.rst	Lint trailing newlines (#54737)	2021-03-30 13:09:52 -07:00
torch.nn.quantized.dynamic.rst	Forbid trailing whitespace (#53406)	2021-03-05 17:22:55 -08:00
torch.nn.quantized.rst	[quant] add docs for embedding/embedding_bag (#51770)	2021-02-05 11:43:15 -08:00
torch.overrides.rst	Add documentation for torch.overrides submodule. (#48170)	2020-11-30 11:25:31 -08:00
torch.quantization.rst	Lint trailing newlines (#54737)	2021-03-30 13:09:52 -07:00
torch.rst	Support __pos__ and positive (#55891)	2021-04-27 13:23:59 -07:00
type_info.rst	DOC: split quantization.rst into smaller pieces (#41321)	2020-07-25 23:59:40 -07:00