Skip to content

Commit 630644e

Browse files
author
chengduo
authored
Merge pull request #5036 from luotao1/nccl
fix nccl compiler error
2 parents fa72e54 + 35434c3 commit 630644e

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

cmake/external/nccl.cmake

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ INCLUDE(ExternalProject)
22

33
SET(NCCL_SOURCE_DIR ${THIRD_PARTY_PATH}/nccl)
44

5-
INCLUDE_DIRECTORIES(${NCCL_SOURCE_DIR}/src/extern_nccl)
5+
INCLUDE_DIRECTORIES(${NCCL_SOURCE_DIR}/src/extern_nccl/src)
66

77

88
if(WITH_DSO)

paddle/operators/math/im2col.h

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,8 @@ class Im2ColFunctor {
7575
void operator()(const platform::DeviceContext& context,
7676
const framework::Tensor& im, framework::Tensor& col,
7777
int stride_height, int stride_width, int padding_up,
78-
int padding_down, int padding_left, int padding_right);
78+
int padding_down, int padding_left = 0,
79+
int padding_right = 0);
7980
};
8081

8182
template <ColFormat Format, typename Place, typename T>
@@ -84,7 +85,7 @@ class Col2ImFunctor {
8485
void operator()(const platform::DeviceContext& context, framework::Tensor& im,
8586
const framework::Tensor& col, int stride_height,
8687
int stride_width, int padding_up, int padding_down,
87-
int padding_left, int padding_right);
88+
int padding_left = 0, int padding_right = 0);
8889
};
8990

9091
} // namespace math

0 commit comments

Comments
 (0)