OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fix pytorch linux build issues (#9273)

Browse Source 
Summary:
Breaking out of #8338

This fixes the build issues with pytorch on linux machines after BUILD_CAFFE2 and BUILD_ATEN are removed.

cc orionr
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9273

Reviewed By: orionr

Differential Revision: D8768869

Pulled By: mingzhe09088

fbshipit-source-id: 2730426ed1bed398eb5dc804c7348aeeb27c93d3
This commit is contained in:
Mingzhe Li 2018-07-09 14:37:05 -07:00 committed by Facebook Github Bot
parent d0ad696f9d
commit a70a90b28f
7 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CMakeLists.txt
View File

@ -201,12 +201,14 @@ if(NOT MSVC)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wextra")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-missing-field-initializers")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-type-limits")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-array-bounds")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unknown-pragmas")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-sign-compare")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-parameter")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-variable")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-function")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-result")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-strict-aliasing")
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-error=deprecated-declarations")
# These flags are not available in GCC-4.8.5. Set only when using clang.
# Compared against https://gcc.gnu.org/onlinedocs/gcc-4.8.5/gcc/Option-Summary.html

2
aten/src/THNN/generic/SpatialMaxUnpooling.c
View File

@ -10,7 +10,7 @@ static void THNN_(SpatialMaxUnpooling_updateOutput_frame)(real *input_p, real *o
{
int k;
int has_error = 0;
THIndex_t error_index;
THIndex_t error_index = 0;
#pragma omp parallel for private(k)
for (k = 0; k < nslices; k++)
{

2
aten/src/THNN/generic/VolumetricMaxUnpooling.c
View File

@ -67,7 +67,7 @@ static void THNN_(VolumetricMaxUnpooling_updateOutput_frame)(
{
int k;
int has_error = 0;
THIndex_t error_index;
THIndex_t error_index = 0;
#pragma omp parallel for private(k)
for (k = 0; k < nslices; k++)
{

2
caffe2/operators/abs_op.cc
View File

@ -78,7 +78,7 @@ Y: [0.3005476 1.551666 1.3591481 0.39191285 0.21866608]
</details>
)DOC")
.Input(0, "X", "*(type: Tensor<float\\>)* Input tensor.")
.Input(0, "X", "*(type: Tensor<float>)* Input tensor.")
.Output(
0,
"Y",

2
caffe2/operators/pack_segments.cc
View File

@ -132,7 +132,7 @@ bool UnpackSegmentsOp<CPUContext>::DoRunWithType2() {
output->Resize(shape);
// create output tensor
auto* out = static_cast<char*>(output->raw_mutable_data(data.meta()));
if (!(data.dim(0) * data.dim(1))) {
if (!(data.dim(0) && data.dim(1))) {
return true;
}
auto block_size = data.size_from_dim(2);

4
caffe2/operators/sequence_ops.cc
View File

@ -385,7 +385,7 @@ lengths_out: [5]
.Output(
0,
"data_out",
"*(type: Tensor)* Padded data tensor ($T<N + 2*padding\\_width, "
"*(type: Tensor)* Padded data tensor ($T<N + 2*padding_width, "
"D_1, ..., D_n>$).")
.Output(
1,
@ -483,7 +483,7 @@ lengths_out_rm: [3]
0,
"data_out",
"*(type: Tensor)* Padded data tensor "
"($T<N + 2*padding\\_width, D_1, ..., D_n>$).")
"($T<N + 2*padding_width, D_1, ..., D_n>$).")
.Output(
1,
"lengths_out",

6
cmake/Utils.cmake
View File

@ -276,10 +276,11 @@ function(target_enable_style_warnings TARGET)
-Wmissing-include-dirs
-Woverloaded-virtual
-Wredundant-decls
-Wno-shadow
-Wsign-promo
-Wstrict-overflow=5
-fdiagnostics-show-option
-Wconversion
-Wno-conversion
-Wpedantic
-Wundef
)
@ -301,5 +302,8 @@ function(target_enable_style_warnings TARGET)
list(APPEND WARNING_OPTIONS "-Werror")
endif()
endif()
if(APPLE)
set(WARNING_OPTIONS -Wno-gnu-zero-variadic-macro-arguments)
endif()
target_compile_options(${TARGET} PRIVATE ${WARNING_OPTIONS})
endfunction()