mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
9a6c72891b
Summary: In order to pybind, we need transform in core. It's a basically finished product, with a big test suite. It's safe. We can begin hooking up observers after this, and I have a diff coming up that pybinds some apply_transform function. Reviewed By: bwasti Differential Revision: D5522200 fbshipit-source-id: dea6aa606fc689af84e2533569d1ef348cb5f3f2
15 lines
448 B
CMake
15 lines
448 B
CMake
# ---[ Get non-tests
|
|
file(GLOB tmp *.cc)
|
|
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} ${tmp})
|
|
|
|
# exclude test files
|
|
file(GLOB tmp *_test.cc)
|
|
exclude(Caffe2_CPU_SRCS "${Caffe2_CPU_SRCS}" ${tmp})
|
|
|
|
# ---[ Get tests
|
|
file(GLOB tmp *_test.cc)
|
|
set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} ${tmp})
|
|
|
|
# ---[ Send the lists to the parent scope.
|
|
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} PARENT_SCOPE)
|