pytorch/torch
Mike Ruberry dc87cf5fe1 Fixes mem_get_info when querying on a device other than the current device (#69640)
Summary:
Also fixes the documentation failing to appear and adds a test to validate that op works with multiple devices properly.

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

Reviewed By: ngimel

Differential Revision: D32965391

Pulled By: mruberry

fbshipit-source-id: 4fe502809b353464da8edf62d92ca9863804f08e
2021-12-08 23:04:30 -08:00
..
_C [JIT] optimize_for_inference on methods other than forward (#69367) 2021-12-07 12:36:47 -08:00
_masked Strided masked var. (#68738) 2021-12-01 19:19:37 -08:00
ao Getting NS to work with Ferraris (#68908) 2021-12-08 12:26:00 -08:00
autograd Make make_dual also return namedtuple when level less than zero (#68628) 2021-12-08 17:54:40 -08:00
backends [Linalg] Add a runtime switch to let pytorch prefer a backend impl in linalg functions on GPU (#67980) 2021-12-03 19:06:30 -08:00
contrib
cpu Add fp16/fp32 autocasting to JIT/TorchScript (#63939) 2021-10-27 12:11:36 -07:00
csrc Fixes mem_get_info when querying on a device other than the current device (#69640) 2021-12-08 23:04:30 -08:00
cuda Add nvidia-smi memory and utilization as native Python API (#69104) 2021-12-08 10:33:23 -08:00
distributed Set non-default backend names to lower case (#69400) 2021-12-07 07:58:46 -08:00
distributions Deprecate torch.triangular_solve (#63570) 2021-12-02 13:24:55 -08:00
fft
for_onnx
futures
fx Fix sign op converter (#69580) 2021-12-07 19:04:51 -08:00
jit [JIT] optimize_for_inference on methods other than forward (#69367) 2021-12-07 12:36:47 -08:00
legacy
lib [NOOP][clangformat][codemod] Enable CLANGFORMAT for some folders in caffe2/* (#67746) 2021-11-03 12:23:14 -07:00
linalg Revert D32521980: Add linalg.lu_factor 2021-11-28 17:22:15 -08:00
multiprocessing
nn [quant][embdding qat] Re-land Add FX support for QAT EmbeddingBag (#69334) 2021-12-08 05:57:20 -08:00
onnx Added antialias flag to interpolate (CPU only, bilinear) (#65142) 2021-11-17 09:10:15 -08:00
optim Fix ChainedScheduler.get_last_lr() (#69112) 2021-12-02 13:44:12 -08:00
package
profiler [Reland] Python tracer. (#68325) 2021-11-15 23:32:49 -08:00
quantization
sparse Sparse CSR CUDA: Add torch.sparse.sampled_addmm (#68007) 2021-11-29 15:43:29 -08:00
special
testing Run test_embedding_bag_with_no_grad_tensors only for TensorPipe (#69626) 2021-12-08 18:29:38 -08:00
utils [DataLoader] Implementing communication processes for Map-style DataPipes (#68549) 2021-12-08 07:27:01 -08:00
__config__.py
__future__.py
__init__.py expose return_types in Python (#66614) 2021-12-06 09:05:29 -08:00
_appdirs.py
_classes.py
_deploy.py [deploy] fix TypedStorage serialization (#67499) 2021-10-28 22:33:04 -07:00
_jit_internal.py [package] fix torchscript classes in package (#68028) 2021-11-16 10:01:40 -08:00
_linalg_utils.py
_lobpcg.py torch.lobpcg.backward: do not save non-Variable types with ctx.save_for_backward. (#67994) 2021-11-08 10:02:09 -08:00
_lowrank.py
_namedtensor_internals.py
_ops.py
_python_dispatcher.py
_six.py
_sources.py Disallow annotations on instance attributes outside __init__ (#67051) 2021-10-25 16:20:47 -07:00
_storage_docs.py
_tensor_docs.py [numpy] Alias arctan2 to atan2 (#67010) 2021-11-16 09:41:09 -08:00
_tensor_str.py Add efficient zero tensors (#64837) 2021-12-08 10:37:39 -08:00
_tensor.py [quant] Remove warning for quantized Tensor in __dir__ (#69265) 2021-12-02 10:30:36 -08:00
_torch_docs.py Deprecate torch.triangular_solve (#63570) 2021-12-02 13:24:55 -08:00
_utils_internal.py
_utils.py
_VF.py
_vmap_internals.py More aggressively market functorch.vmap when torch.vmap gets called (#67347) 2021-11-12 16:10:16 -08:00
abi-check.cpp
autocast_mode.py Add fp16/fp32 autocasting to JIT/TorchScript (#63939) 2021-10-27 12:11:36 -07:00
CMakeLists.txt Revert D32010095: [pytorch][PR] Add ability for a mobile::Module to save as flatbuffer 2021-12-02 06:41:40 -08:00
custom_class_detail.h [NOOP][clangformat][codemod] Enable CLANGFORMAT for some folders in caffe2/* (#67746) 2021-11-03 12:23:14 -07:00
custom_class.h [NOOP][clangformat][codemod] Enable CLANGFORMAT (#67854) 2021-11-04 14:07:57 -07:00
deploy.h
extension.h
functional.py Revert D32521980: Add linalg.lu_factor 2021-11-28 17:22:15 -08:00
hub.py making import_module private and deprecating public method (#67990) 2021-11-09 07:27:57 -08:00
library.h [NOOP][clangformat][codemod] Enable CLANGFORMAT for some folders in caffe2/* (#67746) 2021-11-03 12:23:14 -07:00
overrides.py Add efficient zero tensors (#64837) 2021-12-08 10:37:39 -08:00
py.typed
quasirandom.py
random.py
README.txt
return_types.py expose return_types in Python (#66614) 2021-12-06 09:05:29 -08:00
script.h [NOOP][clangformat][codemod] Enable CLANGFORMAT for some folders in caffe2/* (#67746) 2021-11-03 12:23:14 -07:00
serialization.py Avoid dtype mismatch error in torch.save if storages are unallocated (#68787) 2021-11-24 09:51:29 -08:00
storage.py
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.