rzou 0f768c7866 Barebones flat_apply HOP (#146060) ... This PR: - adds pytree.register_constant for registering a class to be treated as a constant by torch.compile/torch.fx - adds a very barebones flat_apply HOP. This should be sufficient to get mark_traceable working. A lot more work is necessary to get the custom operator case working (when make_fx sees a custom operator with PyTree arg types, it needs to emit a call to the flat_apply HOP). - I expect the flat_apply HOP to change a lot, I want to ship this in the current state to unblock the mark_traceable and custom ops work. Test Plan: - It's kind of difficult to test the barebones flat_apply HOP "works" so I added a really simple test. Pull Request resolved: https://github.com/pytorch/pytorch/pull/146060 Approved by: https://github.com/StrongerXi, https://github.com/yanboliang ghstack dependencies: #146059		2025-02-01 16:17:48 +00:00
..
_strobelight
_sympy	Allow replacing unbacked with very large upperbound by returning no-op for FloorToInt(int) (#146001)	2025-01-31 00:25:20 +00:00
backcompat
benchmark
bottleneck
data
hipify
jit
model_dump
serialization	Add option to serialization config to reduce random reads from get_record_offset when loading with mmap=True (#143880)	2025-01-31 17:09:20 +00:00
tensorboard
viz
__init__.py
_backport_slots.py
_config_module.py	config: Don't spam warnings about reference type configs (#145800)	2025-01-30 18:57:16 +00:00
_config_typing.pyi
_content_store.py
_contextlib.py
_cpp_embed_headers.py	PEP585: Missed conversions (#145342)	2025-01-29 05:24:36 +00:00
_cpp_extension_versioner.py
_cxx_pytree.py	[AOTI] Cache treespec_loads calculation (#145815)	2025-01-31 06:38:21 +00:00
_device.py
_exposed_in.py
_filelock.py
_foreach_utils.py
_freeze.py
_functools.py
_get_clean_triton.py
_import_utils.py
_mode_utils.py
_ordered_set.py	[BE]: Update typing of OrderedSet ancestor (#145783)	2025-01-28 04:43:49 +00:00
_python_dispatch.py
_pytree.py	Barebones flat_apply HOP (#146060)	2025-02-01 16:17:48 +00:00
_stats.py
_thunk.py
_traceback.py
_triton.py
_typing_utils.py
_zip.py
backend_registration.py
bundled_inputs.py
checkpoint.py
collect_env.py
cpp_backtrace.py
cpp_extension.py	Set -DPy_LIMITED_API flag for py_limited_api=True extensions (#145764)	2025-01-28 20:11:05 +00:00
deterministic.py
dlpack.py
file_baton.py
flop_counter.py
hooks.py
mkldnn.py
mobile_optimizer.py
model_zoo.py
module_tracker.py
show_pickle.py	Use typing.IO[bytes] instead of io.BytesIO in annotations (#144994)	2025-01-27 18:08:07 +00:00
throughput_benchmark.py
weak.py