|
|
|
@ -8,6 +8,9 @@ function(math_library TARGET)
|
|
|
|
|
set(cu_srcs)
|
|
|
|
|
set(hip_srcs)
|
|
|
|
|
set(math_common_deps device_context framework_proto enforce)
|
|
|
|
|
if (WITH_GPU)
|
|
|
|
|
list(APPEND math_common_deps cub)
|
|
|
|
|
endif()
|
|
|
|
|
set(multiValueArgs DEPS)
|
|
|
|
|
cmake_parse_arguments(math_library "${options}" "${oneValueArgs}"
|
|
|
|
|
"${multiValueArgs}" ${ARGN})
|
|
|
|
@ -37,7 +40,7 @@ math_library(concat_and_split)
|
|
|
|
|
math_library(context_project DEPS im2col math_function)
|
|
|
|
|
math_library(cross_entropy)
|
|
|
|
|
math_library(cos_sim_functor)
|
|
|
|
|
math_library(depthwise_conv DEPS cub)
|
|
|
|
|
math_library(depthwise_conv)
|
|
|
|
|
math_library(im2col)
|
|
|
|
|
math_library(sample_prob)
|
|
|
|
|
math_library(sampler)
|
|
|
|
|