mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
5d6dacaafe
Summary: Make operator QuantDecompZstd buildable in open source. The operator is not built by default. Need to specify -DBUILD_SHARE_DIR=ON -DUSE_ZSTD=ON to build it. Test plans: Build android caffe2 with the change without issue. Run a model with the operator successfully. Closes https://github.com/caffe2/caffe2/pull/1613 Reviewed By: Yangqing Differential Revision: D6556723 Pulled By: sf-wind fbshipit-source-id: 453a7d787a55928f2dea1ed2b99f2df011aa8d26
16 lines
638 B
CMake
16 lines
638 B
CMake
# There is a linking issue that happens in some of the Windows builds.
|
|
# TODO(Yangqing): after the module redesing, enable this back.
|
|
if (NOT MSVC)
|
|
add_subdirectory(contrib)
|
|
endif()
|
|
|
|
# CPU source, test sources, binary sources
|
|
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_CPU_BINARY_SRCS ${Caffe2_CPU_BINARY_SRCS} PARENT_SCOPE)
|
|
|
|
# GPU source, test sources, binary sources
|
|
set(Caffe2_GPU_SRCS ${Caffe2_GPU_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_GPU_TEST_SRCS ${Caffe2_GPU_TEST_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_GPU_BINARY_SRCS ${Caffe2_GPU_BINARY_SRCS} PARENT_SCOPE)
|