From 304427cd93ba5d2819e4e4d809da4f520d00421e Mon Sep 17 00:00:00 2001
From: yujianfeng <yujianfeng5@huawei.com>
Date: Tue, 23 Jun 2020 11:54:51 +0800
Subject: [PATCH] Add an output to apply_proximal_adagrad op register

---
 .../ccsrc/kernel/cpu/sparse_apply_proximal_adagrad_cpu_kernel.h | 1 +
 tests/st/ops/cpu/test_sparse_apply_proximal_adagrad_op.py       | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/mindspore/ccsrc/kernel/cpu/sparse_apply_proximal_adagrad_cpu_kernel.h b/mindspore/ccsrc/kernel/cpu/sparse_apply_proximal_adagrad_cpu_kernel.h
index 082809a9c2..00ca5bc693 100644
--- a/mindspore/ccsrc/kernel/cpu/sparse_apply_proximal_adagrad_cpu_kernel.h
+++ b/mindspore/ccsrc/kernel/cpu/sparse_apply_proximal_adagrad_cpu_kernel.h
@@ -48,6 +48,7 @@ MS_REG_CPU_KERNEL(SparseApplyProximalAdagrad,
                     .AddInputAttr(kNumberTypeFloat32)
                     .AddInputAttr(kNumberTypeFloat32)
                     .AddInputAttr(kNumberTypeInt32)
+                    .AddOutputAttr(kNumberTypeFloat32)
                     .AddOutputAttr(kNumberTypeFloat32),
                   SparseApplyProximalAdagradCPUKernel);
 }  // namespace kernel
diff --git a/tests/st/ops/cpu/test_sparse_apply_proximal_adagrad_op.py b/tests/st/ops/cpu/test_sparse_apply_proximal_adagrad_op.py
index 0eaa11a201..f90e11ec3d 100644
--- a/tests/st/ops/cpu/test_sparse_apply_proximal_adagrad_op.py
+++ b/tests/st/ops/cpu/test_sparse_apply_proximal_adagrad_op.py
@@ -44,4 +44,4 @@ def test_net():
     context.set_context(mode=context.GRAPH_MODE, device_target="CPU")
     sparse_apply_proximal_adagrad = Net()
     output = sparse_apply_proximal_adagrad(gradient, indices)
-    print(output.asnumpy()[0])
+    print(output[0].asnumpy())