pytorch/torch
2023-05-04 04:28:45 +00:00
..
_awaits
_C Publicly exposing torch.backends.cpu.get_cpu_capability() (#100164) 2023-05-03 19:02:07 +00:00
_C_flatbuffer
_decomp [decomp] Bad accuracy for elu_backward (#100284) 2023-04-29 04:21:20 +00:00
_dispatch Renaming all_known_overloads to all_py_loaded_overloads and add comment (#97672) 2023-03-28 14:10:38 +00:00
_dynamo Reland basic dynamo support for traceable collectives (#100476) 2023-05-04 04:25:35 +00:00
_export [export] Migrate internal verifier to subclass export/verifier 2023-05-02 08:50:48 -07:00
_functorch Add AotAutogradFallbackTests to dynamic suite (#100454) 2023-05-04 04:28:45 +00:00
_higher_order_ops [Reland] Initial version of Dynamo capture for HigherOrderOperator (#100544) 2023-05-03 20:49:05 +00:00
_inductor Misc accuracy improvements on minifier (#100447) 2023-05-04 02:51:26 +00:00
_lazy
_logging [dynamo] Add ddp_graphs artifact (#100021) 2023-04-27 03:53:23 +00:00
_prims Add torch.utils._content_store (#99809) 2023-04-26 18:02:59 +00:00
_prims_common Track exact origin_node on best effort basis (#100110) 2023-04-28 04:15:27 +00:00
_refs [primTorch] add ref for polar (#100345) 2023-05-04 01:37:02 +00:00
_subclasses CustomOp simple abstract implementation registration (#99439) 2023-04-28 13:45:39 +00:00
amp refactor macro with AMP (#99285) 2023-04-19 01:00:00 +00:00
ao [PT2E][Quant] Refactor quantizer and qnnpack qantizer code to support dqlinear config (#99399) 2023-05-03 03:23:32 +00:00
autograd fix(docs): torch.autograd.graph.Node.register_hook can override grad_inputs, not grad_outputs (#100272) 2023-04-29 00:10:12 +00:00
backends Publicly exposing torch.backends.cpu.get_cpu_capability() (#100164) 2023-05-03 19:02:07 +00:00
contrib
cpu
csrc [dynamo] Minor fixes and clean-up in eval_frame.c (#100496) 2023-05-04 00:45:15 +00:00
cuda [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
distributed Reland basic dynamo support for traceable collectives (#100476) 2023-05-04 04:25:35 +00:00
distributions Remove in-place operations in NegativeBinomial (#96748) 2023-04-26 14:45:08 +00:00
fft
func
futures
fx Partition modules (#98628) 2023-05-03 23:31:56 +00:00
jit Revert "Added round_with_scale_factor arg to ATen (#97868)" 2023-04-28 20:47:00 +00:00
legacy
lib
linalg
masked [BE] Enable flake8-comprehension rule C417 (#97880) 2023-03-30 14:34:24 +00:00
monitor
mps
multiprocessing Reduce overhead in CUDAGraph Trees (#98529) 2023-04-07 05:46:08 +00:00
nested [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
nn Handle trailing masked column behavior for nested tensor (#100113) 2023-05-03 00:30:17 +00:00
onnx [ONNX] Update 'Functionalize' pass to support pre-decomp graph; Drop 'aten_graph' arg for 'DynamoExporter' (#99667) 2023-05-04 00:01:22 +00:00
optim Optimized EMA implementation (#94820) 2023-04-26 18:02:11 +00:00
package Convert logging f-strings to use % format, part five (#98765) 2023-04-11 13:17:59 +00:00
profiler [Profiler] Support HTML plot output for profiler export_memory_timeline API (#99751) 2023-04-22 04:21:58 +00:00
quantization
signal Fix flake8 lint errors reported by ruff - take 2 (#99798) 2023-04-23 23:09:51 +00:00
sparse
special
testing [primTorch] add ref for polar (#100345) 2023-05-04 01:37:02 +00:00
utils add checkpoint support for custom device (#99626) 2023-05-04 00:23:42 +00:00
__config__.py
__future__.py
__init__.py Publicly exposing torch.backends.cpu.get_cpu_capability() (#100164) 2023-05-03 19:02:07 +00:00
_appdirs.py
_classes.py
_custom_op.py Improved CustomOp API with schema inference (#100127) 2023-04-28 16:53:07 +00:00
_deploy.py
_guards.py Move tracked nn_modules from OutputGraph to TracingContext (#100457) 2023-05-03 02:00:11 +00:00
_jit_internal.py [JIT] Allow tuple and list generics (#98703) 2023-04-09 22:58:58 +00:00
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py Add AotAutogradFallbackTests to dynamic suite (#100454) 2023-05-04 04:28:45 +00:00
_namedtensor_internals.py
_ops.py Revert "Initial version of Dynamo capture for HigherOrderOperator (#99988)" 2023-05-03 14:02:40 +00:00
_python_dispatcher.py
_sources.py
_storage_docs.py
_tensor_docs.py Fix Tensor.uniform_ documentation to mention generator argument (#99510) 2023-04-19 19:23:12 +00:00
_tensor_str.py Fix FakeTensor printing (#99205) 2023-04-18 13:26:27 +00:00
_tensor.py Change 'w.r.t.' to 'wrt' in function docstrings to fix doc rendering (#100028) 2023-04-25 23:53:26 +00:00
_torch_docs.py Modify repeat_interleave docs to highlight potential overloading (#99650) 2023-05-01 17:53:03 +00:00
_utils_internal.py Log PT2 compile to Scuba (#98790) 2023-04-11 20:10:35 +00:00
_utils.py add get_device_index for custom device (#98804) 2023-04-12 23:58:31 +00:00
_VF.py
_vmap_internals.py [BE] Enable C419 rule for any all shortcircuiting (#99890) 2023-04-25 15:02:13 +00:00
_weights_only_unpickler.py
abi-check.cpp
CMakeLists.txt [ROCm] Remove usage of deprecated ROCm component header includes (#97620) 2023-03-28 19:28:38 +00:00
custom_class_detail.h
custom_class.h
extension.h
functional.py
hub.py torch.hub: add safe weights_only option to load_state_dict_from_url (#98479) 2023-04-11 12:44:25 +00:00
library.h
library.py torch.library.Library.impl: add missing param in docstring example (#98619) 2023-04-11 06:09:46 +00:00
overrides.py Persist torch.assert in aten graph (#100101) 2023-04-28 07:31:43 +00:00
py.typed
quasirandom.py
random.py add rng_state support for custom device (#98069) 2023-04-10 22:36:55 +00:00
README.txt
return_types.py
script.h
serialization.py Support non-ASCII characters in model file paths (#99453) 2023-04-26 01:15:49 +00:00
storage.py Fix loading data on different encoding (#94503) 2023-04-25 21:05:20 +00:00
torch_version.py
types.py

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.