未验证 提交 20b7c99c 编写于 作者: C Chen Weihang 提交者: GitHub

fix detail error for scale (#38213)

上级 fc701369
......@@ -24,7 +24,7 @@ add_subdirectory(tests)
# make an unity target for compile deps
set(PTEN_DEPS convert_utils dense_tensor pten_context kernel_factory kernel_context)
set(PTEN_DEPS scale_kernel_eigen)
set(PTEN_DEPS ${PTEN_DEPS} scale_kernel_eigen)
set(PTEN_DEPS ${PTEN_DEPS} math_cpu linalg_cpu creation_cpu manipulation_cpu)
set(PTEN_DEPS ${PTEN_DEPS} nary unary binary)
if(WITH_GPU OR WITH_ROCM)
......
......@@ -36,14 +36,6 @@ void Mean(const CUDAContext& dev_ctx,
DataType out_dtype,
DenseTensor* out);
template <typename T>
void Scale(const CUDAContext& dev_ctx,
const DenseTensor& x,
const Scalar& scale,
float bias,
bool bias_after_scale,
DenseTensor* out);
template <typename T>
void Add(const CUDAContext& dev_ctx,
const DenseTensor& x,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册