diff options
| author | Frank Seide <fseide@microsoft.com> | 2020-05-21 08:51:18 +0300 |
|---|---|---|
| committer | Frank Seide <fseide@microsoft.com> | 2020-05-21 08:51:18 +0300 |
| commit | 77a420740c79782d251bfe43a10279e3cfb10781 (patch) | |
| tree | 1c275cb393b88f3813347799dfe17aea62031ae5 /CMakeLists.txt | |
| parent | 9842e7b6e29af992274740c7b7d73ec3697a703a (diff) | |
Merged PR 12958: ONNX support
This branch adds functionality to export ONNX models (with limitations).
Diffstat (limited to 'CMakeLists.txt')
| -rw-r--r-- | CMakeLists.txt | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index 4cf8bf92..8e7b643c 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -195,6 +195,12 @@ if(USE_SENTENCEPIECE) set(EXT_LIBS ${EXT_LIBS} sentencepiece sentencepiece_train) endif() +if(USE_ONNX) + message(STATUS "Enabling experimental ONNX support") + set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DUSE_ONNX") + set(EXT_LIBS ${EXT_LIBS} protobuf) +endif() + # Find packages set(EXT_LIBS ${EXT_LIBS} ${CMAKE_DL_LIBS}) |