Zachary DeVito
Zachary DeVito
Current we have two overloads for things where 0-dim tensors can occur: Tensor + Tensor Tensor + Scalar Instead we should only have Tensor + Tensor. However, we still need...
@killeent @colesbury @gchanan This issue is just to track the progress on advanced indexing plans in ATen. To summarize our discussion, the initial idea is to add 'low-level' support for...
Currently when you include `ATen/Tensor.h` in a compilation unit, you also need to include `ATen/TensorMethods.h`. This is handled by including `ATen/ATen.h`, but since we do no checks about what clients...
Tensor objects do not have to be defined: ``` Tensor foo; // undefined tensor reference auto zeros = CPU(kFloat).zeros({3,4}); // will crash because foo is not defined. add_out(zeros,zeros,foo); ``` In...
Similar to the size language of the original aten, but implemented as C++ utility functions that match dimensions of tensors to variables and report and error if a variable is...
Track headers correctly. Build dependency for making libOpt.a on API/o.t
Example code: from functorch import vmap import torch def f(x): return torch.ones(4)[x] y = torch.arange(2) print(vmap(f)(y)) Output: RuntimeError Traceback (most recent call last) [](https://r6s5at9hp4a-496ff2e9c6d22116-0-colab.googleusercontent.com/outputframe.html?vrz=colab-20220422-060045-RC00_443622635#) in () 5 6 y =...
Stack from [ghstack](https://github.com/ezyang/ghstack): * **#161 [wip] some clean and code to package models** * #158 List names, demucs cleanup
Stack from [ghstack](https://github.com/ezyang/ghstack): * #161 [wip] some clean and code to package models * **#158 List names, demucs cleanup** * Separate listing the names of modules from actually loading one....
* check parallel ranks have consistent data * remove a potential race condition when saving checkpoints that lets sequences_consumed get a head of number of iterations * Add code to...