pytorch/torch
zpcore 50d8168c8b [DTensor] Support in gradient placement for local_map() (#155181)
Support `in_grad_placements` argument in torch.distributed.tensor.experimental.local_map().  The argument helps enforce placement of gradient of the input Dtensor.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155181
Approved by: https://github.com/wanchaol
2025-06-12 17:07:04 +00:00
..
_awaits
_C Move _storage_Use_Count to be gerneric (#155451) 2025-06-12 01:39:04 +00:00
_C_flatbuffer
_custom_op
_decomp Fix clamp type promotion in inductor decomposition (#154471) 2025-05-28 23:24:25 +00:00
_dispatch
_dynamo Support tuning of _grouped_mm (#153953) 2025-06-12 15:39:35 +00:00
_export [export] Fix serialization for call_torchbind hop with as_none argument (#155647) 2025-06-12 02:59:03 +00:00
_functorch Revert "[PT2][partitioners] Add aten.split to view_ops list (#155424)" 2025-06-12 01:38:34 +00:00
_higher_order_ops [refactor] make do_auto_functionalize_v2 take HopInstance (#154192) 2025-06-11 22:52:37 +00:00
_inductor [inductor][cutlass backend] Log prescreening elpase (#155508) 2025-06-12 16:48:52 +00:00
_lazy
_library Custom Op handle 1-element tuples (#155447) 2025-06-11 03:43:40 +00:00
_logging Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
_numpy fix numpy compatibility for 2d small list indices (#154806) 2025-06-04 01:58:52 +00:00
_prims Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
_prims_common Add view_simple as meta function for view, and avoid calling reshape_view_helper. (#154757) 2025-06-12 09:58:15 +00:00
_refs [dynamic shapes] skip fused linear path if not definitely contiguous (#155051) 2025-06-12 15:55:21 +00:00
_strobelight
_subclasses Add view_simple as meta function for view, and avoid calling reshape_view_helper. (#154757) 2025-06-12 09:58:15 +00:00
_vendor
accelerator Add torch.accelerator.device_index as accelerator's device switch context (#148864) 2025-04-25 09:45:25 +00:00
amp [Intel GPU] skip a cuda api call in amp to save some host overhead on xpu (#151111) 2025-04-13 06:37:07 +00:00
ao Fix incorrect get_default_qat_qconfig in prepare_qat_fx docs. (#155100) 2025-06-04 18:51:40 +00:00
autograd set_grad_enabled add str and repr for prints (#155681) 2025-06-11 16:01:03 +00:00
backends [coreml-backend-tool] fix pytorch-backended issue on new coremltools (#155543) 2025-06-11 20:52:26 +00:00
compiler Combine sticky pgo key with job id (#154863) 2025-06-03 07:58:38 +00:00
contrib
cpu [device_mesh] improve device selection logic (#150897) 2025-05-14 06:29:16 +00:00
csrc [Easy][Code Clean] Remove the unused and undefined function in pickler (#155772) 2025-06-12 13:03:36 +00:00
cuda [Memory Snapshot] Add Flag to Toggle Global and Local Callbacks for Annotations (#154932) 2025-06-04 23:15:19 +00:00
distributed [DTensor] Support in gradient placement for local_map() (#155181) 2025-06-12 17:07:04 +00:00
distributions Type hints for distributions/utils (#154712) 2025-05-30 15:50:31 +00:00
export [Export] Fix some typos in docstring (#155485) 2025-06-11 16:44:38 +00:00
fft
func
futures Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
fx unify symbolic_shapes and sizevars dynamic shapes APIs naming 1 (#154774) 2025-06-12 16:11:55 +00:00
jit Fix broken URLs (#152237) 2025-04-27 09:56:42 +00:00
legacy
lib Revert "Use 3.27 as the minimum CMake version (#153153)" 2025-05-31 02:14:24 +00:00
linalg Fix for ambiguity in linalg.norm()'s ord argument of +2 & -2 (#155148) 2025-06-04 21:15:20 +00:00
masked Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
monitor
mps
mtia Add getDeviceProperties api to torch mtia device (#153577) 2025-05-27 11:55:58 +00:00
multiprocessing
nativert [nativert] Move DelegateExecutor to PyTorch core (#155581) 2025-06-12 04:33:31 +00:00
nested Add torch.Tensor._make_wrapper_subclass to torch/_C/__init__.pyi (#154022) 2025-05-27 14:10:00 +00:00
nn Fix docs build (#155129) 2025-06-09 22:25:20 +00:00
onnx Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
optim Fix lr_scheduler unexpectedly calls step() when init argument last_epoch is larger than -1 (#149312) 2025-05-22 08:42:37 +00:00
package [BE]: Enable ruff YTT linter for Python version checks (#153547) 2025-05-14 21:09:16 +00:00
profiler [Profiler] Induce Inductor Import before Profiling (#155243) 2025-06-07 23:58:50 +00:00
quantization
signal
sparse fix numpy compatibility for 2d small list indices (#154806) 2025-06-04 01:58:52 +00:00
special Add doc for missing functions for torch.special module (#155074) 2025-06-09 22:28:26 +00:00
standalone/macros Move c10/macros/Export.h to torch/standalone (#154850) 2025-06-03 06:18:59 +00:00
testing [MPS] Migrate hardswish (forward and backward) to Metal kernel (#155479) 2025-06-11 20:58:46 +00:00
utils [inductor][triton pin] add support for new TMA API for mm.py templates (#155723) 2025-06-12 06:25:47 +00:00
xpu Correct torch.xpu.is_bf16_supported return False if no XPU detected (#152317) 2025-05-06 10:03:17 +00:00
__config__.py
__future__.py
__init__.py Replace frame_traced_fn hook with get_traced_code() util (#155249) 2025-06-10 22:40:58 +00:00
_appdirs.py Fix broken URLs (#152237) 2025-04-27 09:56:42 +00:00
_classes.py
_compile.py [precompile] Ensure @disable()-ed function won't trigger recompile from precompile bytecode. (#155363) 2025-06-10 16:13:38 +00:00
_custom_ops.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_deploy.py
_environment.py
_guards.py Replace frame_traced_fn hook with get_traced_code() util (#155249) 2025-06-10 22:40:58 +00:00
_jit_internal.py BE: Type previously untyped decorators (#154515) 2025-05-29 00:36:34 +00:00
_linalg_utils.py
_lobpcg.py Fixed rerr computation in lobpcg (#152789) 2025-05-08 12:22:31 +00:00
_lowrank.py
_meta_registrations.py Add view_simple as meta function for view, and avoid calling reshape_view_helper. (#154757) 2025-06-12 09:58:15 +00:00
_namedtensor_internals.py
_ops.py [hop] auto functionalize invoke_subgraph (#154072) 2025-06-11 22:52:28 +00:00
_python_dispatcher.py
_size_docs.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_sources.py
_storage_docs.py Fix docstring for torch.UntypedStorage.from_file (#155067) 2025-06-05 14:30:49 +00:00
_streambase.py
_tensor_docs.py [docs] Add docstring indicating UB for converting inf to int (#154781) 2025-06-10 14:04:50 +00:00
_tensor_str.py
_tensor.py Avoid triggering ignored requires_grad warning in our code (#152686) 2025-05-05 23:56:40 +00:00
_thread_safe_fork.py
_torch_docs.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
_utils_internal.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
_utils.py User-controlled sparse tensor validation when loading data from external storage (#154610) 2025-06-02 10:17:07 +00:00
_VF.py
_vmap_internals.py Fix broken URLs (#152237) 2025-04-27 09:56:42 +00:00
_weights_only_unpickler.py
CMakeLists.txt Move c10/macros/Export.h to torch/standalone (#154850) 2025-06-03 06:18:59 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py Optimize cdist param description (#151178) 2025-04-14 13:53:10 +00:00
header_only_apis.txt Move c10/macros/Export.h to torch/standalone (#154850) 2025-06-03 06:18:59 +00:00
hub.py
library.h Overload Library::def rather than templating it (#151626) 2025-04-18 22:51:16 +00:00
library.py Render Example: and not Example:: in docs (#153978) 2025-05-21 01:03:26 +00:00
overrides.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
py.typed
quasirandom.py
random.py Update description for torch.random.fork_rng (#151881) 2025-04-23 16:59:29 +00:00
return_types.py
script.h
serialization.py Update serialization docs (#153631) 2025-05-19 20:22:07 +00:00
storage.py [Multiprocesing] Fix _release_ipc_counter missing in rebuilding cuda ipc tensor with UntypedStorage (#155312) 2025-06-12 10:41:58 +00:00
torch_version.py
types.py
version.py.tpl