From 2534ca7ebd39586b5c54921d852f19185d415a90 Mon Sep 17 00:00:00 2001 From: zhangkaihuo Date: Wed, 26 Oct 2022 16:45:09 +0800 Subject: [PATCH] test success on cuda11.7 (#47348) --- paddle/phi/api/yaml/sparse_ops.yaml | 3 ++- paddle/phi/kernels/sparse/empty_kernel.cc | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/paddle/phi/api/yaml/sparse_ops.yaml b/paddle/phi/api/yaml/sparse_ops.yaml index a7111e5dee3..a7b0d7a323b 100644 --- a/paddle/phi/api/yaml/sparse_ops.yaml +++ b/paddle/phi/api/yaml/sparse_ops.yaml @@ -379,7 +379,8 @@ args : (Tensor input, Tensor x, Tensor y, float alpha=1.0, float beta=1.0) output : Tensor(out) infer_meta : - func : AddmmInferMeta + func : UnchangedInferMeta + param : [input] kernel : func : addmm_csr_dense {dense, sparse_csr, dense -> dense}, addmm_csr_csr {sparse_csr, sparse_csr, sparse_csr -> sparse_csr}, diff --git a/paddle/phi/kernels/sparse/empty_kernel.cc b/paddle/phi/kernels/sparse/empty_kernel.cc index 96a7301c589..49a377ca70f 100644 --- a/paddle/phi/kernels/sparse/empty_kernel.cc +++ b/paddle/phi/kernels/sparse/empty_kernel.cc @@ -31,6 +31,7 @@ void EmptyLikeCooKernel(const Context& dev_ctx, const DenseTensor& x_values = x.values(); DenseTensor* out_values = out->mutable_values(); out_values->Resize(x_values.dims()); + out->set_meta(x.meta()); dev_ctx.template Alloc(out_values); } @@ -44,6 +45,7 @@ void EmptyLikeCsrKernel(const Context& dev_ctx, const DenseTensor& x_values = x.values(); DenseTensor* out_values = out->mutable_values(); out_values->Resize(x_values.dims()); + out->set_meta(x.meta()); dev_ctx.template Alloc(out_values); } -- GitLab