|
|
|
@ -1,7 +1,7 @@
|
|
|
|
|
add_subdirectory(detail)
|
|
|
|
|
|
|
|
|
|
if(WITH_GPU)
|
|
|
|
|
nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc im2col.cu DEPS cblas device_context)
|
|
|
|
|
nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc im2col.cu DEPS cblas device_context framework_proto)
|
|
|
|
|
nv_test(math_function_gpu_test SRCS math_function_test.cu DEPS math_function tensor)
|
|
|
|
|
nv_library(selected_rows_functor SRCS selected_rows_functor.cc selected_rows_functor.cu DEPS selected_rows math_function)
|
|
|
|
|
nv_test(selected_rows_functor_gpu_test SRCS selected_rows_functor_test.cu DEPS selected_rows_functor)
|
|
|
|
@ -15,7 +15,7 @@ if(WITH_GPU)
|
|
|
|
|
nv_library(lstm_compute SRCS lstm_compute.cc lstm_compute.cu DEPS device_context activation_functions)
|
|
|
|
|
nv_library(gru_compute SRCS gru_compute.cc gru_compute.cu DEPS device_context activation_functions math_function)
|
|
|
|
|
else()
|
|
|
|
|
cc_library(math_function SRCS math_function.cc im2col.cc DEPS cblas device_context)
|
|
|
|
|
cc_library(math_function SRCS math_function.cc im2col.cc DEPS cblas device_context framework_proto)
|
|
|
|
|
cc_library(selected_rows_functor SRCS selected_rows_functor.cc DEPS selected_rows math_function)
|
|
|
|
|
cc_library(softmax SRCS softmax.cc DEPS device_context)
|
|
|
|
|
cc_library(cross_entropy SRCS cross_entropy.cc DEPS device_context)
|
|
|
|
|