Skip to content

Commit 28e3e4f

Browse files
committed
Fix
1 parent 1ec3f64 commit 28e3e4f

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

paddle/phi/kernels/funcs/math.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717
#include "math.h" // NOLINT
1818
#include "paddle/common/hostdevice.h"
19+
#include "paddle/phi/common/bfloat16.h"
1920
#include "paddle/phi/common/float16.h"
2021

2122
namespace phi {

paddle/phi/kernels/gpu/distributed_fused_lamb_kernel.cu

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1739,8 +1739,6 @@ void DistributedFusedLambKernel(
17391739
int64_t global_rank = 0, local_rank = 0;
17401740
ncclComm_t global_comm = nullptr, local_comm = nullptr,
17411741
external_comm = nullptr;
1742-
paddle::platform::NCCLComm *nccl_comm_handle = nullptr,
1743-
*local_nccl_comm_handle = nullptr;
17441742
distributed::NCCLCommContext *comm_ctx = nullptr, *local_comm_ctx = nullptr,
17451743
*external_comm_ctx = nullptr;
17461744

0 commit comments

Comments
 (0)