diff --git a/src/memory/HeapMemory_unit_tests.hpp b/src/memory/HeapMemory_unit_tests.hpp index 60b2b502430c6e9b6f90f53609b5a9902b7dfc41..7c2a78fb56dc1917224fd8a4708ec62a147e8df9 100644 --- a/src/memory/HeapMemory_unit_tests.hpp +++ b/src/memory/HeapMemory_unit_tests.hpp @@ -8,8 +8,12 @@ #ifndef HEAPMEMORY_UNIT_TESTS_HPP_ #define HEAPMEMORY_UNIT_TESTS_HPP_ +#include "config.h" + #include "memory/HeapMemory.hpp" +#ifdef NVCC #include "memory/CudaMemory.cuh" +#endif BOOST_AUTO_TEST_SUITE( HeapMemory_test ) @@ -79,7 +83,9 @@ template<typename T> void test() BOOST_AUTO_TEST_CASE( use ) { test<HeapMemory>(); +#ifdef NVCC test<CudaMemory>(); +#endif }