diff --git a/CMakeLists.txt b/CMakeLists.txt index aeb87eaa7e3..a842fe3ba9e 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -135,6 +135,7 @@ option(BUILD_TEST "Build C++ test binaries (need gtest and gbenchmark)" OFF) option(BUILD_MOBILE_BENCHMARKS "Build C++ test binaries for mobile (ARM) targets(need gtest and gbenchmark)" OFF) option(BUILD_MOBILE_TEST "Build C++ test binaries for mobile (ARM) targets(need gtest and gbenchmark)" OFF) option(BUILD_JNI "Build JNI bindings" OFF) +option(BUILD_MOBILE_AUTOGRAD "Build autograd function in mobile build (in development)" OFF) cmake_dependent_option( INSTALL_TEST "Install test binaries if BUILD_TEST is on" ON "BUILD_TEST" OFF) @@ -397,6 +398,11 @@ if(INTERN_BUILD_MOBILE AND NOT BUILD_CAFFE2_MOBILE) if(NOT BUILD_SHARED_LIBS AND NOT "${SELECTED_OP_LIST}" STREQUAL "") string(APPEND CMAKE_CXX_FLAGS " -DNO_EXPORT") endif() + if(BUILD_MOBILE_AUTOGRAD) + set(INTERN_DISABLE_AUTOGRAD OFF) + else() + set(INTERN_DISABLE_AUTOGRAD ON) + endif() set(BUILD_PYTHON OFF) set(BUILD_CAFFE2_OPS OFF) set(USE_DISTRIBUTED OFF) @@ -405,7 +411,6 @@ if(INTERN_BUILD_MOBILE AND NOT BUILD_CAFFE2_MOBILE) set(USE_FBGEMM OFF) set(USE_QNNPACK OFF) set(INTERN_DISABLE_ONNX ON) - set(INTERN_DISABLE_AUTOGRAD ON) set(INTERN_USE_EIGEN_BLAS ON) # Disable developing mobile interpreter for actual mobile build. # Enable it elsewhere to capture build error. diff --git a/cmake/Summary.cmake b/cmake/Summary.cmake index cfd024c80c0..c8750a272fe 100644 --- a/cmake/Summary.cmake +++ b/cmake/Summary.cmake @@ -45,6 +45,7 @@ function(caffe2_print_configuration_summary) message(STATUS " BUILD_SHARED_LIBS : ${BUILD_SHARED_LIBS}") message(STATUS " BUILD_TEST : ${BUILD_TEST}") message(STATUS " BUILD_JNI : ${BUILD_JNI}") + message(STATUS " BUILD_MOBILE_AUTOGRAD : ${BUILD_MOBILE_AUTOGRAD}") message(STATUS " INTERN_BUILD_MOBILE : ${INTERN_BUILD_MOBILE}")