From 8b53a3d2311154201cb4ade5c8f42f5c226eefb1 Mon Sep 17 00:00:00 2001
From: Da Zheng <zhengda1936@gmail.com>
Date: Sat, 12 May 2018 23:19:37 -0700
Subject: [PATCH] support L2Norm backward for GPU. (#10874)

* support gpu.

* Fix.
---
 src/operator/tensor/broadcast_reduce_op_value.cu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/operator/tensor/broadcast_reduce_op_value.cu b/src/operator/tensor/broadcast_reduce_op_value.cu
index f1b19b4a8574..5065b9fcc7bb 100644
--- a/src/operator/tensor/broadcast_reduce_op_value.cu
+++ b/src/operator/tensor/broadcast_reduce_op_value.cu
@@ -104,5 +104,9 @@ NNVM_REGISTER_OP(norm)
 .set_attr<FCompute>("FCompute<gpu>", L2NormCompute<gpu>)
 .set_attr<FComputeEx>("FComputeEx<gpu>", L2NormComputeEx<gpu>);
 
+NNVM_REGISTER_OP(_backward_norm)
+.set_attr<FCompute>("FCompute<gpu>", L2NormGradCompute<gpu>)
+.set_attr<FComputeEx>("FComputeEx<gpu>", L2NormGradComputeEx<gpu>);
+
 }  // namespace op
 }  // namespace mxnet