fix gpu buils issue on windows test=develop

revert-15207-remove_op_handle_lock_and_fix_var
peizhilin 7 years ago
parent c981bf0f9d
commit 9ae50dd07d

@ -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

Loading…
Cancel
Save