提交 9ae50dd0 编写于 作者: P peizhilin

fix gpu buils issue on windows test=develop

上级 c981bf0f
......@@ -447,6 +447,9 @@ function(nv_test TARGET_NAME)
set(multiValueArgs SRCS DEPS)
cmake_parse_arguments(nv_test "${options}" "${oneValueArgs}" "${multiValueArgs}" ${ARGN})
cuda_add_executable(${TARGET_NAME} ${nv_test_SRCS})
if(WIN32)
list(APPEND nv_test_DEPS shlwapi)
endif(WIN32)
target_link_libraries(${TARGET_NAME} ${nv_test_DEPS} paddle_gtest_main lod_tensor memory gtest gflags glog)
add_dependencies(${TARGET_NAME} ${nv_test_DEPS} paddle_gtest_main lod_tensor memory gtest gflags glog)
common_link(${TARGET_NAME})
......
......@@ -15,6 +15,7 @@
#include <gtest/gtest.h>
#include <algorithm>
#include <iostream>
#include <map>
#include <random>
#define PADDLE_CUDA_FP16
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册