diff --git a/paddle/majel/CMakeLists.txt b/paddle/majel/CMakeLists.txt index 47f33dd83210e405763faa059f29524ece8e4f4f..cb8bece00e1c77a8ffacf4cf8cc7a73fd508fbd3 100644 --- a/paddle/majel/CMakeLists.txt +++ b/paddle/majel/CMakeLists.txt @@ -1,4 +1,5 @@ -cc_library(majel SRCS place.cc ddim.cc) +cc_library(place SRCS place.cc) +cc_library(ddim SRCS ddim.cc) if(WITH_TESTING) add_subdirectory(test) diff --git a/paddle/majel/dim.h b/paddle/majel/dim.h index 5eff4c05cc240de1f98fe09e8f3fa07efdd4b976..c4b0c6aea683384d4657dd5db6f419b9e1108704 100644 --- a/paddle/majel/dim.h +++ b/paddle/majel/dim.h @@ -5,8 +5,8 @@ #include #include +#include "paddle/majel/detail/cuda_assert.h" #include "paddle/majel/detail/hostdevice.h" -#include "paddle/majel/detail/util.h" namespace majel { diff --git a/paddle/majel/test/CMakeLists.txt b/paddle/majel/test/CMakeLists.txt index 424537995261116f8bf9a471ebd43553a488ea79..9d632d568e9908c5a1505f23f43c8bd3daecff18 100644 --- a/paddle/majel/test/CMakeLists.txt +++ b/paddle/majel/test/CMakeLists.txt @@ -1,12 +1,12 @@ cc_test(place_test SRCS place_test.cc - DEPS majel) + DEPS place) cc_test(ddim_test SRCS ddim_test.cc - DEPS majel) + DEPS ddim) if(WITH_GPU) nv_test(cuda_test SRCS cuda_test.cu) - nv_test(dim_test SRCS dim_test.cu DEPS majel) + nv_test(dim_test SRCS dim_test.cu DEPS ddim) endif()