| commit | 90a52c39045a9f5aa5d695256af520fae00019f6 | [log] [tgz] |
|---|---|---|
| author | Luke Yeager <lukeyeager@users.noreply.github.com> | Mon Jun 19 12:10:46 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Mon Jun 19 12:21:24 2017 -0700 |
| tree | 7e3bb0e47f8c4167d04ce19872a895d5d2d5d199 | |
| parent | 980e2a6b59d3983a1766a9873597235dd9217175 [diff] |
Skip TestInferDevice if no GPU support Summary: Working towards https://github.com/caffe2/caffe2/pull/817. ``` E AttributeError: Method CopyCPUToGPU is not a registered operator. Did you mean: [] ``` https://travis-ci.org/caffe2/caffe2/jobs/243867951 Closes https://github.com/caffe2/caffe2/pull/818 Differential Revision: D5276735 Pulled By: akyrola fbshipit-source-id: 35d9df19330ae522037e8a5d721d83dc2e5aa4dc
diff --git a/caffe2/python/core_test.py b/caffe2/python/core_test.py index ca0ff03..44138e8 100644 --- a/caffe2/python/core_test.py +++ b/caffe2/python/core_test.py
@@ -441,6 +441,7 @@ self.assertEqual(net.Proto().op[0].name, '{}:{}'.format(filename, line)) +@unittest.skipIf(not workspace.has_gpu_support, 'No GPU support') class TestInferDevice(test_util.TestCase): def setUp(self):