Fix XMakeLists.txt to function as a submodule (#13)

dmlc · May 31, 2017 · 29f826a · 29f826a
1 parent a6e09b5
commit 29f826a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -47,9 +47,9 @@ if(BUILD_DOCS)
     add_subdirectory(docs)
 endif(BUILD_DOCS)
 
-set(DLPACK_INCLUDE_DIR ${CMAKE_SOURCE_DIR}/include)
+set(DLPACK_INCLUDE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/include)
 
 include_directories(${DLPACK_INCLUDE_DIR})
-add_executable(mock ${CMAKE_SOURCE_DIR}/src/mock.cc)
+add_executable(mock ${CMAKE_CURRENT_SOURCE_DIR}/src/mock.cc)