Angela Yi 58268137f1 [pytree] Allow register_pytree_node to take in 5 inputs (#108256) ... Summary: This is currently breaking internal old torch.packages from when _register_pytree_node took 5 inputs. Test Plan: CI Differential Revision: D48834742 Pull Request resolved: https://github.com/pytorch/pytorch/pull/108256 Approved by: https://github.com/zhxchen17, https://github.com/zou3519		2023-08-31 11:06:17 +00:00
..
_sympy	[inductor] Ignore sympy.PolynomialError while simplifying (#108280)	2023-08-30 22:57:15 +00:00
backcompat
benchmark	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00
bottleneck	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00
data	fix typos (#108006)	2023-08-28 19:49:09 +00:00
hipify	Expose cudaStreamCaptureMode in CUDA Graphs, use local setting in inductor (#107407)	2023-08-25 01:44:26 +00:00
jit	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00
model_dump
tensorboard	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00
viz	MemoryViz: print stream 0 if other streams exist (#106483)	2023-08-03 00:42:13 +00:00
__init__.py	Add torch.utils to the docs page, remove dead code and fix docstrings (#105142)	2023-07-26 14:24:58 +00:00
_content_store.py	Make conversions from/to sparse semi-structured always @torch.compile-d (#105272)	2023-07-18 04:51:28 +00:00
_contextlib.py	grad_mode decorators without paren (#107086)	2023-08-15 05:25:33 +00:00
_cpp_extension_versioner.py
_cuda_trace.py
_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	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00
_mode_utils.py	Switch most Python RAII guard usages to context manager (#102642)	2023-06-01 16:28:37 +00:00
_python_dispatch.py	add return_and_correct_aliasing() util for wrapper subclasses (#107915)	2023-08-29 14:27:19 +00:00
_pytree.py	[pytree] Allow register_pytree_node to take in 5 inputs (#108256)	2023-08-31 11:06:17 +00:00
_stats.py
_traceback.py	Change how caching/cleanup for CapturedTraceback works (#107471)	2023-08-22 01:03:13 +00:00
_zip.py	[caffe2] Create deterministic zip archives (#102903)	2023-06-15 17:45:19 +00:00
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	Improve activation checkpoint docs wording (#107296)	2023-08-16 17:36:52 +00:00
collect_env.py	[BE]: Apply ruff PERF fixes to torch (#104917)	2023-07-11 20:45:21 +00:00
cpp_backtrace.py
cpp_extension.py	[BE]: Update ruff to 0.285 (#107519)	2023-08-22 23:16:38 +00:00
dlpack.py	[BE]: Update Ruff to 0.0.280 (#105724)	2023-07-22 23:03:34 +00:00
file_baton.py
flop_counter.py	[BE] f-stringify torch/ and scripts (#105538)	2023-07-21 19:35:24 +00:00
hooks.py	[BE]: Update ruff to 0.285 (#107519)	2023-08-22 23:16:38 +00:00
mkldnn.py
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	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00
weak.py	[BE] Enable ruff's UP rules and autoformat utils/ (#105424)	2023-07-18 20:17:25 +00:00