OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
02c7ab2f9b
pytorch/torch
History
Valentine233 02c7ab2f9b [cpp wrapper] add AOTI shim for collective ops (#154492) 
Implementations:
1. Move collective ops to c10d namespace, so that we can call them externally.
2. Add AOTI shims for collective ops.

Testing
1. Add c10d functional UT for cpu.
2. Include the above one in cpp wrapper UT.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/154492
Approved by: https://github.com/desertfire
2025-06-25 01:20:05 +00:00
..
_awaits
_C Revert "Add unified memory APIs for torch.accelerator (#152932)" 2025-06-25 00:11:35 +00:00
_C_flatbuffer
_custom_op pyfmt lint torch/_custom_op/* (#155782) 2025-06-12 23:04:11 +00:00
_decomp Fix non-bitwise type annotations for Tensor operators (see #145838) (#146845) 2025-06-24 15:41:34 +00:00
_dispatch Improve torch.ops typing (#154555) 2025-06-22 15:52:27 +00:00
_dynamo Enable C++ dynamic shape guards by default (#140756) 2025-06-24 21:10:17 +00:00
_export Improve torch.ops typing (#154555) 2025-06-22 15:52:27 +00:00
_functorch Replace deprecated is_compiling method (#154476) 2025-06-24 05:16:40 +00:00
_higher_order_ops Revert "[invoke_subgraph] make collect_meta_analysis fake prop cachable (#156347)" 2025-06-23 18:19:29 +00:00
_inductor [cpp wrapper] add AOTI shim for collective ops (#154492) 2025-06-25 01:20:05 +00:00
_lazy
_library Custom Op handle 1-element tuples (#155447) 2025-06-11 03:43:40 +00:00
_logging Add logging for async compile worker statistics (#155820) 2025-06-19 23:10:15 +00:00
_numpy fix numpy compatibility for 2d small list indices (#154806) 2025-06-04 01:58:52 +00:00
_prims Fix non-bitwise type annotations for Tensor operators (see #145838) (#146845) 2025-06-24 15:41:34 +00:00
_prims_common Improve torch.ops typing (#154555) 2025-06-22 15:52:27 +00:00
_refs remove gso from vector_norm (#156530) 2025-06-21 08:42:36 +00:00
_strobelight
_subclasses Revert "[invoke_subgraph] make collect_meta_analysis fake prop cachable (#156347)" 2025-06-23 18:19:29 +00:00
_vendor
accelerator Revert "Add unified memory APIs for torch.accelerator (#152932)" 2025-06-25 00:11:35 +00:00
amp
ao Improve torch.ops typing (#154555) 2025-06-22 15:52:27 +00:00
autograd [EZ/Profiler] Change 'b' to 'B' in FunctionEvent Frontend (#156250) 2025-06-23 23:25:04 +00:00
backends [coreml-backend-tool] fix pytorch-backended issue on new coremltools (#155543) 2025-06-11 20:52:26 +00:00
compiler Implement guard collectives (optimized version) (#156562) 2025-06-24 04:59:49 +00:00
contrib
cpu
csrc [cpp wrapper] add AOTI shim for collective ops (#154492) 2025-06-25 01:20:05 +00:00
cuda Revert "Add unified memory APIs for torch.accelerator (#152932)" 2025-06-25 00:11:35 +00:00
distributed [dcp_poc] Introduce a new simple rank local checkpointer (#156142) 2025-06-25 01:19:40 +00:00
distributions Fix non-bitwise type annotations for Tensor operators (see #145838) (#146845) 2025-06-24 15:41:34 +00:00
export Document each of the private member variables on ExportedProgram (#156704) 2025-06-24 19:56:40 +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 [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
fx Revert "[aotd] Support mutations of the same input in fw and bw (#155354)" 2025-06-24 04:42:14 +00:00
jit [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
legacy
lib [2/N] Fix cppcoreguidelines-init-variables suppression (#146237) 2025-06-19 23:26:42 +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 [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
monitor
mps [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
mtia [BE][PYFMT] migrate PYFMT for torch/[e-n]*/ to ruff format (#144553) 2025-06-17 08:18:47 +00:00
multiprocessing Fix MacOS MP hang in Python-3.12+ (#155698) 2025-06-24 12:13:35 +00:00
nativert [nativert] Move PrimKernelRegistry to PyTorch core (#156506) 2025-06-24 21:42:41 +00:00
nested mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
nn fix hack to check if register_buffer has been overridden (#155963) 2025-06-18 01:50:30 +00:00
onnx [ONNX] Preserve all legacy exporter params in fallback (#156659) 2025-06-24 05:28:55 +00:00
optim Fix non-bitwise type annotations for Tensor operators (see #145838) (#146845) 2025-06-24 15:41:34 +00:00
package [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
profiler [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
quantization [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
signal [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
sparse [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
special [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
standalone/macros Move c10/macros/Export.h to torch/standalone (#154850) 2025-06-03 06:18:59 +00:00
testing [MPS] Revert cumsum/cumprod to MPSGraph implementation (#156708) 2025-06-24 18:12:18 +00:00
utils Revert "Simplify nvtx3 CMake handling, always use nvtx3 (#153784)" 2025-06-24 20:02:07 +00:00
xpu [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
__config__.py
__future__.py
__init__.py Revert "Simplify nvtx3 CMake handling, always use nvtx3 (#153784)" 2025-06-24 20:02:07 +00:00
_appdirs.py
_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
_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 Improve documentation for torch.lobpcg (#156139) 2025-06-25 00:39:34 +00:00
_lowrank.py
_meta_registrations.py [Reland] [Intel GPU] Make SDPA output has the same stride as Query. (#154340) 2025-06-24 06:09:59 +00:00
_namedtensor_internals.py
_ops.py Improve torch.ops typing (#154555) 2025-06-22 15:52:27 +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 [docs] Add docstring indicating UB for converting inf to int (#154781) 2025-06-10 14:04:50 +00:00
_tensor_str.py fix tensor print behavior for MAIA (#155609) 2025-06-14 01:04:12 +00:00
_tensor.py Fix non-bitwise type annotations for Tensor operators (see #145838) (#146845) 2025-06-24 15:41:34 +00:00
_thread_safe_fork.py
_torch_docs.py Optimize dim description in torch.max (#156153) 2025-06-24 20:50:40 +00:00
_utils_internal.py Migrate from lru_cache to cache (#155613) 2025-06-11 19:44:18 +00:00
_utils.py Disable pinning check when loading sparse tensors (#154638) 2025-06-18 14:33:36 +00:00
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
CMakeLists.txt Revert "Simplify nvtx3 CMake handling, always use nvtx3 (#153784)" 2025-06-24 20:02:07 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py
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
library.py
overrides.py Revert "Upgrade to DLPack 1.0. (#145000)" 2025-06-20 15:32:47 +00:00
py.typed
quasirandom.py
random.py
return_types.py
script.h
serialization.py mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
storage.py mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
torch_version.py
types.py
version.py.tpl