pytorch/torch
Lakshay Garg a4110fedcf Use insert_or_assign instead of erase+emplace (#164868)
insert_or_assign does effectively the same thing as
erase+emplace but more efficiently since the search
does not need to be repeated

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164868
Approved by: https://github.com/eqy
2025-10-08 19:13:49 +00:00
..
_awaits
_C Add split-K control to cuBLAS reduced-precision settings (#164766) 2025-10-08 18:48:45 +00:00
_C_flatbuffer
_custom_op [2/N] Fix ruff warnings (#164460) 2025-10-04 03:40:32 +00:00
_decomp remove more (#164753) 2025-10-08 14:23:38 +00:00
_dispatch Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
_dynamo Add split-K control to cuBLAS reduced-precision settings (#164766) 2025-10-08 18:48:45 +00:00
_export Pyrefly suppressions 7/n (#164913) 2025-10-08 07:27:17 +00:00
_functorch Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
_higher_order_ops Revert "fix flex attention eager bwd: more rounding (#164317)" 2025-10-08 14:29:10 +00:00
_inductor Revert "Limit path search within range (#164581)" 2025-10-08 14:56:21 +00:00
_lazy Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
_library [opaque_obj] Add make_fx tracing support (#163278) 2025-10-08 09:09:16 +00:00
_logging Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
_numpy [2/N] Fix ruff warnings (#164460) 2025-10-04 03:40:32 +00:00
_prims Add pyrefly suppressions 2/n (#164513) 2025-10-03 02:46:13 +00:00
_prims_common remove more (#164753) 2025-10-08 14:23:38 +00:00
_refs Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
_strobelight Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
_subclasses remove more (#164753) 2025-10-08 14:23:38 +00:00
_vendor
accelerator
amp Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
ao Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
autograd Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
backends Add split-K control to cuBLAS reduced-precision settings (#164766) 2025-10-08 18:48:45 +00:00
compiler Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
contrib
cpu Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
csrc Use insert_or_assign instead of erase+emplace (#164868) 2025-10-08 19:13:49 +00:00
cuda Revert "[CUDA] Add experimental green context support for SM carveout (#159104)" 2025-10-07 20:51:22 +00:00
distributed [PP]: Optimize memory by early releasing stage inputs' gradients (#164329) 2025-10-08 16:12:00 +00:00
distributions Add pyrefly suppressions (#164748) 2025-10-07 17:31:18 +00:00
export Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
fft
func
futures [5/N] Apply ruff UP035 rule (#164423) 2025-10-02 07:31:11 +00:00
fx Fix refine_ranges corner case (#164075) (#164846) 2025-10-08 18:42:37 +00:00
headeronly Move version.h to torch/headeronly (#164381) 2025-10-07 17:47:30 +00:00
jit Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
legacy
lib
linalg Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
masked Add pyrefly suppressions (3/n) (#164588) 2025-10-03 22:03:03 +00:00
monitor
mps Add type annotations to MPS profiler utilities (#163486) 2025-09-27 23:00:53 +00:00
mtia Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
multiprocessing Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
nativert Add SVE128 ISA (#158932) 2025-09-29 14:49:19 +00:00
nested use check_size instead of check_is_size in ops.py (#164668) 2025-10-08 14:23:38 +00:00
nn Pyrefly suppressions 7/n (#164913) 2025-10-08 07:27:17 +00:00
numa Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
onnx [BC-Breaking] Remove long-deprecated casting functions from native_functions.yaml (#164641) 2025-10-08 08:27:58 +00:00
optim Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
package Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
profiler Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
quantization Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
signal Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
sparse Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
special
testing Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
utils Pyrefly suppressions 7/n (#164913) 2025-10-08 07:27:17 +00:00
xpu Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
__config__.py
__future__.py
__init__.py CUDA 13.0 builds fix on Amazon Linux 2023 (#164870) 2025-10-07 22:52:53 +00:00
_appdirs.py
_classes.py
_compile.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
_custom_ops.py
_environment.py
_guards.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
_jit_internal.py Revert "Enable all SIM rules except disabled ones (#164645)" 2025-10-05 19:32:21 +00:00
_linalg_utils.py
_lobpcg.py Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
_lowrank.py
_meta_registrations.py remove more (#164753) 2025-10-08 14:23:38 +00:00
_namedtensor_internals.py
_ops.py Revert "Enable all SIM rules except disabled ones (#164645)" 2025-10-05 19:32:21 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py Revert "Support setting grad_dtype on leaf tensors (#162815)" 2025-10-03 23:14:28 +00:00
_tensor_str.py Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
_tensor.py Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
_thread_safe_fork.py
_torch_docs.py Update documentation for torch.index_select (#163616) 2025-09-25 18:29:17 +00:00
_utils_internal.py Pyrefly suppressions 6/n (#164877) 2025-10-08 02:30:57 +00:00
_utils.py Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
_VF.py
_vmap_internals.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
_weights_only_unpickler.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
CMakeLists.txt Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)" 2025-09-25 13:47:46 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py [2/N] Fix ruff warnings (#164460) 2025-10-04 03:40:32 +00:00
header_only_apis.txt Migrate DeviceType to torch/headeronly (#163999) 2025-09-30 23:13:27 +00:00
hub.py Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
library.h
library.py Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
overrides.py Revert "Support setting grad_dtype on leaf tensors (#162815)" 2025-10-03 23:14:28 +00:00
py.typed
quasirandom.py
random.py Revert "Add device argument to torch.random.get_rng_state (#163034)" 2025-10-04 15:25:45 +00:00
return_types.py
script.h
serialization.py Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
storage.py Add initial suppressions for pyrefly (#164177) 2025-10-02 20:57:41 +00:00
torch_version.py
types.py [4/N] Apply ruff UP035 rule to python code (#164206) 2025-10-01 19:05:53 +00:00
version.py.tpl