nn.functional.group_norm 在float16格式下，数据排布为NHWC且输入矩阵rank=3时报错FatalError: Erroneous arithmetic operation

[Eddie-Wang1120]

bug描述 Describe the Bug

nn.functional.group_norm 在float16格式下，数据排布为NHWC且输入矩阵rank=3时报错。
在相同的数据格式和矩阵rank场景下，nn.functional.group_norm支持NCHW数据排布。

复现步骤（最小代码集）

import paddle

x = paddle.arange(72, dtype="float16").reshape((2, 6, 6))
group_norm_out = paddle.nn.functional.group_norm(x, num_groups=6, data_format='NHWC')

报错信息

grep: warning: GREP_OPTIONS is deprecated; please use an alias or script
W0416 08:34:06.418946  6860 gpu_resources.cc:119] Please NOTE: device: 0, GPU Compute Capability: 8.6, Driver API Version: 12.3, Runtime API Version: 12.0
W0416 08:34:06.448040  6860 gpu_resources.cc:164] device: 0, cuDNN Version: 8.8.


--------------------------------------
C++ Traceback (most recent call last):
--------------------------------------
0   paddle::pybind::eager_api_group_norm(_object*, _object*, _object*)
1   group_norm_ad_func(paddle::Tensor const&, paddle::optional<paddle::Tensor> const&, paddle::optional<paddle::Tensor> const&, float, int, std::string)
2   paddle::experimental::group_norm_intermediate(paddle::Tensor const&, paddle::optional<paddle::Tensor> const&, paddle::optional<paddle::Tensor> const&, float, int, std::string const&)

----------------------
Error Message Summary:
----------------------
FatalError: `Erroneous arithmetic operation` is detected by the operating system.
  [TimeInfo: *** Aborted at 1713256446 (unix time) try "date -d @1713256446" if you are using GNU date ***]
  [SignalInfo: *** SIGFPE (@0x7fd7b9c6e5bb) received by PID 6860 (TID 0x7fd8415c2740) from PID 18446744072531404219 ***]

Floating point exception

复现环境

paddlepaddle-gpu develop
最新commit
NVIDIA GeForce RTX 3050 16G

其他补充信息 Additional Supplementary Information

group_norm函数在非常多的模型中都需要用到，出现报错的场景也是模型推理中使用频率非常高的，非常希望该问题能得到重视以及解决！非常感谢！

[yuanlehome]

问题已经收到，感谢使用Paddle。
你的问题是由于当前paddle.nn.functional.group_norm动态图api仅支持NCHW格式的输入导致的，至于对于NHWC格式的支持计划，我需要向其他同事了解一下再做回复。

[Eddie-Wang1120]

问题已经收到，感谢使用Paddle。 你的问题是由于当前paddle.nn.functional.group_norm动态图api仅支持NCHW格式的输入导致的，至于对于NHWC格式的支持计划，我需要向其他同事了解一下再做回复。

谢谢回复！非常希望问题得到解决，再次向paddle员工表示感谢！

[Eddie-Wang1120]

问题已经收到，感谢使用Paddle。 你的问题是由于当前paddle.nn.functional.group_norm动态图api仅支持NCHW格式的输入导致的，至于对于NHWC格式的支持计划，我需要向其他同事了解一下再做回复。

还有一个问题，我测试后发现这个api是支持fp32精度下的NHWC格式输入的，只有半精度不支持，可以参考一下这个测试结果

[yuanlehome]

你跑的是GPU？我看GPU已经支持了半精度的NHWC格式，出现报错的原因可能是有些BUG。。。

[yuanlehome]

你试一下rank为4呢？目前只支持了rank==4的情况。
是对rank==3有强需求吗？

[Eddie-Wang1120]

你试一下rank为4呢？目前只支持了rank==4的情况。 是对rank==3有强需求吗？

跑的是gpu，确实是有强需求的，模型推理时会用到rank==3的情况

[zhwesky2010]

@Eddie-Wang1120 请问目前是缺rank=3还是NHWC/NCHW还是float16

[Eddie-Wang1120]

@Eddie-Wang1120 请问目前是缺rank=3还是NHWC/NCHW还是float16

缺少nn.functional.group_norm 在float16格式下，数据排布为NHWC且输入矩阵rank=3时的支持，当前情况下会报错。

[zhwesky2010]

@Eddie-Wang1120 请问目前是缺rank=3还是NHWC/NCHW还是float16

缺少nn.functional.group_norm 在float16格式下，数据排布为NHWC且输入矩阵rank=3时的支持，当前情况下会报错。

需求收到，我们会进行开发。

[Eddie-Wang1120]

请问目前开发的进度如何？ @yuanlehome @zhwesky2010