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

fix compilation failure (#24090)

上级 797f7cbd
...@@ -8,6 +8,9 @@ function(math_library TARGET) ...@@ -8,6 +8,9 @@ function(math_library TARGET)
set(cu_srcs) set(cu_srcs)
set(hip_srcs) set(hip_srcs)
set(math_common_deps device_context framework_proto enforce) set(math_common_deps device_context framework_proto enforce)
if (WITH_GPU)
list(APPEND math_common_deps cub)
endif()
set(multiValueArgs DEPS) set(multiValueArgs DEPS)
cmake_parse_arguments(math_library "${options}" "${oneValueArgs}" cmake_parse_arguments(math_library "${options}" "${oneValueArgs}"
"${multiValueArgs}" ${ARGN}) "${multiValueArgs}" ${ARGN})
...@@ -37,7 +40,7 @@ math_library(concat_and_split) ...@@ -37,7 +40,7 @@ math_library(concat_and_split)
math_library(context_project DEPS im2col math_function) math_library(context_project DEPS im2col math_function)
math_library(cross_entropy) math_library(cross_entropy)
math_library(cos_sim_functor) math_library(cos_sim_functor)
math_library(depthwise_conv DEPS cub) math_library(depthwise_conv)
math_library(im2col) math_library(im2col)
math_library(sample_prob) math_library(sample_prob)
math_library(sampler) math_library(sampler)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册