提交 630644e9 编写于 作者: C chengduo 提交者: GitHub

Merge pull request #5036 from luotao1/nccl

fix nccl compiler error
......@@ -2,7 +2,7 @@ INCLUDE(ExternalProject)
SET(NCCL_SOURCE_DIR ${THIRD_PARTY_PATH}/nccl)
INCLUDE_DIRECTORIES(${NCCL_SOURCE_DIR}/src/extern_nccl)
INCLUDE_DIRECTORIES(${NCCL_SOURCE_DIR}/src/extern_nccl/src)
if(WITH_DSO)
......
......@@ -75,7 +75,8 @@ class Im2ColFunctor {
void operator()(const platform::DeviceContext& context,
const framework::Tensor& im, framework::Tensor& col,
int stride_height, int stride_width, int padding_up,
int padding_down, int padding_left, int padding_right);
int padding_down, int padding_left = 0,
int padding_right = 0);
};
template <ColFormat Format, typename Place, typename T>
......@@ -84,7 +85,7 @@ class Col2ImFunctor {
void operator()(const platform::DeviceContext& context, framework::Tensor& im,
const framework::Tensor& col, int stride_height,
int stride_width, int padding_up, int padding_down,
int padding_left, int padding_right);
int padding_left = 0, int padding_right = 0);
};
} // namespace math
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册