File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change 1
1
add_subdirectory (detail )
2
2
3
3
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 )
5
5
nv_test (math_function_gpu_test SRCS math_function_test.cu DEPS math_function tensor )
6
6
nv_library (selected_rows_functor SRCS selected_rows_functor.cc selected_rows_functor.cu DEPS selected_rows math_function )
7
7
nv_test (selected_rows_functor_gpu_test SRCS selected_rows_functor_test.cu DEPS selected_rows_functor )
@@ -15,7 +15,7 @@ if(WITH_GPU)
15
15
nv_library (lstm_compute SRCS lstm_compute.cc lstm_compute.cu DEPS device_context activation_functions )
16
16
nv_library (gru_compute SRCS gru_compute.cc gru_compute.cu DEPS device_context activation_functions math_function )
17
17
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 )
19
19
cc_library (selected_rows_functor SRCS selected_rows_functor.cc DEPS selected_rows math_function )
20
20
cc_library (softmax SRCS softmax.cc DEPS device_context )
21
21
cc_library (cross_entropy SRCS cross_entropy.cc DEPS device_context )
You can’t perform that action at this time.
0 commit comments