File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -2,7 +2,7 @@ INCLUDE(ExternalProject)
2
2
3
3
SET (NCCL_SOURCE_DIR ${THIRD_PARTY_PATH} /nccl )
4
4
5
- INCLUDE_DIRECTORIES (${NCCL_SOURCE_DIR} /src/extern_nccl )
5
+ INCLUDE_DIRECTORIES (${NCCL_SOURCE_DIR} /src/extern_nccl/src )
6
6
7
7
8
8
if (WITH_DSO )
Original file line number Diff line number Diff line change @@ -75,7 +75,8 @@ class Im2ColFunctor {
75
75
void operator ()(const platform::DeviceContext& context,
76
76
const framework::Tensor& im, framework::Tensor& col,
77
77
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 );
79
80
};
80
81
81
82
template <ColFormat Format, typename Place, typename T>
@@ -84,7 +85,7 @@ class Col2ImFunctor {
84
85
void operator ()(const platform::DeviceContext& context, framework::Tensor& im,
85
86
const framework::Tensor& col, int stride_height,
86
87
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 );
88
89
};
89
90
90
91
} // namespace math
You can’t perform that action at this time.
0 commit comments