Skip to content

Commit 80ac92c

Browse files
Joao Gomesfacebook-github-bot
authored andcommitted
[fbsync] Revert "[proto] Enable GPU tests on prototype (#6665)" (#6809)
Summary: This reverts commit d0de55d. Reviewed By: YosuaMichael Differential Revision: D40588158 fbshipit-source-id: 877a172d7dd807b0c90255bd14129a90768bcc76
1 parent cbe6af1 commit 80ac92c

File tree

2 files changed

+1
-84
lines changed

2 files changed

+1
-84
lines changed

.github/workflows/prototype-tests-gpu.yml

Lines changed: 0 additions & 80 deletions
This file was deleted.

test/test_prototype_transforms_functional.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -174,10 +174,7 @@ def test_cuda_vs_cpu(self, info, args_kwargs):
174174
output_cpu = info.kernel(input_cpu, *other_args, **kwargs)
175175
output_cuda = info.kernel(input_cuda, *other_args, **kwargs)
176176

177-
try:
178-
assert_close(output_cuda, output_cpu, check_device=False, **info.closeness_kwargs)
179-
except AssertionError:
180-
pytest.xfail("CUDA vs CPU tolerance issue to be fixed")
177+
assert_close(output_cuda, output_cpu, check_device=False, **info.closeness_kwargs)
181178

182179
@sample_inputs
183180
@pytest.mark.parametrize("device", cpu_and_gpu())

0 commit comments

Comments
 (0)