提交 ac993460 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!3288 delete useless code for all reduce

Merge pull request !3288 from yangzhenzhang/delete-useless-code
......@@ -100,11 +100,6 @@ class AllReduce(PrimitiveWithInfer):
self.add_prim_attr('fusion', 0)
self.add_prim_attr('index', 0)
def vm_impl(self, x):
"""Implement by vm mode."""
x = x.asnumpy()
return Tensor(x)
def infer_shape(self, x_shape):
return x_shape
......
......@@ -294,9 +294,6 @@ TEST_F(TestStepParallel, CreatOpInstance) {
ASSERT_TRUE(allreduce_ptr);
if (nullptr != allreduce_ptr) {
MS_LOG(INFO) << "Get PrimitivePyPtr: " << allreduce_ptr->name();
if (!allreduce_ptr->HasComputeFunction()) {
MS_LOG(EXCEPTION) << "" << allreduce_ptr->name() << "'s compute function is not implemented";
}
std::vector<py::object> arglist;
(void)std::transform(attrs.begin(), attrs.end(), std::back_inserter(arglist),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册