提交 d0ad82cf 编写于 作者: Q Qiao Longfei 提交者: GitHub

fix nv_library (#4370)

* fix nv_library

* fix symbol in gpu_info.h
上级 17827d7e
......@@ -253,7 +253,7 @@ function(nv_library TARGET_NAME)
foreach(source_file ${nv_library_SRCS})
string(REGEX REPLACE "\\.[^.]*$" "" source ${source_file})
if(EXISTS ${CMAKE_CURRENT_SOURCE_DIR}/${source}.h)
list(APPEND cc_library_HEADERS ${CMAKE_CURRENT_SOURCE_DIR}/${source}.h)
list(APPEND nv_library_HEADERS ${CMAKE_CURRENT_SOURCE_DIR}/${source}.h)
endif()
endforeach()
add_style_check_target(${TARGET_NAME} ${nv_library_SRCS} ${nv_library_HEADERS})
......
......@@ -36,7 +36,7 @@ int GetCurrentDeviceId();
//! Set the GPU device id for next execution.
void SetDeviceId(int device_id);
//Get the memory usage of current GPU device.
//! Get the memory usage of current GPU device.
void GpuMemoryUsage(size_t &available, size_t &total);
//! Get the maximum allocation size of current GPU device.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册