未验证 提交 66a9d71a 编写于 作者: C Chen Weihang 提交者: GitHub

add scale lost deps (#38237)

上级 d80fe268
if(WITH_GPU) if(WITH_GPU)
nv_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function) nv_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor kernel_context kernel_factory eigen_function)
elseif(WITH_ROCM) elseif(WITH_ROCM)
hip_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function) hip_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor kernel_context kernel_factory eigen_function)
else() else()
cc_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor eigen_function) cc_library(scale_kernel_eigen SRCS scale_kernel.cc DEPS dense_tensor kernel_context kernel_factory eigen_function)
endif() endif()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册