pytorch/torch
2024-08-02 04:18:57 +00:00
..
_awaits
_C Populate submodules of torch._C to sys.modules recursively (#132216) 2024-08-01 12:04:59 +00:00
_C_flatbuffer
_custom_op Tighten torch.library.infer_schema input types (#130705) 2024-07-29 16:01:19 +00:00
_decomp Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
_dispatch [BE][Easy][15/19] enforce style for empty lines in import segments in torch/_d*/ (#129767) 2024-07-31 21:18:11 +00:00
_dynamo [Dynamo] Support abc.MutableMapping.get (#132363) 2024-08-02 04:17:35 +00:00
_export Set weights_only=False in export deserialize_torch_artifact (#132348) 2024-08-01 23:25:07 +00:00
_functorch Revert "Ban decorator usage of dynamo_timed (#132328)" 2024-08-01 20:20:28 +00:00
_higher_order_ops [Easy] Fix argument name collision in HigherOrderOperator dispatched functions (#132377) 2024-08-01 17:13:37 +00:00
_inductor [AOTI] Fix number type for AOTI (#132180) 2024-08-01 22:43:28 +00:00
_lazy Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
_library [torchbind] don't warning for certain skippable methods. (#132306) 2024-08-01 22:40:42 +00:00
_logging Dump files that look like FX graphs to structured log (#132100) 2024-07-31 18:45:28 +00:00
_numpy [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_prims [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_prims_common [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_refs [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_strobelight [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_subclasses fastpath FunctionalTensor sizes() (#132084) 2024-08-01 21:09:22 +00:00
_vendor
amp
ao Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
autograd Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
backends Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
compiler
contrib
cpu
csrc Fix some issues detected by static analysis tools (#131989) 2024-08-02 04:18:57 +00:00
cuda [CUDA] is_bf16_supported() should not crash if there are no GPUs (#132313) 2024-08-02 02:50:43 +00:00
distributed Only make wait_tensor as a side_effect op (#132341) 2024-08-02 01:24:40 +00:00
distributions [BE][Easy][18/19] enforce style for empty lines in import segments in torch/d*/ (#129770) 2024-08-01 04:22:50 +00:00
export Revert "[export] Add print_readable to unflattener (#128617)" 2024-08-01 23:47:29 +00:00
fft
func
futures
fx add src map to data-dependent errors (#132393) 2024-08-02 00:31:12 +00:00
jit Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
legacy
lib
linalg
masked Add functions from torch.masked._ops to __all__ for torch.masked (#131288) 2024-08-01 05:45:08 +00:00
monitor
mps
mtia [Land Internally] MTIA equivalent of torch.cuda.memory_stats (#132007) 2024-07-29 20:47:18 +00:00
multiprocessing Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
nested Grouped Query Attention (#128898) 2024-07-31 22:58:51 +00:00
nn Always use high precision for SDPA math backend (#128922) 2024-08-01 18:55:48 +00:00
onnx [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
optim [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
package [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
profiler [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
quantization [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
signal Revert "[BE] typing for decorators - signal/windows/windows (#131582)" 2024-07-28 03:29:31 +00:00
sparse [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
special
testing Correct sample creation of torch.histogram in UT op_db to align PyTorch defined operator semantics (#131630) 2024-08-02 01:51:09 +00:00
utils Skip frame if torch dispatch mode enabled (#131828) 2024-08-01 19:06:20 +00:00
xpu [BE][Easy][19/19] enforce style for empty lines in import segments in torch/[o-z]*/ (#129771) 2024-08-01 17:07:14 +00:00
__config__.py
__future__.py
__init__.py Populate submodules of torch._C to sys.modules recursively (#132216) 2024-08-01 12:04:59 +00:00
_appdirs.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_classes.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
_compile.py
_custom_ops.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_deploy.py
_guards.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
_jit_internal.py [Easy] reorder functions in torch._jit_internal (#130531) 2024-07-31 02:12:29 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py Add lowering for updated _scaled_mm (fixing submodules) (#130422) 2024-07-30 23:48:48 +00:00
_namedtensor_internals.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
_ops.py [Easy] Fix argument name collision in HigherOrderOperator dispatched functions (#132377) 2024-08-01 17:13:37 +00:00
_python_dispatcher.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
_size_docs.py
_sources.py
_storage_docs.py
_streambase.py
_tensor_docs.py [MTIA] Support module.mtia() (#131499) 2024-07-25 04:23:48 +00:00
_tensor_str.py fix tensor print behavior for XPU (#130523) 2024-07-17 02:03:32 +00:00
_tensor.py [easy] Small rendering fix in Tensor.module_load doc (#130489) 2024-07-12 22:12:53 +00:00
_torch_docs.py docs(multinomial): Add reference to Multinomial class (#131904) 2024-07-30 18:05:07 +00:00
_utils_internal.py Write trace_structured events to scuba (#130955) 2024-07-19 06:02:47 +00:00
_utils.py
_VF.py
_vmap_internals.py
_weights_only_unpickler.py Blocklist certain modules for weights_only load (#131259) 2024-07-22 18:23:21 +00:00
abi-check.cpp
CMakeLists.txt Revert "Add wrappers for synchronous GPUDirect Storage APIs (#130633)" 2024-07-26 18:08:20 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py
hub.py
library.h [3/N] Fix Wunused-parameter warnings (#131271) 2024-07-20 23:31:03 +00:00
library.py Fix public API tests (#131386) 2024-07-30 18:42:54 +00:00
overrides.py Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
py.typed
quasirandom.py
random.py
README.txt
return_types.py
script.h
serialization.py Updating Types in torch/_dynamo/utils.py (#131001) 2024-07-23 18:25:52 +00:00
storage.py Fix public API tests (#131386) 2024-07-30 18:42:54 +00:00
torch_version.py Add mypy typing to torch_version.py (#131447) 2024-07-23 17:31:07 +00:00
types.py FakeTensor cache SymInt support (#127596) 2024-07-21 19:26:38 +00:00
version.py.tpl

Note [TH abstraction violation]
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

TH/THC provide some hpp headers, which are proper C++ headers rather than
C headers.  These headers serve double duty as *internal implementation
detail* headers, whose contents should largely not be used by external
clients.

Ideally, we would not install these headers at all; instead, you should
use public functions (in headers like `THTensor.h`, NOT `THTensor.hpp`)
to manipulate these structs.  However, there are a few places
in torch/csrc where we violate this abstraction.  They are marked with
a pointer to this note.  Each of those sites will have to be refactored
when we refactor the guts of THTensor and related structures.