diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2014-07-12 06:58:35 -0700 |
|---|---|---|
| committer | Jeff Donahue <jeff.donahue@gmail.com> | 2014-07-14 02:17:58 -0700 |
| commit | e5ce15d29d183f8a2cdc48ab12e90d254cf4fad1 (patch) | |
| tree | 329475ccacc57365f43c21880cde9ad6b5325593 /src/caffe/test/test_common.cpp | |
| parent | 9c61462cac1cea0856d180bc024e9937f357b274 (diff) | |
| download | caffeonacl-e5ce15d29d183f8a2cdc48ab12e90d254cf4fad1.tar.gz caffeonacl-e5ce15d29d183f8a2cdc48ab12e90d254cf4fad1.tar.bz2 caffeonacl-e5ce15d29d183f8a2cdc48ab12e90d254cf4fad1.zip | |
Move test headers to include/.
Add a test param to test both CPU and GPU (with both float and double
Dtypes).
Diffstat (limited to 'src/caffe/test/test_common.cpp')
| -rw-r--r-- | src/caffe/test/test_common.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/caffe/test/test_common.cpp b/src/caffe/test/test_common.cpp index 13c2d951..a452b612 100644 --- a/src/caffe/test/test_common.cpp +++ b/src/caffe/test/test_common.cpp @@ -13,7 +13,7 @@ namespace caffe { class CommonTest : public ::testing::Test {}; -TEST_F(CommonTest, TestCublasHandler) { +TEST_F(CommonTest, TestCublasHandlerGPU) { int cuda_device_id; CUDA_CHECK(cudaGetDevice(&cuda_device_id)); EXPECT_TRUE(Caffe::cublas_handle()); @@ -53,10 +53,10 @@ TEST_F(CommonTest, TestRandSeedGPU) { SyncedMemory data_b(10 * sizeof(unsigned int)); Caffe::set_random_seed(1701); CURAND_CHECK(curandGenerate(Caffe::curand_generator(), - reinterpret_cast<unsigned int*>(data_a.mutable_gpu_data()), 10)); + static_cast<unsigned int*>(data_a.mutable_gpu_data()), 10)); Caffe::set_random_seed(1701); CURAND_CHECK(curandGenerate(Caffe::curand_generator(), - reinterpret_cast<unsigned int*>(data_b.mutable_gpu_data()), 10)); + static_cast<unsigned int*>(data_b.mutable_gpu_data()), 10)); for (int i = 0; i < 10; ++i) { EXPECT_EQ(((const unsigned int*)(data_a.cpu_data()))[i], ((const unsigned int*)(data_b.cpu_data()))[i]); |