diff options
| author | Iurii Zdebskyi <iuriiz@fb.com> | 2019-04-02 16:10:43 -0700 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-04-02 16:17:05 -0700 |
| commit | b832b99afb241a8b6ea9fc34698d1f3bfd451f00 (patch) | |
| tree | 0717b959c1718010605006193fb070557df6e1be /caffe2/python/pybind_state.cc | |
| parent | b77e3c2ca1790cfd4414d15f0a0952a630dc22c9 (diff) | |
| download | pytorch-b832b99afb241a8b6ea9fc34698d1f3bfd451f00.tar.gz pytorch-b832b99afb241a8b6ea9fc34698d1f3bfd451f00.tar.bz2 pytorch-b832b99afb241a8b6ea9fc34698d1f3bfd451f00.zip | |
Bool Tensor for CUDA (#18166)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18166
ghimport-source-id: a8e2ba2d966e49747a55701c4f6863c5e24d6f14
Stack from [ghstack](https://github.com/ezyang/ghstack):
* **#18166 Bool Tensor for CUDA**
* #18165 Resolved comments from Bool Tensor for CPU PR
------
This PR enables bool tensor creation and some basic operations for the CPU backend. This is a part of Bool Tensor feature implementation work. The whole plan looks like this:
1. Storage Implementation [Done]
2. Tensor Creation.
a) CPU [Done]
b) CUDA [This PR]
3. Tensor Conversions.
4. Tensor Indexing.
5. Tensor Operations.
6. Back compatibility related changes.
Change:
Enable bool tensor in CUDA with the following operations:
torch.zeros
torch.tensor
torch.ones
torch.rand/rand_like/randint/randint_like
torch.full
torch.full_like
torch.empty
torch.empty_like
Tested via unit tests and local scripts.
Differential Revision: D14605104
fbshipit-source-id: b7d7340a7d70edd03a109222d271e68becba762c
Diffstat (limited to 'caffe2/python/pybind_state.cc')
0 files changed, 0 insertions, 0 deletions