mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
45c9ed825a
Summary:
Running commands:
```bash
shopt -s globstar
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i CMakeLists.txt
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i caffe2/**/CMakeLists.txt
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i torch/**/CMakeLists.txt
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i c10/**/CMakeLists.txt
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i cmake/**/*.cmake
sed -e 's/IF (/if(/g' -e 's/IF(/if(/g' -e 's/if (/if(/g' -e 's/ELSE (/else(/g' -e 's/ELSE(/else(/g' -e 's/else (/else(/g' -e 's/ENDif(/endif(/g' -e 's/ELSEif(/elseif(/g' -i cmake/**/*.cmake.in
```
We may further convert all the commands into lowercase according to the following issue: 77543bde41.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35521
Differential Revision: D20704382
Pulled By: malfet
fbshipit-source-id: 42186b9b1660c34428ab7ceb8d3f7a0ced5d2e80
69 lines
2.3 KiB
CMake
69 lines
2.3 KiB
CMake
if(INTERN_BUILD_MOBILE AND NOT BUILD_CAFFE2_MOBILE)
|
|
list(APPEND Caffe2_CPU_SRCS
|
|
"${CMAKE_CURRENT_SOURCE_DIR}/common.cc"
|
|
)
|
|
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE)
|
|
return()
|
|
endif()
|
|
|
|
# ---[ GPU files
|
|
# ------[ cuDNN
|
|
if(USE_CUDNN)
|
|
file(GLOB tmp *_cudnn.cc)
|
|
set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${tmp})
|
|
endif()
|
|
# ------[ general GPU
|
|
file(GLOB tmp *_gpu.cc)
|
|
set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${tmp})
|
|
# ------[ CUDA sources
|
|
file(GLOB tmp *.cu)
|
|
set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${tmp})
|
|
# exclude test files
|
|
file(GLOB tmp *_test.cc)
|
|
exclude(Caffe2_GPU_SRCS "${Caffe2_GPU_SRCS}" ${tmp})
|
|
|
|
# ---[ general HIP files
|
|
file(GLOB tmp hip/*.cc)
|
|
set(Caffe2_HIP_SRCS ${Caffe2_HIP_SRCS} ${tmp})
|
|
# ------[ HIP sources
|
|
file(GLOB tmp hip/*.hip)
|
|
set(Caffe2_HIP_SRCS ${Caffe2_HIP_SRCS} ${tmp})
|
|
# exclude test files
|
|
file(GLOB tmp hip/*_test.cc)
|
|
exclude(Caffe2_HIP_SRCS "${Caffe2_HIP_SRCS}" ${tmp})
|
|
|
|
# ---[ CPU files.
|
|
file(GLOB tmp *.cc)
|
|
# Manually remove the cudnn files since we might be using USE_CUDNN=OFF
|
|
# TODO: when we move to explicit file list, this would not be needed.
|
|
file(GLOB tmp_cudnn *_cudnn.cc)
|
|
exclude(tmp "${tmp}" ${tmp_cudnn})
|
|
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} ${tmp})
|
|
# exclude test files and gpu files
|
|
file(GLOB tmp *_test.cc)
|
|
exclude(Caffe2_CPU_SRCS "${Caffe2_CPU_SRCS}" ${tmp})
|
|
exclude(Caffe2_CPU_SRCS "${Caffe2_CPU_SRCS}" ${Caffe2_GPU_SRCS})
|
|
exclude(Caffe2_CPU_SRCS "${Caffe2_CPU_SRCS}" ${Caffe2_HIP_SRCS})
|
|
|
|
# ---[ GPU test files
|
|
file(GLOB tmp *_gpu_test.cc)
|
|
set(Caffe2_GPU_TEST_SRCS ${Caffe2_GPU_TEST_SRCS} ${tmp})
|
|
|
|
# ---[ HIP test files
|
|
file(GLOB tmp hip/*_test.cc)
|
|
set(Caffe2_HIP_TEST_SRCS ${Caffe2_HIP_TEST_SRCS} ${tmp})
|
|
|
|
# ---[ CPU test files
|
|
file(GLOB tmp *_test.cc)
|
|
set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} ${tmp})
|
|
exclude(Caffe2_CPU_TEST_SRCS "${Caffe2_CPU_TEST_SRCS}" ${Caffe2_GPU_TEST_SRCS})
|
|
exclude(Caffe2_CPU_TEST_SRCS "${Caffe2_CPU_TEST_SRCS}" ${Caffe2_HIP_TEST_SRCS})
|
|
|
|
# ---[ Send the lists to the parent scope.
|
|
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_HIP_SRCS ${Caffe2_HIP_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_GPU_TEST_SRCS ${Caffe2_GPU_TEST_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_HIP_TEST_SRCS ${Caffe2_HIP_TEST_SRCS} PARENT_SCOPE)
|