mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
92f2220589
Summary: This helps adjusting mobile build sizes when necessary. Closes https://github.com/caffe2/caffe2/pull/228 Differential Revision: D4795135 Pulled By: Yangqing fbshipit-source-id: 70a0dc35b31d5c8038081aedeb464e47e4284217
33 lines
790 B
CMake
33 lines
790 B
CMake
|
|
if (__caffe2_whitelist_included)
|
|
return()
|
|
endif()
|
|
|
|
set (__caffe2_whitelist_included TRUE)
|
|
|
|
set(CAFFE2_WHITELISTED_FILES)
|
|
if (NOT CAFFE2_WHITELIST)
|
|
return()
|
|
endif()
|
|
|
|
# First read the whitelist file and break it by line.
|
|
file(READ "${CAFFE2_WHITELIST}" whitelist_content)
|
|
# Convert file contents into a CMake list
|
|
string(REGEX REPLACE "\n" ";" whitelist_content ${whitelist_content})
|
|
|
|
foreach(item ${whitelist_content})
|
|
file(GLOB_RECURSE tmp ${item})
|
|
set(CAFFE2_WHITELISTED_FILES ${CAFFE2_WHITELISTED_FILES} ${tmp})
|
|
endforeach()
|
|
|
|
macro(caffe2_do_whitelist output whitelist)
|
|
set(_tmp)
|
|
foreach(item ${${output}})
|
|
list(FIND ${whitelist} ${item} _index)
|
|
if (${_index} GREATER -1)
|
|
set(_tmp ${_tmp} ${item})
|
|
endif()
|
|
endforeach()
|
|
set(${output} ${_tmp})
|
|
endmacro()
|