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
..
benchmark	Set RUNPATH so installed tests can find the required shared libraries (#136627)	2024-10-25 09:38:08 +00:00
core	Add admonition about as_float_unchecked() (#141742)	2024-11-28 06:25:18 +00:00
cuda	Add API query for available per-process CUDA memory (#140620)	2024-12-03 00:24:03 +00:00
hip	Fix hardcoded ROCm paths in Caffe2Targets.cmake (#136283)	2024-09-26 00:34:43 +00:00
macros	[PyTorch] Unbreak C10_ALWAYS_INLINE_ATTRIBUTE on MSVC (#139363)	2024-10-31 07:40:05 +00:00
mobile	[2/N] Fix extra warnings brought by clang-tidy-17 (#137459)	2024-10-08 19:05:02 +00:00
test	Remove unneeded std::make_optional (#141567)	2024-11-28 00:05:21 +00:00
util	Switch c10::string_view to std::string_view (#139635)	2024-11-27 01:41:18 +00:00
xpu	Friendly catch exception when fail to initialize XPU devices (#141658)	2024-11-28 05:17:08 +00:00
BUCK.oss
BUILD.bazel
build.bzl
CMakeLists.txt	[pytorch][monitoring] Dynamic backend for WaitCounter (#135967)	2024-09-15 18:07:49 +00:00
ovrsource_defs.bzl