| commit | be7725b0ba329751edf3ea9683fc6635cdb8ff62 | [log] [tgz] |
|---|---|---|
| author | Luke Yeager <lukeyeager@users.noreply.github.com> | Wed Jul 05 14:19:40 2017 -0700 |
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | Wed Jul 05 14:32:12 2017 -0700 |
| tree | 4a754b6527f869dd287b8c8397b44465c1a93dfb | |
| parent | 87730360d15aa87469c4b69d0c13caf6e4855318 [diff] |
Tests: fix dpm test when only 1 GPU present Summary: https://github.com/caffe2/caffe2/commit/b33894e95da4a7db0b038b9735714f7e8108a58f removed this line: ```py unittest.skipIf(workspace.NumCudaDevices() < 2, "Need at least 2 GPUs.") ``` but forgot to add it back later. ``` _________________________________ DataParallelModelTest.test_equiv __________________________________ ... if p2p_access_pattern is not None and not p2p_access_pattern[ > devices[0], peer ]: E IndexError: index 1 is out of bounds for axis 1 with size 1 ... WARNING:data_parallel_model:** Only 1 GPUs available, GPUs [0, 1] requested ``` /cc akyrola Closes https://github.com/caffe2/caffe2/pull/888 Reviewed By: akyrola Differential Revision: D5341310 Pulled By: harouwu fbshipit-source-id: 8d7f06913c7b5a42009a4033dbb6a48a8e812822
Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.
Caffe2 research award competition request for proposals
Please use Github issues (https://github.com/caffe2/caffe2/issues) to ask questions, report bugs, and request new features.
Please participate in our survey (https://www.surveymonkey.com/r/caffe2). We will send you information about new releases and special developer events/webinars.
Caffe2 is released under the BSD 2-Clause license.