diff --git a/test/Reconstruction_Test.cpp b/test/Reconstruction_Test.cpp index b130d47..999775f 100644 --- a/test/Reconstruction_Test.cpp +++ b/test/Reconstruction_Test.cpp @@ -2,6 +2,7 @@ #include #include +#include "CudaMatrix.h" #include "Function1D.h" #include "Function3D.h" #include "MatlabReader.h" @@ -14,6 +15,9 @@ #include "transmissionReconstruction/reconstruction/buildMatrix/DGradient.h" #include "transmissionReconstruction/reconstruction/buildMatrix/FMM.h" #include "transmissionReconstruction/reconstruction/buildMatrix/buildMatrix.h" + +#include "transmissionReconstruction/reconstruction/buildMatrix/buildMatrix.cuh" + #include "transmissionReconstruction/reconstruction/reconstruction.h" #include "transmissionReconstruction/reconstruction/solvingEquationSystem/TVAL/TVAL.h" @@ -23,6 +27,9 @@ #include "reflectionReconstruction/preprocessData/determineOptimalPulse.h" +#include +#include + @@ -52,6 +59,14 @@ protected: } }; +TEST_F(Reconstruction_Test, GPUMajor) { + int cudaDevice; + cudaGetDevice(&cudaDevice); + cudaDeviceProp prop; + cudaGetDeviceProperties(&prop, cudaDevice); + EXPECT_GT(prop.major,2)<<"GPU major:"<