Driss Guessous 35c611d30f Add mem efficient backend flag (#87946) ... # Summary Add in a torch.backends.cuda flag and update context manager to pic between the three implementations of the scaled_dot_product_attention. cc @cpuhrsch @jbschlosser @bhosmer @mikaylagawarecki Pull Request resolved: https://github.com/pytorch/pytorch/pull/87946 Approved by: https://github.com/cpuhrsch		2022-10-28 15:51:10 +00:00
..
_static	[maskedtensor] add docs (#84887)	2022-10-19 20:44:34 +00:00
_templates	Fix left nav (#78552)	2022-06-01 00:49:53 +00:00
community	Add General Project Policies (#87385)	2022-10-20 21:02:09 +00:00
elastic	Add watchdog to TorchElastic agent and trainers (#84081)	2022-09-07 00:17:20 +00:00
notes	Fix typo under docs directory (#87583)	2022-10-24 23:52:44 +00:00
rpc
scripts	[ONNX] Update ONNX documentation to include unsupported operators (#84496)	2022-09-16 23:48:37 +00:00
amp.rst	Remove deprecated torch.matrix_rank (#70981)	2022-09-22 17:40:46 +00:00
autograd.rst	Change torch.autograd.graph.disable_saved_tensors_hooks to be public API (#85994)	2022-10-03 16:25:01 +00:00
backends.rst	Add mem efficient backend flag (#87946)	2022-10-28 15:51:10 +00:00
benchmark_utils.rst
bottleneck.rst	add itt unit test and docstrings (#84848)	2022-09-28 01:39:58 +00:00
checkpoint.rst
complex_numbers.rst	Add a note on CUDA 11.6 (#80363)	2022-06-27 21:34:24 +00:00
conf.py	Unify SymIntNode and SymFloatNode into SymNode (#87817)	2022-10-27 20:56:02 +00:00
config_mod.rst
cpp_extension.rst
cpp_index.rst
cuda._sanitizer.rst	Rework printing tensor aliases in CSAN error message (#85008)	2022-09-21 13:41:52 +00:00
cuda.rst	(Re-open) Adds cudaMallocAsync as an alternative backend for the CUDA allocator (#82682)	2022-10-12 03:44:21 +00:00
cudnn_persistent_rnn.rst
cudnn_rnn_determinism.rst
data.rst	Extend collate function that can register collate functions to handle specific types (#85748)	2022-09-30 13:30:18 +00:00
ddp_comm_hooks.rst	Fix two small typos in ddp_comm_hooks.rst (#82047)	2022-07-23 19:10:57 +00:00
deploy.rst	Delete torch::deploy from pytorch core (#85953)	2022-10-06 07:20:16 +00:00
distributed.algorithms.join.rst
distributed.elastic.rst
distributed.optim.rst
distributed.rst	[docs] batch_isend_irecv and P2POp of torch.distributed (#86438)	2022-10-25 00:11:50 +00:00
distributions.rst
dlpack.rst
docutils.conf
fft.rst
fsdp.rst
futures.rst
fx.rst	CSE Pass and common pass Tests (#81742)	2022-07-22 03:45:09 +00:00
hub.rst
index.rst	Set up new module torch.signal.windows (#85599)	2022-10-14 11:33:32 +00:00
jit_builtin_functions.rst
jit_language_reference_v2.rst	Fix typos in docs (#80602)	2022-08-29 23:32:44 +00:00
jit_language_reference.rst	(Re-open) Adds cudaMallocAsync as an alternative backend for the CUDA allocator (#82682)	2022-10-12 03:44:21 +00:00
jit_python_reference.rst
jit_unsupported.rst	(Re-open) Adds cudaMallocAsync as an alternative backend for the CUDA allocator (#82682)	2022-10-12 03:44:21 +00:00
jit_utils.rst	Create __init__.py (#78629)	2022-06-03 18:14:21 +00:00
jit.rst	torch.jit doc link for nvfuser readme.md (#77780)	2022-07-07 23:25:35 +00:00
library.rst	Add docs for Python Registration	2022-06-13 23:21:23 +00:00
linalg.rst	[Array API] Add linalg.vecdot (#70542)	2022-07-12 14:28:54 +00:00
masked.rst	[maskedtensor] fix docs formatting (#87387)	2022-10-20 20:48:25 +00:00
math-quantizer-equation.png
mobile_optimizer.rst
model_zoo.rst
monitor.rst
multiprocessing.rst
name_inference.rst
named_tensor.rst	Add torch.unflatten and improve its docs (#81399)	2022-07-29 15:02:42 +00:00
nested.rst	Fix typo under docs directory (#87583)	2022-10-24 23:52:44 +00:00
nn.functional.rst	Add Dropout1d module	2022-06-15 14:39:07 +00:00
nn.init.rst	update nn.init doc to reflect the no_grad (#80882)	2022-07-07 17:19:29 +00:00
nn.rst	Add Dropout1d module	2022-06-15 14:39:07 +00:00
onnx_supported_aten_ops.rst	[ONNX] Update ONNX documentation to include unsupported operators (#84496)	2022-09-16 23:48:37 +00:00
onnx.rst	[ONNX] Update user documentation (#85819)	2022-09-30 19:35:34 +00:00
optim.rst	[doc] LR scheduler example fix (#86629)	2022-10-11 21:41:50 +00:00
package.rst	Fix typos in torch.package documentation (#82994)	2022-08-08 20:19:17 +00:00
pipeline.rst
profiler.rst	Fix ITT unit-tests if PyTorch is compiled with USE_ITT=OFF (#86199)	2022-10-04 21:57:05 +00:00
quantization-accuracy-debugging.rst	Fix typo under docs directory (#87583)	2022-10-24 23:52:44 +00:00
quantization-backend-configuration.rst
quantization-support.rst	[quant][ao_migration] nn.intrinsic.quantized migration to ao (#86172)	2022-10-08 00:01:38 +00:00
quantization.rst	[ao] Adding FAQ to docs (#87322)	2022-10-25 20:18:04 +00:00
random.rst
rpc.rst	Fix typo under docs directory and RELEASE.md (#85896)	2022-09-29 21:41:59 +00:00
signal.rst	Reimplement Kaiser window (#87330)	2022-10-27 21:01:01 +00:00
sparse.rst	Fix typo under docs directory (#87583)	2022-10-24 23:52:44 +00:00
special.rst	[primTorch] special: j0, j1, spherical_j0 (#86049)	2022-10-04 18:21:46 +00:00
storage.rst	Fix typos in docs (#80602)	2022-08-29 23:32:44 +00:00
tensor_attributes.rst	Chore: Add 'mps' to the docs of tensor_attributes (#86585)	2022-10-14 19:59:33 +00:00
tensor_view.rst
tensorboard.rst
tensors.rst	Remove deprecated torch.lstsq (#70980)	2022-09-23 00:16:55 +00:00
testing.rst	Fix links in torch.testing docs (#80353)	2022-07-11 19:15:53 +00:00
torch.ao.ns._numeric_suite_fx.rst
torch.ao.ns._numeric_suite.rst
torch.overrides.rst	Revert "Revert "Implement sym_sizes to create proper IR for sym ints representing tensor sizes (#76836)""	2022-05-18 18:40:57 +00:00
torch.rst	[primTorch] Add a ref for narrow_copy (#86748)	2022-10-17 10:16:05 +00:00
type_info.rst	ENH: Convert finfo.tiny to finfo.smallest_normal (#76292)	2022-05-20 00:59:48 +00:00