Fix

Author: co63oc

paddle/phi/kernels/impl/renorm_grad_kernel_impl.h

@@ -37,6 +37,9 @@ void RenormGradKernel(const Context& dev_ctx,
   auto dimension_each = input_dims[dim];
   dx->Resize(x.dims());
   dev_ctx.template Alloc<T>(dx);
+  if (dx && dx->numel() == 0) {
+    return;
+  }
   phi::funcs::RenormGradFunc(dev_ctx,
                              x_data,
                              dout_data,

paddle/phi/kernels/impl/renorm_kernel_impl.h

@@ -11,6 +11,7 @@
 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 // See the License for the specific language governing permissions and
 // limitations under the License.
+#pragma once
 
 #include "paddle/phi/core/dense_tensor.h"
 #include "paddle/phi/kernels/funcs/eigen/common.h"
@@ -28,6 +29,9 @@ void RenormKernel(const Context& dev_ctx,
                   DenseTensor* out) {
   out->Resize(x.dims());
   dev_ctx.template Alloc<T>(out);
+  if (out && out->numel() == 0) {
+    return;
+  }
   auto x_ptr = x.template data<T>();
   auto numel = x.numel();
   int dim = axis;

test/legacy_test/test_renorm_op.py

@@ -122,6 +122,41 @@ def test_dygraph_api(self):
             np.testing.assert_allclose(expected, np.array(y), rtol=1e-05)
 
 
+class TestRenormAPI_ZeroSize(unittest.TestCase):
+    def input_data(self):
+        self.shape = [2, 0, 3]
+        self.data_x = np.random.random(self.shape).astype('float64')
+        self.p = 1.0
+        self.dim = 2
+        self.max_norm = 2.05
+
+    def test_renorm_api(self):
+        paddle.enable_static()
+        self.input_data()
+
+        # case 1:
+        with paddle.static.program_guard(
+            paddle.static.Program(), paddle.static.Program()
+        ):
+            x = paddle.static.data(name="x", shape=self.shape, dtype='float64')
+            z = paddle.renorm(x, self.p, self.dim, self.max_norm)
+            exe = base.Executor(base.CPUPlace())
+            (res,) = exe.run(
+                feed={"x": self.data_x}, fetch_list=[z], return_numpy=False
+            )
+        np.testing.assert_allclose(np.array(res).shape, self.shape)
+
+    def test_dygraph_api(self):
+        self.input_data()
+        with base.dygraph.guard(base.CPUPlace()):
+            x = paddle.to_tensor(self.data_x, stop_gradient=False)
+            y = paddle.renorm(x, 1.0, 2, 2.05)
+            np.testing.assert_allclose(np.array(y).shape, self.shape)
+            z = paddle.mean(y)
+            z.backward(retain_graph=True)
+            np.testing.assert_allclose(x.grad.shape, x.shape)
+
+
 if __name__ == '__main__':
     paddle.enable_static()
     unittest.main()