mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
b282297559
Summary: Fixes https://github.com/pytorch/pytorch/issues/41757 I've replaced all the whitelist with allowlist for this issue. Pull Request resolved: https://github.com/pytorch/pytorch/pull/42067 Reviewed By: pbelevich Differential Revision: D22791690 Pulled By: malfet fbshipit-source-id: 638c13cf49915f5c83bd79c7f4a39b8390cc15b4
33 lines
786 B
CMake
33 lines
786 B
CMake
|
|
if(__caffe2_allowlist_included)
|
|
return()
|
|
endif()
|
|
|
|
set(__caffe2_allowlist_included TRUE)
|
|
|
|
set(CAFFE2_ALLOWLISTED_FILES)
|
|
if(NOT CAFFE2_ALLOWLIST)
|
|
return()
|
|
endif()
|
|
|
|
# First read the allowlist file and break it by line.
|
|
file(READ "${CAFFE2_ALLOWLIST}" allowlist_content)
|
|
# Convert file contents into a CMake list
|
|
string(REGEX REPLACE "\n" ";" allowlist_content ${allowlist_content})
|
|
|
|
foreach(item ${allowlist_content})
|
|
file(GLOB_RECURSE tmp ${item})
|
|
set(CAFFE2_ALLOWLISTED_FILES ${CAFFE2_ALLOWLISTED_FILES} ${tmp})
|
|
endforeach()
|
|
|
|
macro(caffe2_do_allowlist output allowlist)
|
|
set(_tmp)
|
|
foreach(item ${${output}})
|
|
list(FIND ${allowlist} ${item} _index)
|
|
if(${_index} GREATER -1)
|
|
set(_tmp ${_tmp} ${item})
|
|
endif()
|
|
endforeach()
|
|
set(${output} ${_tmp})
|
|
endmacro()
|