soulitzer c9eb8d8d90 Add set_checkpoint_debug_enabled that overrides local setting (#110728) ... People access activation checkpoint through many layers of config and it is not always guaranteed that all the layers of wrapping around checkpoint properly propagate all the kwargs, e.g. debug mode. This context manager offers an alternative way to enable debug mode that bypasses the need for all layers to propagate kwargs. Pull Request resolved: https://github.com/pytorch/pytorch/pull/110728 Approved by: https://github.com/albanD ghstack dependencies: #110673, #110674, #110675, #110676		2023-10-11 02:12:31 +00:00
..
_sympy	[reland] Add symbolic singleton int (#110674)	2023-10-10 19:37:17 +00:00
backcompat
benchmark	Doc test non packages (#110568)	2023-10-06 14:16:01 +00:00
bottleneck
data	StackDataset batched sampling (#110694)	2023-10-10 22:05:51 +00:00
hipify	[CUDA Host Allocator] Add support of CudaHostRegister (#108488)	2023-10-06 04:13:02 +00:00
jit
model_dump
tensorboard	Add global_step parameter to SummaryWriter.add_hparams (#109572)	2023-09-22 12:37:01 +00:00
viz	MemoryViz: print stream 0 if other streams exist (#106483)	2023-08-03 00:42:13 +00:00
__init__.py
_content_store.py
_contextlib.py	grad_mode decorators without paren (#107086)	2023-08-15 05:25:33 +00:00
_cpp_extension_versioner.py
_cuda_trace.py
_cxx_pytree.py	[pytree] Extract reusable generic tests for pytree (#110395)	2023-10-04 23:40:50 +00:00
_device.py	asarray: take the default device into consideration. (#106779)	2023-08-11 13:16:42 +00:00
_foreach_utils.py	add xpu support for foreach kernels (#106021)	2023-08-09 07:51:02 +00:00
_freeze.py
_mode_utils.py
_python_dispatch.py	_return_and_correct_aliasing: fix for schemas with mutable tensor in kwargs (#109662)	2023-09-22 07:09:04 +00:00
_pytree.py	[pytree] Extract reusable generic tests for pytree (#110395)	2023-10-04 23:40:50 +00:00
_stats.py
_traceback.py	Change how caching/cleanup for CapturedTraceback works (#107471)	2023-08-22 01:03:13 +00:00
_triton.py	[Inductor] Generalize inductor triton backend device agnostic (#109486)	2023-09-24 07:49:20 +00:00
_zip.py
backend_registration.py	Fix the api of privateuse1 in comment (#106537)	2023-08-04 16:07:26 +00:00
bundled_inputs.py	[BE]: Update ruff to 0.285 (#107519)	2023-08-22 23:16:38 +00:00
checkpoint.py	Add set_checkpoint_debug_enabled that overrides local setting (#110728)	2023-10-11 02:12:31 +00:00
collect_env.py	Include onnx and onnxscript information in collect_env.py (#110560)	2023-10-05 01:29:04 +00:00
cpp_backtrace.py
cpp_extension.py	Add 9.0a to cpp_extension supported compute archs (#110587)	2023-10-05 17:41:06 +00:00
dlpack.py
file_baton.py
flop_counter.py	Add option to flop counter formula registration to get raw values (#110591)	2023-10-05 21:14:41 +00:00
hooks.py	[BE]: Update ruff to 0.285 (#107519)	2023-08-22 23:16:38 +00:00
mkldnn.py	add fp16 support for mkldnn conv and deconv on CPU (#99496)	2023-09-19 12:37:28 +00:00
mobile_optimizer.py	[BE]: Update ruff to 0.285 (#107519)	2023-08-22 23:16:38 +00:00
model_zoo.py
show_pickle.py
throughput_benchmark.py
weak.py