tests/cuda/cuda_main.cpp

0001 /**
0002  * TRACCC library, part of the ACTS project (R&D line)
0003  *
0004  * (c) 2022 CERN for the benefit of the ACTS project
0005  *
0006  * Mozilla Public License Version 2.0
0007  */
0008
0009 #include <cuda_runtime.h>
0010 #include <gtest/gtest.h>
0011
0012 int main(int argc, char** argv) {
0013   testing::InitGoogleTest(&argc, argv);
0014
0015   int r = RUN_ALL_TESTS();
0016
0017   cudaError_t cErr = cudaDeviceReset();
0018
0019   if (cErr == cudaSuccess || cErr == cudaErrorNoDevice ||
0020       cErr == cudaErrorInsufficientDriver) {
0021     return r;
0022   } else {
0023     return static_cast<int>(cErr);
0024   }
0025 }