mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Update onnx submodule to onnx/onnx@bae6333 (#10961)
Summary: ONNX v1.3.0 release Pull Request resolved: https://github.com/pytorch/pytorch/pull/10961 Reviewed By: houseroad Differential Revision: D9543998 Pulled By: bddppq fbshipit-source-id: b7f0a0553d832d609d3b7613a608f7bf4a2582ef
This commit is contained in:
committed by
Facebook Github Bot
parent
56c737a9b7
commit
302e9cb815
@ -144,6 +144,8 @@ option(USE_DISTRIBUTED "Use THD (distributed)" OFF)
|
||||
# Used when building Caffe2 through setup.py
|
||||
option(BUILDING_WITH_TORCH_LIBS "Tell cmake if Caffe2 is being built alongside torch libs" OFF)
|
||||
|
||||
SET(ONNX_NAMESPACE "onnx_c2" CACHE STRING "onnx namespace")
|
||||
|
||||
if (ANDROID OR IOS)
|
||||
set(BUILD_ATEN_MOBILE ON)
|
||||
endif()
|
||||
|
@ -36,6 +36,7 @@ backend_test.exclude(r'(test_hardsigmoid' # Does not support Hardsigmoid.
|
||||
'|test_.*pool_.*same.*' # Does not support pool same.
|
||||
'|test_maxpool_with_argmax.*' # MaxPool outputs indices in different format.
|
||||
'|test_convtranspose.*' # ConvTranspose needs some more complicated translation
|
||||
'|test_mvn.*' # MeanVarianceNormalization is experimental and not supported.
|
||||
')')
|
||||
|
||||
# Quick patch to unbreak master CI, is working on the debugging.
|
||||
|
@ -772,9 +772,6 @@ endif()
|
||||
|
||||
# ---[ Onnx
|
||||
if (CAFFE2_CMAKE_BUILDING_WITH_MAIN_REPO)
|
||||
if (NOT DEFINED ONNX_NAMESPACE)
|
||||
SET(ONNX_NAMESPACE "onnx_c2")
|
||||
endif()
|
||||
if(EXISTS "${CAFFE2_CUSTOM_PROTOC_EXECUTABLE}")
|
||||
set(ONNX_CUSTOM_PROTOC_EXECUTABLE ${CAFFE2_CUSTOM_PROTOC_EXECUTABLE})
|
||||
endif()
|
||||
@ -782,6 +779,7 @@ if (CAFFE2_CMAKE_BUILDING_WITH_MAIN_REPO)
|
||||
# We will build onnx as static libs and embed it directly into the binary.
|
||||
set(BUILD_SHARED_LIBS OFF)
|
||||
set(ONNX_USE_MSVC_STATIC_RUNTIME ${CAFFE2_USE_MSVC_STATIC_RUNTIME})
|
||||
set(ONNX_USE_LITE_PROTO ${CAFFE2_USE_LITE_PROTO})
|
||||
# If linking local protobuf, make sure ONNX has the same protobuf
|
||||
# patches as Caffe2 and Caffe proto. This forces some functions to
|
||||
# not be inline and instead route back to the statically-linked protobuf.
|
||||
@ -790,7 +788,7 @@ if (CAFFE2_CMAKE_BUILDING_WITH_MAIN_REPO)
|
||||
endif()
|
||||
add_subdirectory(${CMAKE_CURRENT_LIST_DIR}/../third_party/onnx)
|
||||
include_directories(${ONNX_INCLUDE_DIRS})
|
||||
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DONNX_NAMESPACE=${ONNX_NAMESPACE}")
|
||||
add_definitions(-DONNX_NAMESPACE=${ONNX_NAMESPACE})
|
||||
# In mobile build we care about code size, and so we need drop
|
||||
# everything (e.g. checker, optimizer) in onnx but the pb definition.
|
||||
if (ANDROID OR IOS)
|
||||
|
2
third_party/onnx
vendored
2
third_party/onnx
vendored
Submodule third_party/onnx updated: 6146a85d37...bae6333e14
Reference in New Issue
Block a user