From ce66dffeaf5d905c7ced57c0b095a9a0d4c62622 Mon Sep 17 00:00:00 2001 From: oneflow-ci-bot Date: Wed, 8 Jan 2025 07:03:50 +0000 Subject: [PATCH] auto format by CI --- oneflow/core/autograd/gradient_funcs/layer_norm.cpp | 2 +- oneflow/user/kernels/layer_norm_cpu_kernel.cpp | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/oneflow/core/autograd/gradient_funcs/layer_norm.cpp b/oneflow/core/autograd/gradient_funcs/layer_norm.cpp index 515c754c05b..24fdc0e523c 100644 --- a/oneflow/core/autograd/gradient_funcs/layer_norm.cpp +++ b/oneflow/core/autograd/gradient_funcs/layer_norm.cpp @@ -119,7 +119,7 @@ Maybe LayerNorm::Apply(const LayerNormCaptureState* ctx, const TensorTuple *in_grads = *JUST(functional::FuseLayerNormGrad( dy, x, mean, inv_variance, gamma, begin_norm_axis, begin_params_axis, ctx->epsilon)); } else { - UNIMPLEMENTED(); + UNIMPLEMENTED(); } } } else { diff --git a/oneflow/user/kernels/layer_norm_cpu_kernel.cpp b/oneflow/user/kernels/layer_norm_cpu_kernel.cpp index 473b01bc019..757476be533 100644 --- a/oneflow/user/kernels/layer_norm_cpu_kernel.cpp +++ b/oneflow/user/kernels/layer_norm_cpu_kernel.cpp @@ -68,8 +68,8 @@ class FuseLayerNormGradCpuKernel final : public user_op::OpKernel { void Compute(user_op::KernelComputeContext* ctx) const override { TODO(); }; }; -#define REGISTER_FUSE_LAYER_NORM_GRAD_CPU_KERNEL(dtype) \ - REGISTER_USER_KERNEL("fuse_layer_norm_grad") \ +#define REGISTER_FUSE_LAYER_NORM_GRAD_CPU_KERNEL(dtype) \ + REGISTER_USER_KERNEL("fuse_layer_norm_grad") \ .SetCreateFn>() \ .SetIsMatchedHob((user_op::HobDeviceType() == DeviceType::kCPU) \ && (user_op::HobDataType("dy", 0) == GetDataType::value));