frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

enable unit tests (#25963)

Browse Source 
Summary:
These unit tests pass after landing all the warp size awareness patches.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25963

Differential Revision: D17319124

Pulled By: bddppq

fbshipit-source-id: 22f5d5f1ca9c67e66a7ccf983b2d2f889a74e729
This commit is contained in:
J M Dieterich
2019-09-11 12:29:16 -07:00
committed by Facebook Github Bot
parent 075adb4d2d
commit 00d967c39d
5 changed files with 1 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_jit_fuser.py
View File

@ -47,7 +47,6 @@ class TestFuser(JitTestCase):
self._test_fused_abs(device="cuda")
@unittest.skipIf(not RUN_CUDA, "requires CUDA")
@skipIfRocm
def test_zero_element_tensors(self):
def decode(sin_t, cos_t):
theta = torch.atan2(sin_t.float(), cos_t.float())
@ -358,7 +357,6 @@ class TestFuser(JitTestCase):
# If this is a real problem, we'll need to revisit Torchscript Function
# lifetimes in Python.
@unittest.skipIf(not RUN_CUDA, "fuser requires CUDA")
@skipIfRocm
def test_lerp(self):
start = torch.randn(4, 1, dtype=torch.float, device='cuda')
end = torch.randn(1, 4, dtype=torch.float, device='cuda')