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