未验证 提交 a4c04cb1 编写于 作者: Q Qinghe JING 提交者: GitHub

Merge pull request #62 from qjing666/new_version

fix bugs in New version
......@@ -106,9 +106,9 @@ template <typename DeviceContext, typename T>
class MpcElementwiseAddKernel : public MpcOpKernel<T> {
public:
void ComputeImpl(const framework::ExecutionContext &ctx) const override {
auto *in_x_t = ctx.Input<LoDTensor>("X");
auto *in_y_t = ctx.Input<LoDTensor>("Y");
auto *out_t = ctx.Output<LoDTensor>("Out");
auto *in_x_t = ctx.Input<Tensor>("X");
auto *in_y_t = ctx.Input<Tensor>("Y");
auto *out_t = ctx.Output<Tensor>("Out");
int axis = ctx.Attr<int>("axis");
......
......@@ -22,7 +22,7 @@ import platform
from setuptools import find_packages
from setuptools import setup
from paddle_fl.version import fl_version
def python_version():
"""
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册