mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
1cd166d330
Summary: Closes https://github.com/caffe2/caffe2/pull/88 Differential Revision: D4404292 Pulled By: bwasti fbshipit-source-id: 8a4351c2dee5136aaa12b90f1a61fd7afee51994
16 lines
508 B
CMake
16 lines
508 B
CMake
# ---[ CPU files.
|
|
set(Caffe2_CPU_PYTHON_SRCS
|
|
"/pybind_state.cc"
|
|
"/pybind_state_mkl.cc"
|
|
)
|
|
# ---[ GPU files
|
|
set(Caffe2_GPU_PYTHON_SRCS
|
|
${Caffe2_CPU_PYTHON_SRCS}
|
|
"/pybind_state_gpu.cc"
|
|
)
|
|
|
|
prepend(Caffe2_CPU_PYTHON_SRCS ${CMAKE_CURRENT_SOURCE_DIR} ${Caffe2_CPU_PYTHON_SRCS})
|
|
prepend(Caffe2_GPU_PYTHON_SRCS ${CMAKE_CURRENT_SOURCE_DIR} ${Caffe2_GPU_PYTHON_SRCS})
|
|
|
|
set(Caffe2_CPU_PYTHON_SRCS ${Caffe2_CPU_PYTHON_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_GPU_PYTHON_SRCS ${Caffe2_GPU_PYTHON_SRCS} PARENT_SCOPE)
|