frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
c509a7864591a0bf517602a728c77e0266203102
pytorch/torch
History
Chien-Chin Huang 8461b63f2c [CP] Replace context_parallel context manager with functional APIs (#164500) 
`context_parallel()` being a context manager has annoyed users. Now that we plan to redesign CP's UX to explicitly ask users to:

1. Wrap the attention op into an `nn.Module`
2. Lift any buffers that are not sequence agnostic to input

We can replace `context_parallel()` with two functional APIs: `_context_parallel_shard` and `_enable_context_parallel_dispatcher`.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164500
Approved by: https://github.com/XilunWu
ghstack dependencies: #162542
2025-10-13 06:30:18 +00:00
..
_awaits
…
_C
[opaque_obj_v2] PyObject custom op schema type (#165004)
2025-10-10 21:31:56 +00:00
_C_flatbuffer
…
_custom_op
[2/N] Fix ruff warnings (#164460)
2025-10-04 03:40:32 +00:00
_decomp
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
_dispatch
Add initial suppressions for pyrefly (#164177)
2025-10-02 20:57:41 +00:00
_dynamo
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
_export
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
_functorch
Revert "[compile] Regional inductor compilation with fx.annotate (#164776)"
2025-10-11 23:14:23 +00:00
_higher_order_ops
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
_inductor
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
_lazy
Add initial suppressions for pyrefly (#164177)
2025-10-02 20:57:41 +00:00
_library
[opaque_obj_v2] PyObject custom op schema type (#165004)
2025-10-10 21:31:56 +00:00
_logging
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
_numpy
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
_prims
Add pyrefly suppressions 2/n (#164513)
2025-10-03 02:46:13 +00:00
_prims_common
[2/N] Use "is" in python type comparison (#165142)
2025-10-10 15:36:44 +00:00
_refs
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
_strobelight
Add initial suppressions for pyrefly (#164177)
2025-10-02 20:57:41 +00:00
_subclasses
Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939)
2025-10-11 01:03:55 +00:00
_vendor
…
accelerator
Add unified memory APIs for torch.accelerator (#152932)
2025-08-08 17:41:22 +00:00
amp
Revert "[AMP][Refactor] Simplify dtype support logic in autocast context manager (#163446)"
2025-10-10 15:12:46 +00:00
ao
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
autograd
Revert "Warn if AccumulateGrad stream does not match producer node stream (#165065)"
2025-10-10 17:29:29 +00:00
backends
Revert "Add SVE128 ISA (#158932)"
2025-10-10 01:17:02 +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
LocalTensor (#164537)
2025-10-12 20:06:41 +00:00
cuda
error message for instantiating CUDA Stream if CUDA not available (#159868)
2025-10-11 23:21:35 +00:00
distributed
[CP] Replace context_parallel context manager with functional APIs (#164500)
2025-10-13 06:30:18 +00:00
distributions
[1/N] Use "is" in python type comparison (#165037)
2025-10-10 12:36:50 +00:00
export
Do not decompose in functionalization/proxy tensor if autograd wouldn't have decomposed (#164939)
2025-10-11 01:03:55 +00:00
fft
[BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553)
2025-06-17 08:18:47 +00:00
func
…
futures
[5/N] Apply ruff UP035 rule (#164423)
2025-10-02 07:31:11 +00:00
fx
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
headeronly
Mark unused parameters in C++ code (#164912)
2025-10-09 06:23:25 +00:00
jit
Fix missing brackets (#165138)
2025-10-10 17:23:31 +00:00
legacy
…
lib
[CodeClean] Replace std::runtime_error with TORCH_CHECK (#164129)
2025-10-09 19:01:07 +00:00
linalg
Add initial suppressions for pyrefly (#164177)
2025-10-02 20:57:41 +00:00
masked
[1/N] Use "is" in python type comparison (#165037)
2025-10-10 12:36:50 +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
Revert "Add SVE128 ISA (#158932)"
2025-10-10 01:17:02 +00:00
nested
[2/N] Use "is" in python type comparison (#165142)
2025-10-10 15:36:44 +00:00
nn
[2/N] Use "is" in python type comparison (#165142)
2025-10-10 15:36:44 +00:00
numa
Add initial suppressions for pyrefly (#164177)
2025-10-02 20:57:41 +00:00
onnx
Revert "[BC-Breaking] Remove long-deprecated casting functions from native_functions.yaml (#164641)"
2025-10-09 15:42:51 +00:00
optim
Revert "[2/N] More ruff SIM fixes (#165031)"
2025-10-10 13:42:14 +00:00
package
[1/N] Use "is" in python type comparison (#165037)
2025-10-10 12:36:50 +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
[BE][PYFMT] migrate PYFMT for torch/[p-z]*/ to ruff format (#144552)
2025-08-07 00:09:56 +00:00
testing
LocalTensor (#164537)
2025-10-12 20:06:41 +00:00
utils
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +00:00
xpu
Add a new API torch.xpu.is_tf32_supported for Intel GPU (#163141)
2025-10-12 12:11:57 +00:00
__config__.py
…
__future__.py
…
__init__.py
[2/N] Use "is" in python type comparison (#165142)
2025-10-10 15:36:44 +00:00
_appdirs.py
…
_classes.py
remove allow-untyped-defs from torch/_classes.py (#157231)
2025-07-08 00:11:52 +00:00
_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 "[2/N] More ruff SIM fixes (#165031)"
2025-10-10 13:42:14 +00:00
_linalg_utils.py
Update is_sparse doc to mention that it is sparse_coo specific (#157378)
2025-07-09 18:22:14 +00:00
_lobpcg.py
Add initial suppressions for pyrefly (#164177)
2025-10-02 20:57:41 +00:00
_lowrank.py
[BE][1/16] fix typos in torch/ (#156311)
2025-07-09 11:02:22 +00:00
_meta_registrations.py
[Fix] missing lambda in torch._check (#165043)
2025-10-10 17:11:55 +00:00
_namedtensor_internals.py
…
_ops.py
Revert "[2/N] More ruff SIM fixes (#165031)"
2025-10-10 13:42:14 +00:00
_python_dispatcher.py
Typo fixes for "overridden" in comments and function names (#155944)
2025-06-14 03:37:38 +00:00
_size_docs.py
…
_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
[reland] Allow setting grad_dtype on leaf tensors (#164751)
2025-10-08 20:23:13 +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
Revert "Call internal log_compilation_event if it exists (#164855)"
2025-10-09 22:38:45 +00:00
_utils.py
Enable ruff rule E721 (#165162)
2025-10-13 01:48:55 +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
Mark unused parameters in C++ code (#164912)
2025-10-09 06:23:25 +00:00
custom_class.h
Mark unused parameters in C++ code (#164912)
2025-10-09 06:23:25 +00:00
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
Mark unused parameters in C++ code (#164912)
2025-10-09 06:23:25 +00:00
library.py
Add initial suppressions for pyrefly (#164177)
2025-10-02 20:57:41 +00:00
overrides.py
Add scaled_mm python API, test (#164142)
2025-10-09 12:43:18 +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
[2/N] Use "is" in python type comparison (#165142)
2025-10-10 15:36:44 +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
…