pytorch/torch
David Berard 9d54d3bec2 [NVFuser] undo v100 OOM skips (#90070)
Summary: I think these were just caused by parallel tests. After adjusting test settings to 1 thread, these stopped OOMing.

Test Plan:
```
$ buck2 test -j 1 mode/dev-nosan //caffe2/torch/csrc/jit/codegen/cuda:nvfuser
```
https://www.internalfb.com/intern/testinfra/testrun/6473924590389963

Differential Revision: D41643827

Pull Request resolved: https://github.com/pytorch/pytorch/pull/90070
Approved by: https://github.com/jjsjann123
2022-12-02 21:58:24 +00:00
..
_C Small fix for torch._C.Graph type hint (#89821) 2022-11-30 21:48:09 +00:00
_C_flatbuffer
_decomp [decomp] Fix native_batch_norm_backward dtype of dweight and dbias (#89740) 2022-11-29 03:15:20 +00:00
_dispatch
_dynamo [Dynamo] tensor.type() should return tensor types with CPU and GPU variants (#90021) 2022-12-02 18:57:43 +00:00
_functorch Revert "Move functorch/_src to torch/_functorch (#88756)" 2022-11-29 17:17:11 +00:00
_inductor [Reland] dont clone args (#89766) 2022-12-02 17:20:40 +00:00
_lazy
_prims Make fake tensors preserve dense strides in type conversion (#89803) 2022-11-30 01:28:51 +00:00
_prims_common [primTorch] Enable regex error testing for some refs (#87765) 2022-11-23 23:36:27 +00:00
_refs [primTorch] Enable regex error testing for some refs (#87765) 2022-11-23 23:36:27 +00:00
_subclasses Make fake tensors preserve dense strides in type conversion (#89803) 2022-11-30 01:28:51 +00:00
amp
ao Add hierarchical module names to torchFX graph.node #87659 (#87742) 2022-12-02 05:58:06 +00:00
autograd Fix gradcheck for CSR and CSC inputs. (#89786) 2022-12-02 12:35:20 +00:00
backends
contrib
cpu
csrc [NVFuser] undo v100 OOM skips (#90070) 2022-12-02 21:58:24 +00:00
cuda Stream actually overrides __new__ so we need to patch it as well (#89592) 2022-11-29 21:43:23 +00:00
distributed Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
distributions Implement gamma cdf (#89955) 2022-12-01 00:12:53 +00:00
fft
futures
fx Reland "Dynamo, FX, Inductor Progress Bars (#88384)" … (#90055) 2022-12-02 13:28:00 +00:00
jit
legacy
lib
linalg update docstring for torch.linalg.lstsq (#89383) 2022-11-25 21:31:53 +00:00
masked
monitor
multiprocessing
nested
nn Revert "[follow-up] Python Attr Serialization (#88913)" 2022-12-02 20:14:11 +00:00
onnx Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
optim Implement post and pre hooks for optimizer (#89176) 2022-12-02 07:03:45 +00:00
package
profiler [Profiler] Memory profiler part 13: Add sizes to timeline. (#89356) 2022-12-02 03:55:22 +00:00
quantization Add hierarchical module names to torchFX graph.node #87659 (#87742) 2022-12-02 05:58:06 +00:00
signal Implement old windows in Python (#87082) 2022-11-25 11:09:28 +00:00
sparse
special
testing Revert "remove torch.equal usages (#89527)" 2022-12-02 21:36:13 +00:00
utils Issue 68576 prefetch factor docstring changes (#89874) 2022-11-30 23:42:56 +00:00
__config__.py
__future__.py
__init__.py Add torch.compile implementation (#89607) 2022-12-01 20:17:52 +00:00
_appdirs.py
_classes.py
_deploy.py
_jit_internal.py
_linalg_utils.py
_lobpcg.py
_lowrank.py
_meta_registrations.py Fix meta registration for aten._cdist_forward (#90042) 2022-12-02 21:13:52 +00:00
_namedtensor_internals.py
_ops.py
_python_dispatcher.py
_six.py
_sources.py
_storage_docs.py
_tensor.py Add type hint to torch.norm and Tensor.norm (#89728) 2022-11-29 02:09:51 +00:00
_tensor_docs.py Update to_sparse docs regarding the layout and blocksize kw arguments. (#89912) 2022-12-02 12:23:15 +00:00
_tensor_str.py
_torch_docs.py Update to_sparse docs regarding the layout and blocksize kw arguments. (#89912) 2022-12-02 12:23:15 +00:00
_utils.py Revert "[follow-up] Python Attr Serialization (#88913)" 2022-12-02 20:14:11 +00:00
_utils_internal.py
_VF.py
_vmap_internals.py
_weights_only_unpickler.py
abi-check.cpp
CMakeLists.txt
custom_class.h
custom_class_detail.h
extension.h
functional.py Add type hint to torch.norm and Tensor.norm (#89728) 2022-11-29 02:09:51 +00:00
hub.py
library.h
library.py
overrides.py Add torch.compile implementation (#89607) 2022-12-01 20:17:52 +00:00
py.typed
quasirandom.py
random.py
README.txt
return_types.py
script.h
serialization.py
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.