override group norm (#5019) (#5022)

Author: guanbaoy

csrc/gpu/aten/operators/GroupNorm.cpp

@@ -1,4 +1,7 @@
 #include "Norm.h"
+#ifdef USE_OVERRIDE_OP
+#include "utils/CustomOperatorRegistration.h"
+#endif
 
 using namespace at::AtenIpexTypeXPU::normalization;
 
@@ -987,3 +990,17 @@ std::tuple<Tensor, Tensor, Tensor> native_group_norm_backward(
 
 } // namespace AtenIpexTypeXPU
 } // namespace at
+
+#ifdef USE_OVERRIDE_OP
+namespace {
+
+IPEX_TORCH_LIBRARY_IMPL(aten, XPU, m) {
+  m.impl(
+      "native_group_norm", TORCH_FN((&at::AtenIpexTypeXPU::native_group_norm)));
+  m.impl(
+      "native_group_norm_backward",
+      TORCH_FN((&at::AtenIpexTypeXPU::native_group_norm_backward)));
+}
+
+} // namespace
+#endif

scripts/tools/torchgen/yaml/xpu_functions.yaml

@@ -63,6 +63,8 @@ supported:
 # - sum.dim_IntList
 # - nansum
 # - nansum.out
+#  - native_group_norm
+#  - native_group_norm_backward
 ####################################
 #  - _aminmax
 #  - _aminmax.dim
@@ -459,8 +461,6 @@ supported:
 #  - nanmedian.dim_values
 #  - native_dropout
 #  - native_dropout_backward
-#  - native_group_norm
-#  - native_group_norm_backward
 #  - native_layer_norm
 #  - native_layer_norm_backward
 #  - ne.Scalar