Skip to content

Commit 00a5155

Browse files
authored
Merge pull request #5717 from Yancey1989/fix_framework_proto_error
Fix no framework proto file
2 parents ba9f252 + d13c3a9 commit 00a5155

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/operators/math/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
add_subdirectory(detail)
22

33
if(WITH_GPU)
4-
nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc im2col.cu DEPS cblas device_context)
4+
nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc im2col.cu DEPS cblas device_context framework_proto)
55
nv_test(math_function_gpu_test SRCS math_function_test.cu DEPS math_function tensor)
66
nv_library(selected_rows_functor SRCS selected_rows_functor.cc selected_rows_functor.cu DEPS selected_rows math_function)
77
nv_test(selected_rows_functor_gpu_test SRCS selected_rows_functor_test.cu DEPS selected_rows_functor)
@@ -15,7 +15,7 @@ if(WITH_GPU)
1515
nv_library(lstm_compute SRCS lstm_compute.cc lstm_compute.cu DEPS device_context activation_functions)
1616
nv_library(gru_compute SRCS gru_compute.cc gru_compute.cu DEPS device_context activation_functions math_function)
1717
else()
18-
cc_library(math_function SRCS math_function.cc im2col.cc DEPS cblas device_context)
18+
cc_library(math_function SRCS math_function.cc im2col.cc DEPS cblas device_context framework_proto)
1919
cc_library(selected_rows_functor SRCS selected_rows_functor.cc DEPS selected_rows math_function)
2020
cc_library(softmax SRCS softmax.cc DEPS device_context)
2121
cc_library(cross_entropy SRCS cross_entropy.cc DEPS device_context)

0 commit comments

Comments
 (0)