diff options
| author | Sergei Nikolaev <snikolaev@nvidia.com> | 2016-02-02 13:48:18 -0800 |
|---|---|---|
| committer | Sergei Nikolaev <snikolaev@nvidia.com> | 2016-02-02 16:37:14 -0800 |
| commit | dd2099786f11033ded6e9f46bc772ef9b2166399 (patch) | |
| tree | 0b6b83bc4ad981265012d6a598117204e3202408 /src/caffe/test/test_caffe_main.cpp | |
| parent | 2ef584785c8ade90260eb117f189146364494183 (diff) | |
| download | caffeonacl-dd2099786f11033ded6e9f46bc772ef9b2166399.tar.gz caffeonacl-dd2099786f11033ded6e9f46bc772ef9b2166399.tar.bz2 caffeonacl-dd2099786f11033ded6e9f46bc772ef9b2166399.zip | |
Nicely prints GPU names
Diffstat (limited to 'src/caffe/test/test_caffe_main.cpp')
| -rw-r--r-- | src/caffe/test/test_caffe_main.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/caffe/test/test_caffe_main.cpp b/src/caffe/test/test_caffe_main.cpp index c8caf5ac..fccf6f16 100644 --- a/src/caffe/test/test_caffe_main.cpp +++ b/src/caffe/test/test_caffe_main.cpp @@ -34,6 +34,7 @@ int main(int argc, char** argv) { cudaGetDevice(&device); cout << "Current device id: " << device << endl; cudaGetDeviceProperties(&CAFFE_TEST_CUDA_PROP, device); + cout << "Current device name: " << CAFFE_TEST_CUDA_PROP.name << endl; #endif // invoke the test. return RUN_ALL_TESTS(); |