pytorch/torch/testing
Jeff Daily d5c4a80cfd Allow ROCm CI to use non-default stream. (#48424)
Summary:
Revert https://github.com/pytorch/pytorch/issues/26394. Fixes https://github.com/pytorch/pytorch/issues/27356.  Not all MIOpen handles were setting their stream to the current stream prior to running the op.

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

Reviewed By: H-Huang

Differential Revision: D25420384

Pulled By: mruberry

fbshipit-source-id: 051683ba9e3d264b71162bd344031a0c58bf6a41
2020-12-10 09:55:11 -08:00
..
_internal Allow ROCm CI to use non-default stream. (#48424) 2020-12-10 09:55:11 -08:00
__init__.py fix assert_allclose doesnt check shape (#47580) 2020-11-10 15:03:25 -08:00
check_kernel_launches.py [pytorch] [PR] Rename cuda kernel checks to C10 (#48615) 2020-12-01 18:19:07 -08:00