diff options
| author | surgan12 <cs17btech11038@iith.ac.in> | 2019-01-01 23:09:45 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2019-01-01 23:12:17 -0800 |
| commit | b52420742d70f01ed79ce9e55ec476f30c6e9423 (patch) | |
| tree | 0b316aa75cc52a4880f243fe5b08bf96a06505f9 /test/test_cuda.py | |
| parent | 8278a8b16f3f7b9e0f45d0596ae6c0f200864dd2 (diff) | |
| download | pytorch-b52420742d70f01ed79ce9e55ec476f30c6e9423.tar.gz pytorch-b52420742d70f01ed79ce9e55ec476f30c6e9423.tar.bz2 pytorch-b52420742d70f01ed79ce9e55ec476f30c6e9423.zip | |
clamp fixes (#15479)
Summary: fix to #15338 .
Differential Revision: D13564343
Pulled By: soumith
fbshipit-source-id: be64b572945533e10ae6f627d335b47f093720a3
Diffstat (limited to 'test/test_cuda.py')
| -rw-r--r-- | test/test_cuda.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/test/test_cuda.py b/test/test_cuda.py index 7fb5ed1b8b..28ccc6a2c3 100644 --- a/test/test_cuda.py +++ b/test/test_cuda.py @@ -19,7 +19,7 @@ from test_torch import _TestTorchMixin from common_methods_invocations import tri_tests_args, tri_large_tests_args, \ run_additional_tri_tests, _compare_trilu_indices, _compare_large_trilu_indices from common_utils import TestCase, get_gpu_type, to_gpu, freeze_rng_state, run_tests, \ - PY3, IS_WINDOWS, NO_MULTIPROCESSING_SPAWN, skipIfRocm, TEST_NUMPY, TEST_WITH_ROCM, load_tests + PY3, IS_WINDOWS, NO_MULTIPROCESSING_SPAWN, skipIfRocm, TEST_NUMPY, TEST_WITH_ROCM, load_tests, iter_indices # load_tests from common_utils is used to automatically filter tests for # sharding on sandcastle. This line silences flake warnings @@ -1230,6 +1230,9 @@ class TestCuda(TestCase): z = torch.cat([x, y], 0) self.assertEqual(z.get_device(), x.get_device()) + def test_clamp(self): + _TestTorchMixin._test_clamp(self, 'cuda') + def test_cat(self): SIZE = 10 for dim in range(-3, 3): |