Skip to content

Commit 5421f12

Browse files
authored
Revert "[proto] Enable GPU tests on prototype (#6665)" (#6809)
This reverts commit d0de55d.
1 parent f88ab12 commit 5421f12

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)