Edward Z. Yang 17eb649d55 Implement guard collectives (optimized version) (#156562) ... This is a remix of https://github.com/pytorch/pytorch/pull/155558 Instead of mediating guard collective via a config option, in this one it's done via a `set_stance` like API. The motivation is that checking for the config value on entry on torch.compile is apparently quite expensive, according to functorch_maml_omniglot. So this makes it a bit cheaper. Signed-off-by: Edward Z. Yang <ezyang@meta.com> Pull Request resolved: https://github.com/pytorch/pytorch/pull/156562 Approved by: https://github.com/Microve		2025-06-24 04:59:49 +00:00
..
__init__.py	Implement guard collectives (optimized version) (#156562)	2025-06-24 04:59:49 +00:00
_cache.py	[precompile] Implement PrecompileContext for recording precompile artifacts, integrate with CompilePackage (#154415)	2025-06-13 14:11:24 +00:00
config.py	Combine sticky pgo key with job id (#154863)	2025-06-03 07:58:38 +00:00