Skip to content

Commit c125f23

Browse files
【CUDA Kernel No.59】Add declaration in group_norm_kernel.h -part(PaddlePaddle#76524)
1 parent 5e4bf73 commit c125f23

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

paddle/phi/kernels/group_norm_kernel.h

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,21 @@ void GroupNormKernel(const Context& dev_ctx,
3838
DenseTensor* mean,
3939
DenseTensor* variance);
4040

41+
template <typename T, typename Context>
42+
void GroupNormNDHWCKernel(const Context& dev_ctx,
43+
const DenseTensor& x,
44+
const paddle::optional<DenseTensor>& residual,
45+
const paddle::optional<DenseTensor>& scale,
46+
const paddle::optional<DenseTensor>& bias,
47+
float epsilon,
48+
int groups,
49+
const std::string& data_layout_str,
50+
const std::string& activation,
51+
DenseTensor* y,
52+
DenseTensor* residual_out,
53+
DenseTensor* mean,
54+
DenseTensor* var);
55+
4156
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
4257
template <typename T, typename AccT = T>
4358
class GroupNormDirectCUDAFunctor {

0 commit comments

Comments
 (0)