pytorch/torch/testing
Xiang Gao 227cb268bc [Reland] Embedding thrust->cub migration (#63806)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/63427

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

Reviewed By: bdhirsh

Differential Revision: D30498255

Pulled By: ngimel

fbshipit-source-id: 78b7085a92a168cf0163f53dcb712bac922f5235
2021-08-24 09:30:32 -07:00
..
_internal [Reland] Embedding thrust->cub migration (#63806) 2021-08-24 09:30:32 -07:00
__init__.py remove randn? from torch.testing namespace (#61840) 2021-07-23 11:51:03 -07:00
_asserts.py Improve error messages of torch.testing.assert_close for sparse inputs (#61583) 2021-07-30 11:23:26 -07:00
_check_kernel_launches.py Paren-matching kernel launch check without external deps (#60778) 2021-06-28 10:18:04 -07:00
_core.py Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00
_deprecated.py Updates internal assert_allclose callsites in favor of assert_close (#61841) 2021-08-19 12:50:41 -07:00