OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
ab8874bd26
pytorch/torch
History
Hao Zhang(张浩) ab8874bd26 Suppress warning when using native arch for jit loading cuda extensions. (#156923) 
Previeusly, if users want to let pytorch determine the cuda arch when jit loading cuda extensions, they should left environment variable `TORCH_CUDA_ARCH_LIST` empty, but which will raise an warning. This commit add an option to set `TORCH_CUDA_ARCH_LIST=native`, to tell pytorch users want to use native cuda arch intentionally.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/156923
Approved by: https://github.com/ezyang
2025-07-09 02:51:20 +00:00
..
_awaits
_C Work: block_current_stream API (#156883) 2025-07-08 23:55:46 +00:00
_C_flatbuffer
_custom_op
_decomp Revert "Fix full_like decomposition to preserve strides (#144765)" 2025-07-02 13:56:03 +00:00
_dispatch Improve torch.ops typing (#154555) 2025-06-22 15:52:27 +00:00
_dynamo [dynamo] removing string literals for weblink generation (#157820) 2025-07-08 23:08:06 +00:00
_export [export] Fix lift constants bug (#157719) 2025-07-08 20:33:53 +00:00
_functorch [inductor] Quiesce Triton compile worker pool after each dynamo compile (#156187) 2025-07-08 22:53:13 +00:00
_higher_order_ops [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
_inductor [Inductor] Support precomputed size args in the FX backend. (#157758) 2025-07-08 23:22:17 +00:00
_lazy
_library [BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551) 2025-06-25 06:16:06 +00:00
_logging [BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551) 2025-06-25 06:16:06 +00:00
_numpy [BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551) 2025-06-25 06:16:06 +00:00
_prims [remove untyped defs] batch 1 (#157011) 2025-06-30 23:54:40 +00:00
_prims_common python definitely_contiguous-> is_contiguous_or_false (#156515) 2025-06-30 17:31:51 +00:00
_refs _broadcast_shapes gso generalizations (#157008) 2025-07-04 05:56:42 +00:00
_strobelight [BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551) 2025-06-25 06:16:06 +00:00
_subclasses [fake tensor] fix issue of no attribute tags (#156689) 2025-07-03 01:16:01 +00:00
_vendor
accelerator Revert "Add unified memory APIs for torch.accelerator (#152932)" 2025-06-25 00:11:35 +00:00
amp [BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554) 2025-07-03 18:56:07 +00:00
ao remove allow-untyped-defs from torch/ao/nn/quantized/modules/rnn.py (#157234) 2025-07-08 00:11:52 +00:00
autograd Add legacy note to autograd.profiler doc. (#157459) 2025-07-08 20:33:23 +00:00
backends remove allow-untyped-defs from torch/backends/cusparselt/__init__.py (#157232) 2025-07-08 00:11:52 +00:00
compiler Automatically load and save dynamo entries via caching_precompile (#155913) 2025-07-07 23:57:17 +00:00
contrib
cpu
csrc Work: block_current_stream API (#156883) 2025-07-08 23:55:46 +00:00
cuda [BE][PYFMT] migrate PYFMT for torch/[a-c]*/ to ruff format (#144554) 2025-07-03 18:56:07 +00:00
distributed Revert "[DTensor][FSDP2] necessary changes to FSDP and TP to unblock EP (#157216)" 2025-07-08 20:48:51 +00:00
distributions Fix non-bitwise type annotations for Tensor operators (see #145838) (#146845) 2025-06-24 15:41:34 +00:00
export Remove is_jit_trace option (#157387) 2025-07-03 09:20:27 +00:00
fft
func
futures Simplify the base classes of _PyFutureMeta (#157757) 2025-07-08 15:39:56 +00:00
fx Add flag to fx.passes.split_module to normalize input names (#157733) 2025-07-08 13:47:24 +00:00
headeronly Reapply D77381084 / #156964: Rename torch::standalone to headeronly (#157251) 2025-06-30 23:25:30 +00:00
jit [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
legacy
lib [2/N] Fix cppcoreguidelines-init-variables suppression (#146237) 2025-06-19 23:26:42 +00:00
linalg
masked [Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735) 2025-07-08 18:10:33 +00:00
monitor
mps [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
mtia
multiprocessing [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
nativert [nativert] Move Executor to PyTorch core (#157514) 2025-07-03 23:31:54 +00:00
nested [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
nn XCCL changes for DDP (#155497) 2025-07-03 05:18:08 +00:00
onnx [BE][12/16] fix typos in torch/ (#156602) 2025-07-02 22:55:29 +00:00
optim [Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735) 2025-07-08 18:10:33 +00:00
package [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
profiler [profiler] add more CUDA API for kernel launcher (#156016) 2025-07-03 15:26:42 +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 [Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735) 2025-07-08 18:10:33 +00:00
special [BE][6/16] fix typos in torch/ (#156316) 2025-06-23 02:57:34 +00:00
testing Add stack trace of exception to MultiProcContinousTest (#157589) 2025-07-08 17:54:35 +00:00
utils Suppress warning when using native arch for jit loading cuda extensions. (#156923) 2025-07-09 02:51:20 +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 correctly import torch.version (#157584) 2025-07-07 21:43:35 +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
_custom_ops.py
_deploy.py
_environment.py
_guards.py [dynamo][fsdp] Consistent behavior of int attributes (#157262) 2025-07-08 22:11:33 +00:00
_jit_internal.py
_linalg_utils.py
_lobpcg.py Improve documentation for torch.lobpcg (#156139) 2025-06-25 00:39:34 +00:00
_lowrank.py
_meta_registrations.py python definitely_contiguous-> is_contiguous_or_false (#156515) 2025-06-30 17:31:51 +00:00
_namedtensor_internals.py
_ops.py Improve torch.ops typing (#154555) 2025-06-22 15:52:27 +00:00
_python_dispatcher.py
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py
_tensor_str.py
_tensor.py Upgrade to DLPack 1.0. (#145000) 2025-06-30 16:58:06 +00:00
_thread_safe_fork.py
_torch_docs.py Documentation update torch.clone #156644 (#157007) 2025-06-27 21:10:09 +00:00
_utils_internal.py Inductor logging + analysis of torch.profile (#149697) 2025-07-07 22:13:34 +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 [1/N] Don't use CUDA.cmake module (#157188) 2025-07-08 03:05:35 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py
header_only_apis.txt Reapply D77381084 / #156964: Rename torch::standalone to headeronly (#157251) 2025-06-30 23:25:30 +00:00
hub.py
library.h
library.py
overrides.py Revert "Fused RMSNorm implementation (#153666)" 2025-07-01 18:46:45 +00:00
py.typed
quasirandom.py
random.py
return_types.py
script.h
serialization.py [Pyrefly][Refactor] Replace dict() calls with literal dict syntax for improved readability (#157735) 2025-07-08 18:10:33 +00:00
storage.py
torch_version.py
types.py
version.py.tpl