mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 00:20:18 +01:00
c373387709
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/62445 PyTorch currently uses the old style of compiling CUDA in CMake which is just a bunch of scripts in `FindCUDA.cmake`. Newer versions support CUDA natively as a language just like C++ or C. Test Plan: Imported from OSS Reviewed By: ejguan Differential Revision: D31503350 fbshipit-source-id: 2ee817edc9698531ae1b87eda3ad271ee459fd55
26 lines
809 B
CMake
26 lines
809 B
CMake
# CMake file to replace the string contents in Google Test and Google Mock
|
|
# Usage example:
|
|
# Patch the cmake file
|
|
# cmake -DFILENAME=internal_utils.cmake
|
|
# -DBACKUP=internal_utils.cmake.bak
|
|
# -DREVERT=0
|
|
# -P GoogleTestPatch.cmake
|
|
# Revert the changes
|
|
# cmake -DFILENAME=internal_utils.cmake
|
|
# -DBACKUP=internal_utils.cmake.bak
|
|
# -DREVERT=1
|
|
# -P GoogleTestPatch.cmake
|
|
|
|
|
|
if(REVERT)
|
|
file(READ ${BACKUP} content)
|
|
file(WRITE ${FILENAME} "${content}")
|
|
file(REMOVE ${BACKUP})
|
|
else(REVERT)
|
|
file(READ ${FILENAME} content)
|
|
file(WRITE ${BACKUP} "${content}")
|
|
string(REGEX REPLACE "[-/]Z[iI]" "/Z7" content "${content}")
|
|
string(REGEX REPLACE "Threads::Threads" "caffe2::Threads" content "${content}")
|
|
file(WRITE ${FILENAME} "${content}")
|
|
endif(REVERT)
|