| # ---[ GPU files |
| # ------[ cuDNN |
| file(GLOB_RECURSE tmp *_cudnn.cc) |
| set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${tmp}) |
| # ------[ general GPU |
| file(GLOB_RECURSE tmp *_gpu.cc) |
| set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${tmp}) |
| # ------[ CUDA sources |
| file(GLOB_RECURSE tmp *.cu) |
| set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} ${tmp}) |
| # exclude test files |
| file(GLOB_RECURSE tmp *_test.cc) |
| exclude(Caffe2_GPU_SRCS "${Caffe2_GPU_SRCS}" ${tmp}) |
| |
| # ---[ CPU files. |
| file(GLOB_RECURSE tmp *.cc) |
| set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} ${tmp}) |
| # exclude test files and gpu files |
| file(GLOB_RECURSE tmp *_test.cc) |
| exclude(Caffe2_CPU_SRCS "${Caffe2_CPU_SRCS}" ${tmp}) |
| exclude(Caffe2_CPU_SRCS "${Caffe2_CPU_SRCS}" ${Caffe2_GPU_SRCS}) |
| |
| # ---[ threadpool/pthreadpool* is a local modification of the NNPACK |
| # pthreadpool with a very similar interface. Neither NNPACK, nor this |
| # thread pool supports Windows. |
| if (MSVC) |
| file(GLOB_RECURSE tmp pthreadpool*) |
| exclude(Caffe2_CPU_SRCS "${Caffe2_CPU_SRCS}" ${tmp}) |
| endif() |
| |
| # ---[ GPU test files |
| file(GLOB_RECURSE tmp *_gpu_test.cc) |
| set(Caffe2_GPU_TEST_SRCS ${Caffe2_GPU_TEST_SRCS} ${tmp}) |
| |
| # ---[ CPU test files |
| file(GLOB_RECURSE 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}) |
| |
| # ---[ 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_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} PARENT_SCOPE) |
| set(Caffe2_GPU_TEST_SRCS ${Caffe2_GPU_TEST_SRCS} PARENT_SCOPE) |
| |
| # ---[ Printing stack traces requires dladdr |
| |
