pytorch/torch/_subclasses
Elias Ellison 688b971876 Extend fake tensor tests to cuda, add support for index put (#82281)
Testing CUDA exposes some failures, such as `index_put` with CUDA self tensor and cpu value tensors
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82281
Approved by: https://github.com/ezyang
2022-07-28 16:07:15 +00:00
..
__init__.py Apply ufmt to torch internal (#81643) 2022-07-22 02:19:50 +00:00
fake_tensor.py Extend fake tensor tests to cuda, add support for index put (#82281) 2022-07-28 16:07:15 +00:00
meta_utils.py Enable complex for meta tensors (#79975) 2022-07-27 22:19:14 +00:00