Merge pull request #2068 from gangliao/gpu_conv

Adjust float precision in img_conv2 for unit test
8 years ago · aaef5867dd
parent f003a63bfd 6dc58cdd25
commit aaef5867dd
1 changed files with 3 additions and 0 deletions
--- a/paddle/gserver/tests/test_NetworkCompare.cpp
+++ b/paddle/gserver/tests/test_NetworkCompare.cpp
@ -261,9 +261,12 @@ TEST(Compare, img_conv2) {
  std::string config_file_a = "./gserver/tests/img_conv_cudnn.py";
  std::string config_file_b = "./gserver/tests/img_conv_exconv.py";
  bool useGpu = FLAGS_use_gpu;
+  double eps = FLAGS_checkgrad_eps;
  FLAGS_use_gpu = true;
+  FLAGS_checkgrad_eps = 1e-2;
  compareNetwork(config_file_a, config_file_b);
  FLAGS_use_gpu = useGpu;
+  FLAGS_checkgrad_eps = eps;
 }
 #endif