Skip to content
Open
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
#include "paddle/phi/kernels/distributed_fused_lamb_init_kernel.h"
#include "paddle/phi/kernels/funcs/algorithm.h"
#include "paddle/phi/kernels/fusion/gpu/cast_with_ptr.h"
#include "paddle/phi/kernels/fusion/gpu/distributed_fused_lamb_init_kernel.cu" //NOLINT

PD_CUSTOM_KERNEL_REGISTER(distributed_fused_lamb_init,
iluvatar_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/fusion/gpu/fused_bias_act_kernel.cu" //NOLINT
#include "paddle/phi/kernels/fused_bias_act_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(fused_bias_act,
iluvatar_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,6 @@
#include "paddle/phi/kernels/distributed_fused_lamb_init_kernel.h"
#include "paddle/phi/kernels/funcs/algorithm.h"
#include "paddle/phi/kernels/fusion/gpu/cast_with_ptr.h"
#include "paddle/phi/kernels/fusion/gpu/distributed_fused_lamb_init_kernel.cu" //NOLINT

PD_CUSTOM_KERNEL_REGISTER(distributed_fused_lamb_init,
metax_gpu,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
// limitations under the License.

#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/fusion/gpu/fused_bias_act_kernel.cu" //NOLINT
#include "paddle/phi/kernels/fused_bias_act_kernel.h"

PD_CUSTOM_KERNEL_REGISTER(fused_bias_act,
metax_gpu,
Expand Down
Loading