fix compilation failure (#24091)

revert-22778-infer_var_type
Zeng Jinle 5 years ago committed by GitHub
parent ddcdd4a709
commit ab2e284235
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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)

Loading…
Cancel
Save