pytorch/torch
Jerry Zhang f9446cb15a [quant][refactor] Remove register api and rename get_*_mapping to get_default_*_mapping (#46337)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/46337

We plan to pass around the mappings instead of using global registration api to keep
the mappings local to the transformations user is performing

Test Plan: Imported from OSS

Reviewed By: vkuzo

Differential Revision: D24317436

fbshipit-source-id: 81569b88f05eeeaa9595447e482a12827aeb961f
2020-10-20 15:53:47 -07:00
..
_C [Metal] Add the Python binding for optimize_for_mobile (#46456) 2020-10-17 10:26:25 -07:00
autograd Allow Tensor-likes in torch.autograd.gradcheck (#45732) 2020-10-09 11:51:27 -07:00
backends Enable typechecking for torch.testing._internal.common_quantized.* (#44805) 2020-09-17 14:24:32 -07:00
contrib Fix exception chaining in torch/ (#43836) 2020-08-31 20:26:23 -07:00
csrc Revert D24395956: [pytorch][PR] Replace flatten tensors with flatten loops. 2020-10-20 15:42:23 -07:00
cuda Use MTA for amp grad unscaling, enforce op math type in MTA functors, and allow op lambdas (#44778) 2020-10-01 07:51:16 -07:00
distributed Add warning on ProcessGroup and ProcessGroup::Work APIs (#46220) 2020-10-14 16:27:37 -07:00
distributions Expose script_if_tracing as public API (#46494) 2020-10-17 17:31:57 -07:00
fft torch.fft: Two dimensional FFT functions (#45164) 2020-10-17 16:23:06 -07:00
for_onnx
futures fix #45552 - adding add_done_callback(fn) to torch.futures.Future (#45675) 2020-10-13 07:47:36 -07:00
fx Revert D24269034: [fx] Refactor Tracer so that find_module and root args creation could be overridden by implementations 2020-10-19 17:29:18 -07:00
jit Replace list(map(...)) constructs by list comprehensions (#46461) 2020-10-19 18:42:49 -07:00
legacy
lib [NCCL] Modularize ncclCommWatchdog (#46051) 2020-10-16 11:06:40 -07:00
linalg Add torch.linalg.norm (#42749) 2020-08-28 18:28:33 -07:00
multiprocessing Add exception classification to torch.multiprocessing.spawn (#45174) 2020-10-09 12:59:41 -07:00
nn Replace list(map(...)) constructs by list comprehensions (#46461) 2020-10-19 18:42:49 -07:00
onnx [ONNX] Update squeeze test for opset 9 (#45369) 2020-10-14 12:53:13 -07:00
optim Replace list(map(...)) constructs by list comprehensions (#46461) 2020-10-19 18:42:49 -07:00
package [packaging] simpler dependency plotting (#45686) 2020-10-06 23:40:00 -07:00
quantization [quant][refactor] Remove register api and rename get_*_mapping to get_default_*_mapping (#46337) 2020-10-20 15:53:47 -07:00
sparse Fix type annotations for torch.sparse, enable in CI (#43108) 2020-08-17 14:40:11 -07:00
testing [quant][refactor] Remove register api and rename get_*_mapping to get_default_*_mapping (#46337) 2020-10-20 15:53:47 -07:00
utils Implement _broadcast_to_and_flatten(pytree, spec) (#46288) 2020-10-20 07:52:14 -07:00
__config__.py
__future__.py
__init__.py [quant] creating quint4x2 dtype for quantized tensors (#44678) 2020-10-01 23:53:34 -07:00
_appdirs.py
_classes.py [BE] Use f-string in various Python functions (#44161) 2020-09-04 07:38:25 -07:00
_jit_internal.py [JIT] Enable @unused syntax for ignoring properties (#45261) 2020-09-29 10:24:25 -07:00
_linalg_utils.py
_lobpcg.py Backward support for generalized eigenvalue solver with LOBPCG in forward [only k-rank SYMEIG case] (#43002) 2020-09-28 07:22:35 -07:00
_lowrank.py Add __torch_function__ for methods (#37091) 2020-08-05 20:44:13 -07:00
_namedtensor_internals.py
_ops.py
_six.py Delete raise_from from torch._six (#43981) 2020-09-01 15:46:18 -07:00
_storage_docs.py
_tensor_docs.py Implement ravel (#46098) 2020-10-12 16:00:44 -07:00
_tensor_str.py [quant] Create PerRowQuantizer for floating point scale and zero_point (#42612) 2020-08-13 11:20:53 -07:00
_torch_docs.py Added support for complex torch.pinverse (#45819) 2020-10-15 12:28:22 -07:00
_utils_internal.py Remove py2 compatible future imports (#44735) 2020-09-16 12:55:57 -07:00
_utils.py [caffe2][torch] correctly re-raise Manifold StorageException 2020-08-28 11:41:10 -07:00
_VF.py Address JIT/Mypy issue with torch._VF (#43454) 2020-08-25 09:23:54 -07:00
_vmap_internals.py Allow vmap to accept nested python data structures as inputs (#46289) 2020-10-20 07:52:17 -07:00
abi-check.cpp
CMakeLists.txt make a way to disable callgrind (#46116) 2020-10-13 16:18:04 -07:00
custom_class_detail.h
custom_class.h Adding a version serialization type to ConvPackedParam (#43086) 2020-08-28 15:41:30 -07:00
extension.h
functional.py Fix typing errors in the torch.distributions module (#45689) 2020-10-12 10:29:45 -07:00
hub.py Add a torch.hub.load_local() function that can load models from any local directory with a hubconf.py (#44204) 2020-09-21 14:17:21 -07:00
library.h Rationalize inlining of kernels into the unboxing wrapper (#42845) 2020-10-15 04:02:51 -07:00
overrides.py Replace list(map(...)) constructs by list comprehensions (#46461) 2020-10-19 18:42:49 -07:00
py.typed
quasirandom.py Type check quasirandom (#45434) 2020-09-28 16:49:38 -07:00
random.py Fix manual seed to unpack unsigned long (#42206) 2020-08-11 18:05:34 -07:00
README.txt
script.h
serialization.py Use storage.cpu() for moving storage to CPU in serialization. (#46028) 2020-10-13 12:51:10 -07:00
storage.py [BE] Use f-string in various Python functions (#44161) 2020-09-04 07:38:25 -07:00
tensor.py Allow consumer ops to sync on GraphRoot's gradient (#45787) 2020-10-07 08:53:53 -07:00
types.py Enable torch.tensor typechecks (#45077) 2020-09-24 08:22:06 -07:00

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.