Benjamin Glass 4959784dac Add API query for available per-process CUDA memory (#140620) ... Certain `cpp_wrapper`-enabled tests were OOM-ing in the CI pipeline, with error messages suggesting that sufficient memory was accessible. This ultimately resulted from an internal memory limitation that was not queryable in the API. This PR adds querying for that limit. Additionally, the failing tests had incorrect memory availability checks, and are updated with measured memory requirements. Pull Request resolved: https://github.com/pytorch/pytorch/pull/140620 Approved by: https://github.com/malfet, https://github.com/eqy ghstack dependencies: #141367		2024-12-03 00:24:03 +00:00
..
_dynamo	[ca] expose option to collect sizes as dynamic (#141153)	2024-11-22 19:26:27 +00:00
__init__.pyi.in	Add API query for available per-process CUDA memory (#140620)	2024-12-03 00:24:03 +00:00
_aoti.pyi	[aoti] Add cpp loader (#135374)	2024-09-11 03:00:01 +00:00
_autograd.pyi	[Profiler] Add API for Dynamic Activity Toggling [2/n] (#133035)	2024-08-09 21:54:54 +00:00
_cpu.pyi	[Inductor][CPP] Add oneDNN BRGEMM config for Half cpp gemm template (#136255)	2024-11-05 05:33:29 +00:00
_cudnn.pyi
_cusparselt.pyi	[sparse] Add cuSPARSELt as a backend (#128534)	2024-08-21 22:06:07 +00:00
_distributed_autograd.pyi	Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)	2024-06-29 09:23:39 +00:00
_distributed_c10d.pyi	Implements user buffer registration using MemPool (#133603)	2024-11-21 01:40:11 +00:00
_distributed_rpc_testing.pyi	Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)	2024-06-29 09:23:39 +00:00
_distributed_rpc.pyi	Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)	2024-06-29 09:23:39 +00:00
_export.pyi	[export] Implement cpp deserializer. (#136398)	2024-11-14 16:34:59 +00:00
_functions.pyi	type annotations for meta_utils (#140203)	2024-11-13 20:07:47 +00:00
_functorch.pyi	Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)	2024-06-29 09:23:39 +00:00
_instruction_counter.pyi	Add compile time instruction count metric (#133834)	2024-08-27 23:29:02 +00:00
_itt.pyi
_lazy_ts_backend.pyi	Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)	2024-06-29 09:23:39 +00:00
_lazy.pyi	Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)	2024-06-29 09:23:39 +00:00
_monitor.pyi	Use Generic TypeAlias (PEP 585) and Union Type (PEP 604) in .pyi stub files (#129419)	2024-06-29 09:23:39 +00:00
_nn.pyi.in	Add padding_side to pad_sequence with "left" and "right" options ("right" as default) (#131884)	2024-08-07 15:53:07 +00:00
_nvtx.pyi	Flip default value for mypy disallow_untyped_defs [1/11] (#127838)	2024-06-08 18:16:33 +00:00
_onnx.pyi	[1/N] [Caffe2] Remove caffe2_aten_fallback code (#128675)	2024-06-17 21:25:59 +00:00
_profiler.pyi	[Profiler] Create Auto-Trace Frontend for Trace ID (#139310)	2024-10-31 19:02:57 +00:00
_VariableFunctions.pyi.in	Flip default value for mypy disallow_untyped_defs [1/11] (#127838)	2024-06-08 18:16:33 +00:00
_verbose.pyi
build.bzl
return_types.pyi.in	[torchgen] reference generated comment to actual location of the generator and template (#130020)	2024-07-05 21:47:14 +00:00