未验证 提交 ab2e2842 编写于 作者: Z Zeng Jinle 提交者: GitHub

fix compilation failure (#24091)

上级 ddcdd4a7
......@@ -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)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册