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 37 Packages Projects Releases Wiki Activity
227cb268bc
pytorch/torch
History
Xiang Gao 227cb268bc [Reland] Embedding thrust->cub migration (#63806) 
Summary:
Fixes https://github.com/pytorch/pytorch/issues/63427

Pull Request resolved: https://github.com/pytorch/pytorch/pull/63806

Reviewed By: bdhirsh

Differential Revision: D30498255

Pulled By: ngimel

fbshipit-source-id: 78b7085a92a168cf0163f53dcb712bac922f5235
2021-08-24 09:30:32 -07:00
..
_C [nnc] Re-enable CPU fusion" (#63665) 2021-08-23 12:42:42 -07:00
ao [pruner] generalize bias hook for conv2d (#62430) 2021-08-05 09:27:17 -07:00
autograd Allow implementing either backward or vjp for Function (#63434) 2021-08-23 07:07:11 -07:00
backends NNAPI: Support const values in binary ops 2021-08-20 21:10:26 -07:00
contrib Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
cpu Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
csrc [JIT] Add variadic stack op (#63578) 2021-08-24 08:20:54 -07:00
cuda rebase for autocast updates to include device_type and dtype flags (#61002) 2021-08-10 20:03:12 -07:00
distributed Separating quantization test from distributed_test (#63058) 2021-08-24 01:44:55 -07:00
distributions Poisson zero rate (#61511) 2021-08-19 08:30:28 -07:00
fft Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
for_onnx
futures Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
fx [fx2trt] Add acc op and converter for torch.pow (#63795) 2021-08-23 18:18:31 -07:00
jit Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00
legacy
lib Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
linalg Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
multiprocessing
nn ENH Adds no_batch_dim for NLLLoss (#62651) 2021-08-24 08:27:27 -07:00
onnx [ONNX] Suppport torch.dot and torch.nn.utils.spectral_norm (#62596) (#62765) 2021-08-20 12:46:56 -07:00
optim To fix the chainability at epoch zero for some schedulers (#63457) 2021-08-19 07:17:03 -07:00
package Revert D29639797: [package] error if we try to mock a module in 3.6 2021-07-09 19:31:04 -07:00
profiler [Profiler] Change FLOP/s to Total FLOPs (#62779) 2021-08-16 13:43:32 -07:00
quantization Updating the names of these functions (#63513) 2021-08-19 13:34:34 -07:00
sparse Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
special [special] use __all__ to hide internal imports (#63135) 2021-08-19 12:45:43 -07:00
testing [Reland] Embedding thrust->cub migration (#63806) 2021-08-24 09:30:32 -07:00
utils Adding DataLoader2 class as future replacement of DataLoader (#63742) 2021-08-23 18:09:07 -07:00
__config__.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
__future__.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
__init__.py Updates set_default_dtype documentation (#63233) 2021-08-17 10:41:03 -07:00
_appdirs.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_classes.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_deploy.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_jit_internal.py Show warning in eager mode for empty containers (#62978) 2021-08-12 16:11:27 -07:00
_linalg_utils.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_lobpcg.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_lowrank.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_namedtensor_internals.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_ops.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_python_dispatcher.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_six.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_sources.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_storage_docs.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_tensor_docs.py [docs] Remove input parameter from Tensor.flatten docs (#63180) 2021-08-13 12:10:16 -07:00
_tensor_str.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_tensor.py Add support for the ONNX Runtime Eager Mode backend (#58248) 2021-08-20 11:17:13 -07:00
_torch_docs.py Modernizes add and mul documentation (#63309) 2021-08-19 12:49:30 -07:00
_utils_internal.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_utils.py Add support for the ONNX Runtime Eager Mode backend (#58248) 2021-08-20 11:17:13 -07:00
_VF.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
_vmap_internals.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
abi-check.cpp
autocast_mode.py Update cuda amp to also check xla device (#63413) 2021-08-18 06:44:10 -07:00
CMakeLists.txt Fix warnings (#62930) 2021-08-11 14:07:10 -07:00
custom_class_detail.h
custom_class.h
deploy.h
extension.h
functional.py document why wrappers exist in torch.functional (#62847) 2021-08-18 11:51:21 -07:00
hub.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
library.h Add support for the ONNX Runtime Eager Mode backend (#58248) 2021-08-20 11:17:13 -07:00
overrides.py Add support for the ONNX Runtime Eager Mode backend (#58248) 2021-08-20 11:17:13 -07:00
py.typed
quasirandom.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
random.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
README.txt
script.h
serialization.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
storage.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
torch_version.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00
types.py Revert D30279364: [codemod][lint][fbcode/c*] Enable BLACK by default 2021-08-12 11:45:01 -07:00

README.txt 

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.