Add classes GPUDeviceTest and CPUDeviceTest.
authorFelix Abecassis <fabecassis@nvidia.com>
Tue, 26 May 2015 18:21:36 +0000 (11:21 -0700)
committerFelix Abecassis <fabecassis@nvidia.com>
Tue, 26 May 2015 20:55:48 +0000 (13:55 -0700)
commit8d2010489a2e291b4747d3b21b5cadc97d84cb00
tree03ae50dc1b6f56a65becb249d4fc2a315bb83e8b
parent43d538f0969601a6cb96b51d64741482b45f9e0b
Add classes GPUDeviceTest and CPUDeviceTest.

These new classes can be used to implement test cases that are only
running on the GPU or the CPU. The goal is to move all calls to
Caffe::set_mode() inside the test framework, to discourage any test to
change the mode halfway through the execution, which is documented to
be illegal.
include/caffe/test/test_caffe_main.hpp