diff options
author | Sergey Karayev <sergeykarayev@gmail.com> | 2014-02-23 13:02:32 -0800 |
---|---|---|
committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-02-26 15:42:37 -0800 |
commit | 4eca4d41cd683d261faadd927bfe87fa06b62c26 (patch) | |
tree | ac3fcba162e53df2cd6355de9f34dfd30500e421 /src/caffe | |
parent | 800ba3e9faeef37db97a07ee4626a85e9ba4e7ad (diff) | |
download | caffeonacl-4eca4d41cd683d261faadd927bfe87fa06b62c26.tar.gz caffeonacl-4eca4d41cd683d261faadd927bfe87fa06b62c26.tar.bz2 caffeonacl-4eca4d41cd683d261faadd927bfe87fa06b62c26.zip |
Adding GPU coverage to the DataLayer test.
Diffstat (limited to 'src/caffe')
-rw-r--r-- | src/caffe/test/test_data_layer.cpp | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/src/caffe/test/test_data_layer.cpp b/src/caffe/test/test_data_layer.cpp index 66e99568..719c50eb 100644 --- a/src/caffe/test/test_data_layer.cpp +++ b/src/caffe/test/test_data_layer.cpp @@ -81,6 +81,7 @@ TYPED_TEST(DataLayerTest, TestRead) { EXPECT_EQ(this->blob_top_label_->channels(), 1); EXPECT_EQ(this->blob_top_label_->height(), 1); EXPECT_EQ(this->blob_top_label_->width(), 1); + // Go through the data 100 times for (int iter = 0; iter < 100; ++iter) { layer.Forward(this->blob_bottom_vec_, &this->blob_top_vec_); @@ -94,6 +95,21 @@ TYPED_TEST(DataLayerTest, TestRead) { } } } + + // Same test, in GPU mode. + Caffe::set_mode(Caffe::GPU); + for (int iter = 0; iter < 100; ++iter) { + layer.Forward(this->blob_bottom_vec_, &this->blob_top_vec_); + for (int i = 0; i < 5; ++i) { + EXPECT_EQ(i, this->blob_top_label_->cpu_data()[i]); + } + for (int i = 0; i < 5; ++i) { + for (int j = 0; j < 24; ++j) { + EXPECT_EQ(i, this->blob_top_data_->cpu_data()[i * 24 + j]) + << "debug: i " << i << " j " << j; + } + } + } } } |