Merge pull request #12659 from sneaxiy/refine_softmax_with_cross_entropy

Fix 'softmax_with_cross_entropy_op' dependency error
7 years ago · 8dda526a45
parent af8b1b993f 2c560623d1
commit 8dda526a45
1 changed files with 6 additions and 1 deletions
--- a/paddle/fluid/operators/CMakeLists.txt
+++ b/paddle/fluid/operators/CMakeLists.txt
@ -235,7 +235,12 @@ else()
 endif()

 op_library(cross_entropy_op DEPS cross_entropy)
-op_library(softmax_with_cross_entropy_op DEPS cross_entropy softmax)
+if(WITH_GPU)
+  op_library(softmax_with_cross_entropy_op DEPS cross_entropy softmax cub)
+else()
+  op_library(softmax_with_cross_entropy_op DEPS cross_entropy softmax)
+endif()
+
 op_library(softmax_op DEPS softmax)
 op_library(sequence_softmax_op DEPS softmax)
 if (WITH_GPU AND TENSORRT_FOUND)