未验证 提交 2534ca7e 编写于 作者: Z zhangkaihuo 提交者: GitHub

test success on cuda11.7 (#47348)

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